add oxiApi

This commit is contained in:
IluaAir
2025-06-25 10:32:51 +03:00
parent 009f007a38
commit 1eb0ff1eca
11 changed files with 129 additions and 33 deletions

View File

@@ -1,9 +1,13 @@
import logging
from functools import cached_property
import requests
from typing import Optional
from oxi.interface.base import BaseDevice
from settings import settings
import oxi.interface.models # noqa
from oxi.interface import device_registry
log = logging.getLogger()
@@ -21,10 +25,7 @@ class OxidizedAPI:
self._session.verify = verify
if username and password:
self._session.auth = (username, password)
@property
def node(self):
return Node(self._session, self.base_url)
self.node = Node(self._session, self.base_url)
def __enter__(self):
return self
@@ -48,15 +49,15 @@ class Node:
self._base_url = base_url
self._data = None
def show(self, name: str) -> 'NodeView':
def __call__(self, name: str) -> 'NodeView':
url = f"{self._base_url}/node/show/{name}"
if not url.endswith('.json'):
url += '.json'
self._data = self._session.get(url)
response = self._session.get(url)
return NodeView(
session=self._session,
base_url=self._base_url,
data=self._data.json()
data=response.json()
)
@@ -84,26 +85,43 @@ class NodeView:
@property
def config(self):
return NodeConfig(self._session, self.full_name, self._base_url)
return NodeConfig(self._session, self.full_name, self.model, self._base_url)
class NodeConfig:
def __init__(self, session: requests.Session, full_name: str, base_url: str):
def __init__(self, session: requests.Session, full_name: str, model: str, base_url: str):
self._session = session
self._full_name = full_name
self._model = model
self._url = f"{base_url}/node/fetch/{full_name}"
self._device: type[BaseDevice] = device_registry.get(self._model.lower())
if self._device is None:
raise ValueError(f"Device model '{self._model}' not found in registry")
self._data = None
def text(self) -> str:
return self._session.get(self._url).text
@cached_property
def _response(self):
log.debug(f"Fetching config from {self._url}")
response = self._session.get(self._url)
response.raise_for_status()
return response
def json(self) -> dict:
return self._session.get(self._url).json()
@property
def text(self):
return self._response.text
def __str__(self) -> str:
return self.text()
@property
def json(self):
return self._response.json()
def __str__(self):
return self.text
oxi = OxidizedAPI(username=settings.oxi_username, password=settings.oxi_password, verify=False)
print(oxi.node.show('Novok_HOME').config)
mikrotik = oxi.node.show('Novok_HOME').model
print(mikrotik.model)
def vlans(self):
return self._device(self.text).parse_config().vlans
def l3interfaces(self):
return self._device(self.text).parse_config().l3interfaces
def vlaninterfaces(self):
return self._device(self.text).parse_config().vlaninterfaces