Add Core response and Core state

README.md

@@ -36,6 +36,10 @@ if __name__ == "__main__":
 
 True if the server is healthy
 
+- `get_core_state() -> dict`
+
+Retrieve the core state
+
 - `get_devices() -> list[DeviceResponse]`
 
 Retrieve a list of devices
@@ -51,3 +55,27 @@ Retrieve the device consumption by name
 - `get_surplus() -> float:`
 
 Retrieve the surplus value
+
+- `set_surplus_margin(margin: float) -> float:`
+
+Set the surplus margin
+
+- `set_grid_margin(margin: float) -> float:`
+
+Set the grid margin
+
+- `set_idle_power(idle_power: float) -> float:`
+
+Set the idle power
+
+- `set_device_max_consumption(device_name: str, max_consumption: float) -> float:`
+
+Set the max consumption for a device
+
+- `set_device_expected_consumption(device_name: str, expected_consumption: float) -> float:`
+
+Set the expected consumption for a device
+
+- `set_device_cooldown(device_name: str, cooldown: int) -> int:`
+
+Set the cooldown for a device

pyosmanager/pyosmanager.py

@@ -6,7 +6,7 @@
 import aiohttp
 import backoff
 
-from pyosmanager.responses import DeviceResponse
+from pyosmanager.responses import CoreResponse, DeviceResponse
 
 logger = logging.getLogger(__name__)
 
@@ -72,7 +72,7 @@ async def __post_data(self, endpoint, data):
         """
         try:
             async with self.session.post(
-                f"{self.base_url}/{endpoint}", json=data
+                f"{self.base_url}/api/{endpoint}", json=data
             ) as response:
                 response.raise_for_status()
                 data = await response.json()
@@ -94,6 +94,15 @@ async def is_healthy(self) -> bool:
         except APIError:
             return False
 
+    async def get_core_state(self) -> dict:
+        """
+        Get the core state of the system.
+
+        :return: Core state of the system
+        """
+        core = await self.__get_data("core")
+        return CoreResponse(**core)
+
     async def get_devices(self) -> list[DeviceResponse]:
         """
         Get a list of all devices.

pyosmanager/responses.py

@@ -19,3 +19,16 @@ class DeviceResponse:
     powered: bool
     cooldown: int | None
     enabled: bool
+
+
+@dataclass
+class CoreResponse:
+    """
+    This class is used to represent the response of the API
+    when querying for the core.
+    """
+
+    surplus: float
+    surplus_margin: float
+    grid_margin: float
+    idle_power: float

setup.py

@@ -6,7 +6,7 @@
 setup(
     name="pyosmanager",
     packages=["pyosmanager"],
-    version="0.2.1",
+    version="0.2.2",
     license="MIT",
     description="Python client for Open Surplus Manager",
     long_description=long_description,