From e633afac1ee0c06fb46963bc8057f8b145b8d487 Mon Sep 17 00:00:00 2001 From: Kurbanov Bulat Date: Sat, 20 Nov 2021 12:37:19 +0300 Subject: [PATCH] Bump --- src/app/services/updaters/__init__.py | 6 +++--- .../updaters/{flibusta_updater.py => fl_updater.py} | 4 ++-- src/core/config.py | 2 ++ 3 files changed, 7 insertions(+), 5 deletions(-) rename src/app/services/updaters/{flibusta_updater.py => fl_updater.py} (99%) diff --git a/src/app/services/updaters/__init__.py b/src/app/services/updaters/__init__.py index 289f7f3..0716e22 100644 --- a/src/app/services/updaters/__init__.py +++ b/src/app/services/updaters/__init__.py @@ -1,13 +1,13 @@ from enum import Enum from app.services.updaters.base import BaseUpdater -from app.services.updaters.flibusta_updater import FlibustaUpdater +from app.services.updaters.fl_updater import FlUpdater class UpdaterTypes(Enum): - FLIBUSTA = 'flibusta' + FL = 'fl' UPDATERS: dict[UpdaterTypes, BaseUpdater] = { - UpdaterTypes.FLIBUSTA: FlibustaUpdater + UpdaterTypes.FL: FlUpdater } diff --git a/src/app/services/updaters/flibusta_updater.py b/src/app/services/updaters/fl_updater.py similarity index 99% rename from src/app/services/updaters/flibusta_updater.py rename to src/app/services/updaters/fl_updater.py index b4bbae9..c36ad1e 100644 --- a/src/app/services/updaters/flibusta_updater.py +++ b/src/app/services/updaters/fl_updater.py @@ -28,7 +28,7 @@ def remove_dots(s: str): return s.replace('.', '') -class FlibustaUpdater(BaseUpdater): +class FlUpdater(BaseUpdater): SOURCE: int FILES = [ @@ -83,7 +83,7 @@ class FlibustaUpdater(BaseUpdater): async def _import_dump(self, filename: str): result = await run( - f"wget -O - http://flibusta.is/sql/{filename}.gz | gunzip | " + f"wget -O - {env_config.FL_BASE_URL}/sql/{filename}.gz | gunzip | " f"mysql -h {env_config.MYSQL_HOST} -u {env_config.MYSQL_USER} " f"-p\"{env_config.MYSQL_PASSWORD}\" {env_config.MYSQL_DB_NAME}" ) diff --git a/src/core/config.py b/src/core/config.py index d7a2ed1..41b0df2 100644 --- a/src/core/config.py +++ b/src/core/config.py @@ -16,5 +16,7 @@ class EnvConfig(BaseSettings): MYSQL_USER: str MYSQL_PASSWORD: str + FL_BASE_URL: str + env_config = EnvConfig()