mirror of
https://github.com/Garmelon/PFERD.git
synced 2023-12-21 10:23:01 +01:00
Remove dummy crawler
This commit is contained in:
parent
cec0a8e1fc
commit
595ba8b7ab
@ -3,10 +3,8 @@ from typing import Callable, Dict
|
||||
|
||||
from ..config import Config
|
||||
from ..crawler import Crawler, CrawlerSection
|
||||
from .dummy import DummyCrawler
|
||||
from .local import LocalCrawler, LocalCrawlerSection
|
||||
|
||||
CRAWLERS: Dict[str, Callable[[str, Config, SectionProxy], Crawler]] = {
|
||||
"dummy": lambda n, c, s: DummyCrawler(n, c, CrawlerSection(s)),
|
||||
"local": lambda n, c, s: LocalCrawler(n, c, LocalCrawlerSection(s)),
|
||||
}
|
||||
|
@ -1,59 +0,0 @@
|
||||
import asyncio
|
||||
import random
|
||||
from pathlib import PurePath
|
||||
from typing import Any
|
||||
|
||||
from rich.markup import escape
|
||||
|
||||
from ..crawler import Crawler
|
||||
from ..utils import ainput
|
||||
|
||||
DUMMY_TREE = {
|
||||
"Blätter": {
|
||||
"Blatt_01.pdf": (),
|
||||
"Blatt_02.pdf": (),
|
||||
"Blatt_03.pdf": (),
|
||||
"Blatt_04.pdf": (),
|
||||
"Blatt_05.pdf": (),
|
||||
"Lösungen": {
|
||||
"Blatt_01_Lösung.pdf": (),
|
||||
"Blatt_02_Lösung.pdf": (),
|
||||
"Blatt_03_Lösung.pdf": True,
|
||||
"Blatt_04_Lösung.pdf": (),
|
||||
"Blatt_05_Lösung.pdf": (),
|
||||
},
|
||||
},
|
||||
"Vorlesungsfolien": {
|
||||
"VL_01.pdf": (),
|
||||
"VL_02.pdf": (),
|
||||
"VL_03.pdf": (),
|
||||
"VL_04.pdf": (),
|
||||
"VL_05.pdf": (),
|
||||
},
|
||||
"noch_mehr.txt": (),
|
||||
"dateien.jar": (),
|
||||
}
|
||||
|
||||
|
||||
class DummyCrawler(Crawler):
|
||||
async def crawl(self) -> None:
|
||||
await self._crawl_entry(PurePath(), DUMMY_TREE)
|
||||
|
||||
async def _crawl_entry(self, path: PurePath, value: Any) -> None:
|
||||
if value is True:
|
||||
async with self.exclusive_output():
|
||||
await ainput(f"File {path}, please press enter: ")
|
||||
if value == () or value is True:
|
||||
n = random.randint(5, 20)
|
||||
async with self.download_bar(path, n) as bar:
|
||||
await asyncio.sleep(random.random() / 2)
|
||||
for i in range(n):
|
||||
await asyncio.sleep(0.5)
|
||||
bar.advance()
|
||||
self.print(f"[green]Downloaded {escape(str(path))}")
|
||||
else:
|
||||
t = random.random() * 2 + 1
|
||||
async with self.crawl_bar(path) as bar:
|
||||
await asyncio.sleep(t)
|
||||
tasks = [self._crawl_entry(path / k, v) for k, v in value.items()]
|
||||
await asyncio.gather(*tasks)
|
Loading…
Reference in New Issue
Block a user