block: add typing hints

This commit is contained in:
Iain Learmonth 2022-06-18 13:17:36 +01:00
parent 5a00d59398
commit d6b1273cce
2 changed files with 8 additions and 8 deletions

View file

@ -14,7 +14,7 @@ class BlockMirrorAutomation(BaseAutomation):
patterns: List[str] patterns: List[str]
def __init__(self): def __init__(self) -> None:
""" """
Constructor method. Constructor method.
""" """
@ -46,8 +46,8 @@ class BlockMirrorAutomation(BaseAutomation):
if rotated: if rotated:
activity = Activity( activity = Activity(
activity_type="block", activity_type="block",
text=(f"[{self.short_name}] ♻ Rotated {len(rotated)} proxies: \n" + text=(f"[{self.short_name}] ♻ Rotated {len(rotated)} proxies: \n"
"\n".join([f"* {proxy_domain} ({origin_domain})" for proxy_domain, origin_domain in rotated])) + "\n".join([f"* {proxy_domain} ({origin_domain})" for proxy_domain, origin_domain in rotated]))
) )
db.session.add(activity) db.session.add(activity)
activity.notify() activity.notify()
@ -55,7 +55,7 @@ class BlockMirrorAutomation(BaseAutomation):
return True, "" return True, ""
@abstractmethod @abstractmethod
def fetch(self): def fetch(self) -> None:
""" """
Fetch the blocklist data. It is the responsibility of the automation task Fetch the blocklist data. It is the responsibility of the automation task
to persist this within the object for the parse step. to persist this within the object for the parse step.
@ -64,7 +64,7 @@ class BlockMirrorAutomation(BaseAutomation):
""" """
@abstractmethod @abstractmethod
def parse(self): def parse(self) -> None:
""" """
Parse the blocklist data. Parse the blocklist data.
@ -73,7 +73,7 @@ class BlockMirrorAutomation(BaseAutomation):
def active_proxies() -> List[Proxy]: def active_proxies() -> List[Proxy]:
return Proxy.query.filter( return Proxy.query.filter( # type: ignore[no-any-return]
Proxy.deprecated.is_(None), Proxy.deprecated.is_(None),
Proxy.destroyed.is_(None) Proxy.destroyed.is_(None)
).all() ).all()

View file

@ -22,8 +22,8 @@ class BlockRoskomsvobodaAutomation(BlockMirrorAutomation):
_data: Any _data: Any
def fetch(self): def fetch(self) -> None:
self._data = requests.get("https://reestr.rublacklist.net/api/v2/domains/json").json() self._data = requests.get("https://reestr.rublacklist.net/api/v2/domains/json").json()
def parse(self): def parse(self) -> None:
self.patterns.extend(["https://" + pattern for pattern in self._data]) self.patterns.extend(["https://" + pattern for pattern in self._data])