Fix tzinfo

Author: dan-r

custom_components/ohme/binary_sensor.py

@@ -5,7 +5,6 @@
     BinarySensorDeviceClass,
     BinarySensorEntity
 )
-from datetime import datetime
 from homeassistant.helpers.update_coordinator import CoordinatorEntity
 from homeassistant.core import HomeAssistant, callback
 from homeassistant.helpers.entity import generate_entity_id
@@ -294,7 +293,7 @@ def unique_id(self) -> str:
     @property
     def extra_state_attributes(self):
         """Attributes of the sensor."""
-        now = datetime.now()
+        now = utcnow()
         slots = self._hass.data[DOMAIN][DATA_SLOTS] if DATA_SLOTS in self._hass.data[DOMAIN] else []
 
         return {

custom_components/ohme/utils.py

@@ -140,8 +140,8 @@ def charge_graph_slot_list(charge_start, points, skip_format=False):
         # Append a dict to the slots list with the start and end time
         slots.append(
             {
-                "start": datetime.fromtimestamp(result[0]),
-                "end": datetime.fromtimestamp(result[1]),
+                "start": datetime.utcfromtimestamp(result[0]).replace(tzinfo=pytz.utc),
+                "end": datetime.utcfromtimestamp(result[1]).replace(tzinfo=pytz.utc),
                 "charge_in_kwh": -(result[3] / 1000),
                 "source": "smart-charge",
                 "location": None