Merge pull request #103 from Snuffy2/Fix-timestamp

Make datetime tz aware
enkama · Nov 21, 2023 · b7ae428 · b7ae428
2 parents 30766f7 + 141a9af
commit b7ae428
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/custom_components/variable/helpers.py b/custom_components/variable/helpers.py
@@ -3,6 +3,8 @@
 import datetime
 import logging
 
+import homeassistant.util.dt as dt_util
+
 _LOGGER = logging.getLogger(__name__)
 
 
@@ -61,6 +63,11 @@ def value_to_type(init_val, dest_type):  # noqa: C901
                 _LOGGER.debug(
                     f"[value_to_type] return value: {value_datetime}, type: {type(value_datetime)}"
                 )
+                if (
+                    value_datetime.tzinfo is None
+                    or value_datetime.tzinfo.utcoffset(value_datetime) is None
+                ):
+                    return value_datetime.replace(tzinfo=dt_util.UTC)
                 return value_datetime
 
         elif dest_type == "number":
@@ -113,6 +120,11 @@ def value_to_type(init_val, dest_type):  # noqa: C901
                 _LOGGER.debug(
                     f"[value_to_type] return value: {value_datetime}, type: {type(value_datetime)}"
                 )
+                if (
+                    value_datetime.tzinfo is None
+                    or value_datetime.tzinfo.utcoffset(value_datetime) is None
+                ):
+                    return value_datetime.replace(tzinfo=dt_util.UTC)
                 return value_datetime
         elif dest_type == "number":
             _LOGGER.debug(