diff --git a/app/terraform/block_external.py b/app/terraform/block_external.py index 3afcee3..de3003f 100644 --- a/app/terraform/block_external.py +++ b/app/terraform/block_external.py @@ -1,5 +1,3 @@ -from typing import List, Dict - from bs4 import BeautifulSoup import requests @@ -16,12 +14,12 @@ class BlockExternalAutomation(BlockMirrorAutomation): _content: bytes - def _fetch(self) -> None: + def fetch(self) -> None: user_agent = {'User-agent': 'BypassCensorship/1.0'} page = requests.get(app.config['EXTERNAL_CHECK_URL'], headers=user_agent) self._content = page.content - def _parse(self) -> None: + def parse(self) -> None: soup = BeautifulSoup(self._content, 'html.parser') h2 = soup.find_all('h2') # pylint: disable=invalid-name div = soup.find_all('div', class_="overflow-auto mb-5") diff --git a/app/terraform/block_mirror.py b/app/terraform/block_mirror.py index cbd22e3..2f51626 100644 --- a/app/terraform/block_mirror.py +++ b/app/terraform/block_mirror.py @@ -50,8 +50,8 @@ class BlockMirrorAutomation(BaseAutomation): "\n".join([f"* {proxy_domain} ({origin_domain})" for proxy_domain, origin_domain in rotated])) ) db.session.add(activity) - db.session.commit() activity.notify() + db.session.commit() return True, "" @abstractmethod