mirror of
https://github.com/ArchiveBox/ArchiveBox.git
synced 2025-05-13 14:44:29 -04:00
94 lines
2.8 KiB
Python
94 lines
2.8 KiB
Python
from typing import List, Optional
|
|
from pathlib import Path
|
|
|
|
from django.apps import AppConfig
|
|
from django.core.checks import Tags, Warning, register
|
|
|
|
from pydantic import (
|
|
Field,
|
|
SerializeAsAny,
|
|
)
|
|
|
|
from pydantic_pkgr import BinProvider, BinName, Binary, EnvProvider, NpmProvider
|
|
|
|
from plugantic.extractors import Extractor, ExtractorName
|
|
from plugantic.plugins import Plugin
|
|
from plugantic.configs import ConfigSet, ConfigSectionName
|
|
|
|
|
|
###################### Config ##########################
|
|
|
|
class SinglefileToggleConfig(ConfigSet):
|
|
section: ConfigSectionName = 'ARCHIVE_METHOD_TOGGLES'
|
|
|
|
SAVE_SINGLEFILE: bool = True
|
|
|
|
|
|
class SinglefileDependencyConfig(ConfigSet):
|
|
section: ConfigSectionName = 'DEPENDENCY_CONFIG'
|
|
|
|
SINGLEFILE_BINARY: str = Field(default='wget')
|
|
SINGLEFILE_ARGS: Optional[List[str]] = Field(default=None)
|
|
SINGLEFILE_EXTRA_ARGS: List[str] = []
|
|
SINGLEFILE_DEFAULT_ARGS: List[str] = ['--timeout={TIMEOUT-10}']
|
|
|
|
class SinglefileOptionsConfig(ConfigSet):
|
|
section: ConfigSectionName = 'ARCHIVE_METHOD_OPTIONS'
|
|
|
|
# loaded from shared config
|
|
SINGLEFILE_USER_AGENT: str = Field(default='', alias='USER_AGENT')
|
|
SINGLEFILE_TIMEOUT: int = Field(default=60, alias='TIMEOUT')
|
|
SINGLEFILE_CHECK_SSL_VALIDITY: bool = Field(default=True, alias='CHECK_SSL_VALIDITY')
|
|
SINGLEFILE_RESTRICT_FILE_NAMES: str = Field(default='windows', alias='RESTRICT_FILE_NAMES')
|
|
SINGLEFILE_COOKIES_FILE: Optional[Path] = Field(default=None, alias='COOKIES_FILE')
|
|
|
|
|
|
|
|
DEFAULT_CONFIG = {
|
|
'CHECK_SSL_VALIDITY': False,
|
|
'SAVE_SINGLEFILE': True,
|
|
'TIMEOUT': 120,
|
|
}
|
|
|
|
PLUGIN_CONFIG = [
|
|
SinglefileToggleConfig(**DEFAULT_CONFIG),
|
|
SinglefileDependencyConfig(**DEFAULT_CONFIG),
|
|
SinglefileOptionsConfig(**DEFAULT_CONFIG),
|
|
]
|
|
|
|
###################### Binaries ############################
|
|
|
|
class SinglefileBinary(Binary):
|
|
name: BinName = 'single-file'
|
|
providers_supported: List[BinProvider] = [EnvProvider(), NpmProvider()]
|
|
|
|
|
|
###################### Extractors ##########################
|
|
|
|
class SinglefileExtractor(Extractor):
|
|
name: ExtractorName = 'singlefile'
|
|
binary: Binary = SinglefileBinary()
|
|
|
|
def get_output_path(self, snapshot) -> Path:
|
|
return Path(snapshot.link_dir) / 'singlefile.html'
|
|
|
|
|
|
###################### Plugins #############################
|
|
|
|
|
|
class SinglefilePlugin(Plugin):
|
|
name: str = 'singlefile'
|
|
configs: List[SerializeAsAny[ConfigSet]] = [*PLUGIN_CONFIG]
|
|
binaries: List[SerializeAsAny[Binary]] = [SinglefileBinary()]
|
|
extractors: List[SerializeAsAny[Extractor]] = [SinglefileExtractor()]
|
|
|
|
PLUGINS = [SinglefilePlugin()]
|
|
|
|
###################### Django Apps #########################
|
|
|
|
class SinglefileConfig(AppConfig):
|
|
name = 'builtin_plugins.singlefile'
|
|
verbose_name = 'SingleFile'
|
|
|
|
def ready(self):
|
|
print('Loaded singlefile plugin')
|