mirror of
https://github.com/nexus-stc/hyperboria
synced 2024-12-03 00:12:54 +01:00
41 lines
1.2 KiB
Python
41 lines
1.2 KiB
Python
|
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()
|