135 lines
3.8 KiB
Python
135 lines
3.8 KiB
Python
from dataclasses import dataclass, field
|
|
from typing import List, Optional, Dict
|
|
|
|
# ===========================
|
|
# Network / Subnet
|
|
# ===========================
|
|
|
|
@dataclass
|
|
class Network:
|
|
name: str
|
|
vlan_id : Optional[int] = None
|
|
ip: Optional[str] = None
|
|
subnet_mask: Optional[str] = None
|
|
|
|
def __init__(self, fields):
|
|
self.name = fields.get("name")
|
|
self.vlan_id = fields.get("vlan", fields.get("vlan_id"))
|
|
self.ip = fields.get("ip")
|
|
self.subnet_mask = fields.get("mask", fields.get("subnet_mask"))
|
|
self.__post_init__()
|
|
|
|
def __post_init__(self):
|
|
if not isinstance(self.name, str) or not self.name:
|
|
raise ValueError("Network 'name' must be a non-empty string")
|
|
|
|
# ===========================
|
|
# Interface Classes
|
|
# ===========================
|
|
|
|
|
|
@dataclass
|
|
class Interface:
|
|
class IP:
|
|
def __init__(self, ip_str: str, default_gateway: Optional[str] = None):
|
|
self.ip_str = ip_str
|
|
self.default_gateway = default_gateway
|
|
|
|
def __str__(self):
|
|
return self.ip_str
|
|
|
|
name: str
|
|
mode: Optional[str] = None
|
|
network: Optional[Network] = None
|
|
ips: Optional[List[IP]] = None
|
|
mac_address: Optional[str] = None
|
|
speed: Optional[str] = None
|
|
|
|
def __init__(self, fields):
|
|
self.name = fields.get("name")
|
|
ips = fields.get("ips", [])
|
|
if isinstance(ips, str):
|
|
ips = [self.IP(ip_str=ip.strip()) for ip in ips.split(",") if ip.strip()]
|
|
self.ips = ips
|
|
|
|
# возможно передаём просто field
|
|
self.network = Network(fields.get("network", {})) if fields.get("network") else None
|
|
|
|
self.mac_address = fields.get("mac")
|
|
self.speed = fields.get("speed")
|
|
self.__post_init__()
|
|
|
|
def __post_init__(self):
|
|
if not isinstance(self.name, str) or not self.name:
|
|
raise ValueError("Interface 'name' must be a non-empty string")
|
|
|
|
|
|
# ===========================
|
|
# Device Base Class (Abstract)
|
|
# ===========================
|
|
|
|
|
|
@dataclass
|
|
class Device:
|
|
name: str
|
|
interfaces: List[Interface] = field(default_factory=list)
|
|
|
|
def __init__(self, fields):
|
|
self.name = fields.get("name")
|
|
# тут точно понять, какие поля передаются
|
|
self.interfaces = [Interface(iface) for iface in fields.get("interfaces", [])]
|
|
self.__post_init__()
|
|
|
|
def __post_init__(self):
|
|
if not isinstance(self.name, str) or not self.name:
|
|
raise ValueError("Device 'name' must be a non-empty string")
|
|
|
|
|
|
# ===========================
|
|
# Host / Router / Switch
|
|
# ===========================
|
|
|
|
|
|
@dataclass
|
|
class Host(Device):
|
|
# operating_system: Optional[str] = None
|
|
|
|
def __init__(self, fields):
|
|
super().__init__(fields)
|
|
# self.operating_system = fields.get("operatingSystem")
|
|
self.__post_init__()
|
|
|
|
def __post_init__(self):
|
|
super().__post_init__()
|
|
|
|
|
|
@dataclass
|
|
class Router(Device):
|
|
# routing_table: Dict[str, str] = field(default_factory=dict)
|
|
# routing_protocols: List[str] = field(default_factory=list)
|
|
|
|
def __init__(self, fields):
|
|
super().__init__(fields)
|
|
# self.routing_table = fields.get("routingTable", {})
|
|
# self.routing_protocols = fields.get("routingProtocols", [])
|
|
self.__post_init__()
|
|
|
|
def __post_init__(self):
|
|
super().__post_init__()
|
|
|
|
|
|
@dataclass
|
|
class Switch(Device):
|
|
# mac_table: Dict[str, str] = field(default_factory=dict)
|
|
# vlan_database: Dict[int, str] = field(default_factory=dict)
|
|
|
|
def __init__(self, fields):
|
|
super().__init__(fields)
|
|
# self.mac_table = fields.get("macTable", {})
|
|
# self.vlan_database = fields.get("vlanDatabase", {})
|
|
self.__post_init__()
|
|
|
|
def __post_init__(self):
|
|
super().__post_init__()
|
|
|