last_stats_time is now float with latest HA update

dreautall · Mar 29, 2023 · e7dbf27 · e7dbf27
1 parent 0b8c99f
commit e7dbf27
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/custom_components/stromnetzgraz/sensor.py b/custom_components/stromnetzgraz/sensor.py
@@ -2,7 +2,7 @@
 from __future__ import annotations
 
 import asyncio
-from datetime import timedelta
+from datetime import timedelta, datetime
 import logging
 import pandas
 import numpy
@@ -229,6 +229,12 @@ async def _insert_statistics(self):
             if stat:
                 _sum = stat[statistic_id][0]["sum"]
                 last_stats_time = stat[statistic_id][0]["start"]
+                if isinstance(last_stats_time, float):
+                    last_stats_time = datetime.fromtimestamp(
+                        stat[statistic_id][0]["start"]
+                    )
+                else:
+                    last_stats_time = last_stats_time.replace(tzinfo=None)
             else:
                 _sum = 0.0
                 last_stats_time = None
@@ -254,7 +260,10 @@ async def _insert_statistics(self):
 
                 # make sure minutes & (micro)seconds is 0, or HA will throw an error
                 start = start.to_pydatetime().replace(minute=0, second=0, microsecond=0)
-                if last_stats_time is not None and start <= last_stats_time:
+                if (
+                    last_stats_time is not None
+                    and start.replace(tzinfo=None) <= last_stats_time
+                ):
                     continue
 
                 _sum += consump