Tidy up and move api_client

custom_components/ohme/__init__.py

@@ -1,6 +1,6 @@
 from homeassistant import core
 from .const import *
-from .client import OhmeApiClient
+from .api_client import OhmeApiClient
 from .coordinator import OhmeUpdateCoordinator, OhmeStatisticsUpdateCoordinator
 
 

custom_components/ohme/client/__init__.py → custom_components/ohme/api_client.py

@@ -1,10 +1,8 @@
 import aiohttp
-import asyncio
 import logging
-import json
 from datetime import datetime, timedelta
 from homeassistant.helpers.entity import DeviceInfo
-from ..const import DOMAIN
+from .const import DOMAIN
 
 _LOGGER = logging.getLogger(__name__)
 

custom_components/ohme/config_flow.py

@@ -1,7 +1,7 @@
 import voluptuous as vol
 from homeassistant.config_entries import (ConfigFlow, OptionsFlow)
 from .const import DOMAIN
-from .client import OhmeApiClient
+from .api_client import OhmeApiClient
 
 
 class OhmeConfigFlow(ConfigFlow, domain=DOMAIN):

custom_components/ohme/sensor.py

@@ -14,6 +14,7 @@
 from .coordinator import OhmeUpdateCoordinator, OhmeStatisticsUpdateCoordinator
 from .utils import charge_graph_next_slot
 
+
 async def async_setup_entry(
     hass: core.HomeAssistant,
     config_entry: config_entries.ConfigEntry,
@@ -114,6 +115,7 @@ def native_value(self):
 
         return None
 
+
 class NextSlotSensor(CoordinatorEntity[OhmeStatisticsUpdateCoordinator], SensorEntity):
     """Sensor for next smart charge slot."""
     _attr_name = "Next Smart Charge Slot"
@@ -158,7 +160,8 @@ def _handle_coordinator_update(self) -> None:
         if self.coordinator.data is None:
             self._state = None
         else:
-            self._state = charge_graph_next_slot(self.coordinator.data['startTime'], self.coordinator.data['chargeGraph']['points'])
+            self._state = charge_graph_next_slot(
+                self.coordinator.data['startTime'], self.coordinator.data['chargeGraph']['points'])
 
         self._last_updated = utcnow()
 

custom_components/ohme/utils.py

@@ -2,18 +2,19 @@
 from datetime import datetime
 import pytz
 
+
 def charge_graph_next_slot(charge_start, points):
     """Get the next charge slot from a list of graph points."""
     # Get start and current timestamp in seconds
     charge_start = round(charge_start / 1000)
     now = int(time())
-    
+
     # Replace relative timestamp (seconds) with real timestamp
     data = [{"t": x["x"] + charge_start, "y": x["y"]} for x in points]
-    
+
     # Filter to points from now onwards
     data = [x for x in data if x["t"] > now]
-    
+
     # Give up if we have less than 3 points
     if len(data) < 3:
         return None
@@ -33,4 +34,3 @@ def charge_graph_next_slot(charge_start, points):
 
     # This needs to be presented with tzinfo or Home Assistant will reject it
     return None if next_ts is None else datetime.utcfromtimestamp(next_ts).replace(tzinfo=pytz.utc)
-