mirror of
https://github.com/Garmelon/PFERD.git
synced 2023-12-21 10:23:01 +01:00
75 lines
2.1 KiB
Python
75 lines
2.1 KiB
Python
import configparser
|
|
from abc import ABC, abstractmethod
|
|
from contextlib import asynccontextmanager
|
|
from pathlib import Path
|
|
# TODO In Python 3.9 and above, AsyncContextManager is deprecated
|
|
from typing import AsyncContextManager, AsyncIterator, Optional
|
|
|
|
from rich.markup import escape
|
|
|
|
from .conductor import ProgressBar, TerminalConductor
|
|
from .limiter import Limiter
|
|
from .transformer import RuleParseException, Transformer
|
|
|
|
|
|
class CrawlerLoadException(Exception):
|
|
pass
|
|
|
|
|
|
class Crawler(ABC):
|
|
def __init__(self, name: str, section: configparser.SectionProxy) -> None:
|
|
"""
|
|
May throw a CrawlerLoadException.
|
|
"""
|
|
|
|
self.name = name
|
|
|
|
self._conductor = TerminalConductor()
|
|
self._limiter = Limiter()
|
|
|
|
try:
|
|
self._transformer = Transformer(section.get("transform", ""))
|
|
except RuleParseException as e:
|
|
e.pretty_print()
|
|
raise CrawlerLoadException()
|
|
|
|
# output_dir = Path(section.get("output_dir", name))
|
|
|
|
def print(self, text: str) -> None:
|
|
self._conductor.print(text)
|
|
|
|
@asynccontextmanager
|
|
async def progress_bar(
|
|
self,
|
|
desc: str,
|
|
total: Optional[int] = None,
|
|
) -> AsyncIterator[ProgressBar]:
|
|
async with self._limiter.limit():
|
|
with self._conductor.progress_bar(desc, total=total) as bar:
|
|
yield bar
|
|
|
|
def crawl_bar(self, path: Path) -> AsyncContextManager[ProgressBar]:
|
|
pathstr = escape(str(path))
|
|
desc = f"[bold magenta]Crawling[/bold magenta] {pathstr}"
|
|
return self.progress_bar(desc)
|
|
|
|
def download_bar(
|
|
self,
|
|
path: Path,
|
|
size: int,
|
|
) -> AsyncContextManager[ProgressBar]:
|
|
pathstr = escape(str(path))
|
|
desc = f"[bold green]Downloading[/bold green] {pathstr}"
|
|
return self.progress_bar(desc, total=size)
|
|
|
|
async def run(self) -> None:
|
|
await self._conductor.start()
|
|
try:
|
|
await self.crawl()
|
|
finally:
|
|
await self._conductor.stop()
|
|
|
|
@abstractmethod
|
|
async def crawl(self) -> None:
|
|
pass
|