First commit

models/base_switch.py

@@ -0,0 +1,125 @@
+# models/base_switch.py
+from typing import List, Optional
+
+class PowerSupply:
+    def __init__(self, id: int, state: str, mode: Optional[str] = None):
+        self.id = id
+        self.state = state
+        self.mode = mode
+        self.summary = f"Alimentation {id} en état {state}"
+
+    def to_dict(self):
+        return {
+            "object": "power_supply",
+            "id": self.id,
+            "state": self.state,
+            "mode": self.mode,
+            "summary": self.summary,
+        }
+
+class Fan:
+    def __init__(self, id: int, state: str):
+        self.id = id
+        self.state = state
+        self.summary = f"Ventilateur {id} en état {state}"
+
+    def to_dict(self):
+        return {
+            "object": "fan",
+            "id": self.id,
+            "state": self.state,
+            "summary": self.summary,
+        }
+
+class MACEntry:
+    def __init__(self, mac: str, vlan: int, port: str, type_: str):
+        self.mac = mac
+        self.vlan = vlan
+        self.port = port
+        self.type = type_
+
+    def to_dict(self):
+        return {
+            "object": "mac_entry",
+            "mac": self.mac,
+            "vlan": self.vlan,
+            "port": self.port,
+            "type": self.type,
+        }
+
+class Port:
+    def __init__(self, name: str):
+        self.object = "port"
+        self.name = name
+        self.status = None
+        self.speed = None
+        self.vlan = None
+        self.link_type = None
+        self.trunk_vlans = []
+        self.bpdu_protection = False
+        self.portfast = False
+        self.loopback_detection = None
+        self.mac_addresses = []
+        self.summary = ""
+
+    def to_dict(self):
+        return {
+            "object": "port",
+            "name": self.name,
+            "status": self.status,
+            "speed": self.speed,
+            "vlan": self.vlan,
+            "link_type": self.link_type,
+            "trunk_vlans": self.trunk_vlans if self.link_type == "trunk" else None,
+            "bpdu_protection": self.bpdu_protection,
+            "portfast": self.portfast,
+            "loopback_detection": self.loopback_detection,
+            "mac_addresses": [m.to_dict() for m in self.mac_addresses],
+            "summary": self.summary,
+        }
+
+class VLAN:
+    def __init__(self, vlan_id: int, name: Optional[str] = None, description: Optional[str] = None):
+        self.vlan_id = vlan_id
+        self.name = name
+        self.description = description
+        self.summary = f"VLAN {vlan_id} : {name or 'non nommé'}"
+
+    def to_dict(self):
+        return {
+            "object": "vlan",
+            "vlan_id": self.vlan_id,
+            "name": self.name,
+            "description": self.description,
+            "summary": self.summary,
+        }
+
+class Switch:
+    def __init__(self):
+        self.hostname = None
+        self.model = None
+        self.firmware_version = None
+        self.boot_image = None
+        self.uptime = None
+        self.ports: List[Port] = []
+        self.vlans: List[VLAN] = []
+        self.fans: List[Fan] = []
+        self.power_supplies: List[PowerSupply] = []
+        self.mac_table: List[MACEntry] = []
+        self.summary = ""
+
+    def to_dict(self):
+        return {
+            "object": "switch",
+            "hostname": self.hostname,
+            "model": self.model,
+            "firmware_version": self.firmware_version,
+            "boot_image": self.boot_image,
+            "uptime": self.uptime,
+            "summary": self.summary,
+            "fans": [f.to_dict() for f in self.fans],
+            "power_supplies": [p.to_dict() for p in self.power_supplies],
+            "ports": [p.to_dict() for p in self.ports],
+            "vlans": [v.to_dict() for v in self.vlans],
+            "mac_table": [m.to_dict() for m in self.mac_table],
+        }