Merge pull request #79 from caiosweet/codex/fix-dpc-alert-component-issue

Fix DPC setup typing and icon mapping

Author: caiosweet

custom_components/dpc/__init__.py

@@ -19,13 +19,10 @@
     CONF_SCAN_INTERVAL,
 )
 from homeassistant.core import HomeAssistant
-import homeassistant.helpers.config_validation as cv
-from homeassistant.helpers.typing import ConfigType
 from homeassistant.exceptions import ConfigEntryNotReady
-from homeassistant.helpers import config_validation as cv
 from homeassistant.helpers import event
-import homeassistant.helpers.config_validation as cv
 from homeassistant.helpers.aiohttp_client import async_get_clientsession
+from homeassistant.helpers.typing import ConfigType
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
 
 from .api import DpcApiClient, DpcApiException
@@ -42,14 +39,13 @@
 CONFIG_SCHEMA = cv.config_entry_only_config_schema(DOMAIN)
 
 
-
-
-async def async_setup(hass: HomeAssistant, config: ConfigType):
+async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
     """Set up this integration using YAML is not supported."""
+    del config
     return True
 
 
-async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry):
+async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
     """Set up this integration using UI."""
     if hass.data.get(DOMAIN) is None:
         hass.data.setdefault(DOMAIN, {})
@@ -90,7 +86,9 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry):
 class DpcDataUpdateCoordinator(DataUpdateCoordinator):
     """Class to manage fetching data from the API."""
 
-    def __init__(self, hass: HomeAssistant, client: DpcApiClient, update_interval) -> None:
+    def __init__(
+        self, hass: HomeAssistant, client: DpcApiClient, update_interval: timedelta
+    ) -> None:
         """Initialize."""
         self.api = client
         self.platforms = []
@@ -118,7 +116,7 @@ async def _async_request_refresh_later(self, _now):
         await self.async_request_refresh()
 
 
-async def async_update_options(hass, config_entry):
+async def async_update_options(hass: HomeAssistant, config_entry: ConfigEntry) -> None:
     """Update options."""
     await hass.config_entries.async_reload(config_entry.entry_id)
 
@@ -141,7 +139,7 @@ async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
     return unloaded
 
 
-async def async_migrate_entry(hass, entry):
+async def async_migrate_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
     LOGGER.info("Migrating DPC entry from Version %s", entry.version)
     if entry.version == 1:
         entry.options = dict(entry.options)

custom_components/dpc/api.py

@@ -102,7 +102,7 @@
     21: "mdi:snowflake-variant",
     30: "mdi:weather-fog",
     31: "mdi:weather-hazy",
-    40: "mdi:wave",
+    40: "mdi:waves",
     41: "mdi:waves",
     42: "mdi:hydro-power",
     50: "mdi:arrow-up-thick",