Merge pull request #33 from faanskit/dev

Added peak battery effect

pycheckwatt/__init__.py

@@ -71,6 +71,7 @@ def __init__(self, username, password, application="pyCheckwatt") -> None:
         self.display_name = None
         self.reseller_id = None
         self.energy_provider_id = None
+        self.month_peak_effect = None
 
     async def __aenter__(self):
         """Asynchronous enter."""
@@ -692,6 +693,39 @@ async def get_spot_price(self):
         except (ClientResponseError, ClientError) as error:
             return await self.handle_client_error(endpoint, headers, error)
 
+    async def get_battery_month_peak_effect(self):
+        """Fetch Price Zone from CheckWatt."""
+        month = datetime.now().strftime("%Y-%m")
+
+        try:
+            endpoint = f"/ems/PeakBoughtMonth?month={month}"
+            # Define headers with the JwtToken
+            headers = {
+                **self._get_headers(),
+                "authorization": f"Bearer {self.jwt_token}",
+            }
+
+            # First fetch the revenue
+            async with self.session.get(
+                self.base_url + endpoint, headers=headers
+            ) as response:
+                response.raise_for_status()
+                if response.status == 200:
+                    peak_data = await response.json()
+                    if "HourPeak" in peak_data:
+                        self.month_peak_effect = peak_data["HourPeak"]
+                        return True
+
+                _LOGGER.error(
+                    "Obtaining data from URL %s failed with status code %d",
+                    self.base_url + endpoint,
+                    response.status,
+                )
+                return False
+
+        except (ClientResponseError, ClientError) as error:
+            return await self.handle_client_error(endpoint, headers, error)
+
     async def get_energy_trading_company(self, input_id):
         """Translate Energy Company Id to Energy Company Name."""
         try:

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pycheckwatt"
-version = "0.2.2"
+version = "0.2.3"
 description = "Read data from CheckWatts EnergyInBalance WEB API"
 authors = ["Marcus Karlsson <[email protected]>", "Anders Yderborg <[email protected]>", "Daniel Nilsson <[email protected]>"]
 license = "MIT License"

setup.py

@@ -4,7 +4,7 @@
 
 MIN_PY_VERSION = "3.10"
 PACKAGES = find_packages()
-VERSION = "0.2.2"
+VERSION = "0.2.3"
 
 setup(
     name="pycheckwatt",