whois/worker.py: fix outdated settings

Signed-off-by: Eduard Kaverinskyi <[email protected]>
hspsh · Jan 18, 2025 · 4fe5c5c · 4fe5c5c
1 parent 3972f3d
commit 4fe5c5c
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/whois/worker.py b/whois/worker.py
@@ -4,15 +4,15 @@
 from whois.data.db.database import Database
 from whois.data.repository.device_repository import DeviceRepository, Device
 from whois.mikrotik import fetch_leases
-from whois.settings import production
+from whois.settings.production import mikrotik_settings
 
 logger = logging.getLogger("mikrotik-worker")
 database = Database()
 device_repository = DeviceRepository(database)
 
 def update_devices() -> int:
     leases = fetch_leases(
-        production.MIKROTIK_URL, production.MIKROTIK_USER, production.MIKROTIK_PASS
+        mikrotik_settings.MIKROTIK_URL, mikrotik_settings.MIKROTIK_USER, mikrotik_settings.MIKROTIK_PASS
     )
 
     for lease in leases:
@@ -27,7 +27,7 @@ def update_devices() -> int:
 
 def run_worker():
     if not all(
-        [production.MIKROTIK_URL, production.MIKROTIK_USER, production.MIKROTIK_PASS]
+        [mikrotik_settings.MIKROTIK_URL, mikrotik_settings.MIKROTIK_USER, mikrotik_settings.MIKROTIK_PASS]
     ):
         raise ValueError("Mikrotik settings not set")
 
@@ -39,7 +39,7 @@ def run_worker():
         except Exception:
             logger.exception("Could not update device information")
 
-        time.sleep(production.worker_frequency_s)
+        time.sleep(mikrotik_settings.WORKER_FREQUENCY_S)
 
 
 if __name__ == "__main__":