Simplify services

Author: andrew-codechimp

custom_components/battery_notes/__init__.py

@@ -51,7 +51,7 @@
 )
 from .discovery import DiscoveryManager
 from .library_updater import LibraryUpdater
-from .services import setup_services
+from .services import async_setup_services
 from .store import (
     async_get_registry,
 )
@@ -138,7 +138,7 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
         _LOGGER.debug("Auto discovery disabled")
 
     # Register custom services
-    setup_services(hass)
+    async_setup_services(hass)
 
     return True
 

custom_components/battery_notes/common.py

@@ -1,6 +1,6 @@
 """Common functions for battery_notes."""
 
-import datetime
+from datetime import datetime
 
 from homeassistant.helpers.device_registry import DeviceEntry
 from homeassistant.util import dt as dt_util