add oxiApi
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user