diff --git a/whois/worker.py b/whois/worker.py index d8d1a9e..383a013 100644 --- a/whois/worker.py +++ b/whois/worker.py @@ -4,7 +4,7 @@ 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() @@ -12,7 +12,7 @@ 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__":