hyperboria/nexus/pylon/drivers/base.py

from typing import (
    Dict,
    List,
    Optional,
)

from nexus.pylon.network_agent import NetworkAgent
from nexus.pylon.prepared_request import PreparedRequest
from nexus.pylon.proxy_manager import ProxyManager
from nexus.pylon.validators.base import BaseValidator
from utils.izihawa_utils.importlib import import_object


class BaseDriver(NetworkAgent):
    def __init__(
        self,
        validator=None,
        downloads_directory: str = '/downloads',
        proxy_list: Optional[List] = None,
        proxy_manager: Optional[ProxyManager] = None,
    ):
        super().__init__(proxy_list=proxy_list, proxy_manager=proxy_manager)

        validator_cls = 'nexus.pylon.validators.PdfValidator'
        if validator and 'class' in validator:
            validator_cls = validator['class']
        validator_cls = import_object(validator_cls)

        self.validator = validator_cls or BaseValidator
        self.downloads_directory = downloads_directory

    def __str__(self):
        return self.__class__.__name__

    async def execute_prepared_file_request(
        self,
        prepared_file_request: PreparedRequest,
        params: Dict,
    ):
        raise NotImplementedError()