add oxiApi

oxi/interface/__init__.py

@@ -1,10 +1,6 @@
-from .vrp import Vrp
-from .qtech import Qtech
-from .mikrotik import Mikrotik
-from .bdcom import BDcom
+from oxi.interface.registry import register_parser, device_registry
+
 __all__ = [
-    'Vrp',
-    'Qtech',
-    'Mikrotik',
-    'BDcom'
+    'register_parser',
+    'device_registry'
 ]

oxi/interface/base.py

@@ -1,6 +1,6 @@
 import re
-from abc import ABC
 from dataclasses import dataclass
+from typing import Protocol
 
 
 @dataclass
@@ -25,7 +25,7 @@ class ParsedDeviceData:
     vlans: list[Vlan]
 
 
-class BaseDevice(ABC):
+class BaseDevice(Protocol):
     anchor_pattern: str = '!'
     hostname_pattern: str = 'hostname'
 
@@ -81,7 +81,7 @@ class BaseDevice(ABC):
             ip_address_match = re.search(r'\s?ip address\s(.+)$', interface_block, re.MULTILINE)
             if not description_match and not ip_address_match:
                 continue
-            ip_address = ip_address_match.group(1) if ip_address_match else None
+            ip_address = ip_address_match.group(1).replace(' ', '/') if ip_address_match else None
             description = description_match.group(1) if description_match else None
             interfaces.append(L3Interface(
                 interface=interface_name,

oxi/interface/models/__init__.py

@@ -0,0 +1,7 @@
+import importlib
+import pkgutil
+
+package = __package__
+
+for loader, module_name, is_pkg in pkgutil.iter_modules(__path__):
+    importlib.import_module(f"{package}.{module_name}")

oxi/interface/models/bdcom.py

@@ -0,0 +1,7 @@
+from oxi.interface import register_parser
+from oxi.interface.models.qtech import Qtech
+
+
+@register_parser("BDCOM")
+class BDcom(Qtech):
+    pass

oxi/interface/models/mikrotik.py

@@ -0,0 +1,7 @@
+from oxi.interface import register_parser
+from oxi.interface.base import BaseDevice
+
+
+@register_parser("Mikrotik")
+class Mikrotik(BaseDevice):
+    ...

oxi/interface/models/qtech.py

@@ -0,0 +1,16 @@
+import re
+
+from oxi.interface import register_parser
+from oxi.interface.base import BaseDevice
+
+
+@register_parser("QTECH")
+class Qtech(BaseDevice):
+
+    def __init__(self, config):
+        self.config: str = self._fix_config(config)
+
+    def _fix_config(self, config):
+        pattern = r"Pending configurations.*"
+        cleaned_text = re.sub(pattern, "", config, flags=re.DOTALL)
+        return cleaned_text

oxi/interface/models/vrp.py

@@ -0,0 +1,30 @@
+import re
+
+from oxi.interface import register_parser
+from oxi.interface.base import BaseDevice, Vlan
+
+
+@register_parser("VRP")
+class Vrp(BaseDevice):
+    anchor_pattern: str = '#'
+    hostname_pattern = 'sysname'
+    unamed_vlan_splitter = ' '
+    unamed_vlan_counter = 'to'
+
+    def _parse_unamed_vlans(self) -> list[Vlan]:
+        vlans = []
+        pattern = self.unamed_vlans_parse_pattern
+        for match in re.finditer(pattern, self.config, re.MULTILINE):
+            tokens = match.group(1).split(self.unamed_vlan_splitter)
+            i = 0
+            while i < len(tokens):
+                if i + 2 < len(tokens) and tokens[i + 1].lower() == 'to':
+                    start = int(tokens[i])
+                    end = int(tokens[i + 2])
+                    for vlan_id in range(start, end + 1):
+                        vlans.append(Vlan(vlan=str(vlan_id), name=None, description=None))
+                    i += 3  # пропустить X, 'to', Y
+                else:
+                    vlans.append(Vlan(vlan=str(tokens[i]), name=None, description=None))
+                    i += 1
+        return vlans

oxi/interface/registry.py

@@ -0,0 +1,12 @@
+from typing import Callable, Type
+
+from oxi.interface.base import BaseDevice
+
+device_registry = {}
+
+
+def register_parser(name: str) -> Callable[[Type[BaseDevice]], Type[BaseDevice]]:
+    def wrapper(cls):
+        device_registry[name.lower()] = cls
+        return cls
+    return wrapper