Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fix mikrotik worker v1.6.3 #85

Open
wants to merge 18 commits into
base: master
Choose a base branch
from
Open
Changes from 1 commit
Commits
Show all changes
18 commits
Select commit Hold shift + click to select a range
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Next Next commit
whois/worker.py: fix worker to use DeviceRepository
Signed-off-by: Eduard Kaverinskyi <[email protected]>
  • Loading branch information
EduKav1813 committed Jan 18, 2025
commit 6d95388b254c3838634e5c316416401892c5450d
18 changes: 9 additions & 9 deletions whois/worker.py
Original file line number Diff line number Diff line change
@@ -2,26 +2,26 @@
import time
from datetime import datetime, timezone

from whois.data.db.database import Device, db
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

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
)

for lease in leases:
with db.atomic():
last_seen_date = datetime.now(timezone.utc) - lease.last_seen
Device.update_or_create(
mac_address=lease.mac_address,
last_seen=last_seen_date,
hostname=lease.host_name,
)
device = Device(lease.mac_address, lease.host_name, lease.last_seen)
if device_repository.get_by_mac_address(lease.mac_address):
device_repository.update(device)
else:
device_repository.insert(device)

return len(leases)