36 lines
1.0 KiB
Python
36 lines
1.0 KiB
Python
from typing import Optional
|
|
from requests import Session
|
|
from .node import Node
|
|
|
|
|
|
class OxiAPI:
|
|
def __init__(
|
|
self,
|
|
url: str,
|
|
session: Optional[Session] = None,
|
|
username: Optional[str] = None,
|
|
password: Optional[str] = None,
|
|
verify: bool = True,
|
|
):
|
|
self.base_url = url.rstrip("/")
|
|
self._session = session or Session()
|
|
self._session.verify = verify
|
|
if username and password:
|
|
self._session.auth = (username, password)
|
|
self.node = Node(self._session, self.base_url)
|
|
|
|
def __enter__(self):
|
|
return self
|
|
|
|
def __exit__(self, *args):
|
|
self._session.close()
|
|
|
|
def get(self, endpoint: str, **kwargs) -> dict:
|
|
url = f"{self.base_url}/{endpoint.lstrip('/')}"
|
|
if not url.endswith(".json"):
|
|
url += ".json"
|
|
result = self._session.get(url, **kwargs)
|
|
if result.status_code == 500:
|
|
raise ValueError(f"page {url} not found")
|
|
return result.json()
|