38 lines
988 B
Python
38 lines
988 B
Python
__package__ = 'abx_plugin_wget'
|
|
|
|
from pathlib import Path
|
|
|
|
from abx_pkg import BinName
|
|
|
|
from abx_spec_extractor import BaseExtractor, ExtractorName
|
|
|
|
from .binaries import WGET_BINARY
|
|
from .wget_util import wget_output_path
|
|
|
|
class WgetExtractor(BaseExtractor):
|
|
name: ExtractorName = 'wget'
|
|
binary: BinName = WGET_BINARY.name
|
|
|
|
def get_output_path(self, snapshot) -> str:
|
|
# wget_index_path = wget_output_path(snapshot.as_link())
|
|
# if wget_index_path:
|
|
# return Path(wget_index_path)
|
|
return 'wget'
|
|
|
|
WGET_EXTRACTOR = WgetExtractor()
|
|
|
|
|
|
class WarcExtractor(BaseExtractor):
|
|
name: ExtractorName = 'warc'
|
|
binary: BinName = WGET_BINARY.name
|
|
|
|
def get_output_path(self, snapshot) -> Path | None:
|
|
warc_files = list((Path(snapshot.link_dir) / 'warc').glob('*.warc.gz'))
|
|
if warc_files:
|
|
return sorted(warc_files, key=lambda x: x.stat().st_size, reverse=True)[0]
|
|
return None
|
|
|
|
|
|
WARC_EXTRACTOR = WarcExtractor()
|
|
|