Improve energy history endpoints again

Teslemetry · Sep 16, 2024 · d4de5c4 · d4de5c4
1 parent b2554ca
commit d4de5c4
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 21 deletions.
diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name="tesla_fleet_api",
-    version="0.7.7",
+    version="0.7.8",
     author="Brett Adams",
     author_email="hello@teslemetry.com",
     description="Tesla Fleet API library for Python",

diff --git a/tesla_fleet_api/energy.py b/tesla_fleet_api/energy.py
@@ -32,10 +32,10 @@ async def backup(
     async def backup_history(
         self,
         energy_site_id: int,
-        start_date: str,
-        end_date: str,
-        period: TeslaEnergyPeriod | str,
-        time_zone: str,
+        period: TeslaEnergyPeriod | str | None,
+        start_date: str | None = None,
+        end_date: str | None = None,
+        time_zone: str | None = None,
     ) -> dict[str, Any]:
         """Returns the backup (off-grid) event history of the site in duration of seconds."""
         return await self._request(
@@ -55,7 +55,7 @@ async def charge_history(
         energy_site_id: int,
         start_date: str,
         end_date: str,
-        time_zone: str,
+        time_zone: str | None = None,
     ) -> dict[str, Any]:
         """Returns the charging history of a wall connector."""
         return await self._request(
@@ -72,10 +72,10 @@ async def charge_history(
     async def energy_history(
         self,
         energy_site_id: int,
-        start_date: str,
-        end_date: str,
-        period: TeslaEnergyPeriod | str,
-        time_zone: str,
+        period: TeslaEnergyPeriod | str | None,
+        start_date: str | None = None,
+        end_date: str | None = None,
+        time_zone: str | None = None,
     ) -> dict[str, Any]:
         """Returns the energy measurements of the site, aggregated to the requested period."""
         return await self._request(

diff --git a/tesla_fleet_api/energyspecific.py b/tesla_fleet_api/energyspecific.py
@@ -29,25 +29,25 @@ async def backup(self, backup_reserve_percent: int) -> dict[str, Any]:
 
     async def backup_history(
         self,
-        start_date: str,
-        end_date: str,
-        period: TeslaEnergyPeriod | str,
-        time_zone: str,
+        period: TeslaEnergyPeriod | str | None,
+        start_date: str | None = None,
+        end_date: str | None = None,
+        time_zone: str | None = None,
     ) -> dict[str, Any]:
         """Returns the backup (off-grid) event history of the site in duration of seconds."""
         return await self._parent.backup_history(
             self.energy_site_id,
+            period,
             start_date,
             end_date,
-            period,
             time_zone,
         )
 
     async def charge_history(
         self,
         start_date: str,
         end_date: str,
-        time_zone: str,
+        time_zone: str | None = None,
     ) -> dict[str, Any]:
         """Returns the charging history of a wall connector."""
         return await self._parent.charge_history(
@@ -59,17 +59,17 @@ async def charge_history(
 
     async def energy_history(
         self,
-        start_date: str,
-        end_date: str,
-        period: TeslaEnergyPeriod | str,
-        time_zone: str,
+        period: TeslaEnergyPeriod | str | None,
+        start_date: str | None = None,
+        end_date: str | None = None,
+        time_zone: str | None = None,
     ) -> dict[str, Any]:
         """Returns the energy measurements of the site, aggregated to the requested period."""
         return await self._parent.energy_history(
             self.energy_site_id,
+            period,
             start_date,
             end_date,
-            period,
             time_zone,
         )