aljaz/python-teos
1
0
Fork 0
mirror of https://github.com/aljazceru/python-teos.git synced 2025-12-17 22:24:23 +01:00
Code Issues Packages Projects Releases Wiki Activity

pluggin - adds pending appointments to towers

Browse Source
This commit is contained in:
Sergi Delgado Segura
2020-04-27 18:49:47 +02:00
parent 75ca281cc4
commit 780cb5c202
1 changed files with 15 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
watchtower-plugin/tower_info.py
View File

@@ -1,5 +1,5 @@
class TowerInfo: class TowerInfo:
def __init__(self, netaddr, available_slots, status="active", appointments=None): def __init__(self, netaddr, available_slots, status="active", appointments=None, pending_appointments=None):
self.netaddr = netaddr self.netaddr = netaddr
self.available_slots = available_slots self.available_slots = available_slots
self.status = status self.status = status
@@ -9,20 +9,31 @@ class TowerInfo:
else: else:
self.appointments = appointments self.appointments = appointments
if not pending_appointments:
self.pending_appointments = []
else:
self.pending_appointments = pending_appointments
@classmethod @classmethod
def from_dict(cls, tower_data): def from_dict(cls, tower_data):
netaddr = tower_data.get("netaddr") netaddr = tower_data.get("netaddr")
available_slots = tower_data.get("available_slots") available_slots = tower_data.get("available_slots")
status = tower_data.get("status") status = tower_data.get("status")
appointments = tower_data.get("appointments") appointments = tower_data.get("appointments")
pending_appointments = tower_data.get("pending_appointments")
if any(v is None for v in [netaddr, available_slots, status, appointments]): if any(v is None for v in [netaddr, available_slots, status, appointments, pending_appointments]):
raise ValueError("Wrong appointment data, some fields are missing") raise ValueError("Wrong appointment data, some fields are missing")
return cls(netaddr, available_slots, status, appointments) return cls(netaddr, available_slots, status, appointments, pending_appointments)
def to_dict(self): def to_dict(self):
return self.__dict__ return self.__dict__
def get_summary(self): def get_summary(self):
return {"netaddr": self.netaddr, "status": self.status, "available_slots": self.available_slots} return {
"netaddr": self.netaddr,
"status": self.status,
"available_slots": self.available_slots,
"pending_appointments": self.pending_appointments,
}