fix(sensor): check for AttributeError timezone

Closes #501
alandtse · Jan 15, 2020 · 84b4c61 · 84b4c61
1 parent 82f8af2
commit 84b4c61
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/custom_components/alexa_media/sensor.py b/custom_components/alexa_media/sensor.py
@@ -178,7 +178,11 @@ def __init__(
     def _fix_alarm_date_time(self, value):
         import pytz
 
-        if self._sensor_property != "date_time" or not value:
+        if (
+            self._sensor_property != "date_time"
+            or not value
+            or isinstance(value[1][self._sensor_property], datetime.datetime)
+        ):
             return value
         naive_time = dt.parse_datetime(value[1][self._sensor_property])
         timezone = pytz.timezone(self._client._timezone)