diff --git a/homeassistant/components/cert_expiry/__init__.py b/homeassistant/components/cert_expiry/__init__.py index 717a55b2027..2387c2a73c3 100644 --- a/homeassistant/components/cert_expiry/__init__.py +++ b/homeassistant/components/cert_expiry/__init__.py @@ -7,21 +7,21 @@ from homeassistant.const import CONF_HOST, CONF_PORT, Platform from homeassistant.core import HomeAssistant from homeassistant.helpers.start import async_at_started -from .const import DOMAIN from .coordinator import CertExpiryDataUpdateCoordinator PLATFORMS = [Platform.SENSOR] +CertExpiryConfigEntry = ConfigEntry[CertExpiryDataUpdateCoordinator] -async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: + +async def async_setup_entry(hass: HomeAssistant, entry: CertExpiryConfigEntry) -> bool: """Load the saved entities.""" host: str = entry.data[CONF_HOST] port: int = entry.data[CONF_PORT] coordinator = CertExpiryDataUpdateCoordinator(hass, host, port) - hass.data.setdefault(DOMAIN, {}) - hass.data[DOMAIN][entry.entry_id] = coordinator + entry.runtime_data = coordinator if entry.unique_id is None: hass.config_entries.async_update_entry(entry, unique_id=f"{host}:{port}") diff --git a/homeassistant/components/cert_expiry/sensor.py b/homeassistant/components/cert_expiry/sensor.py index 6a55e630a35..674f7bb6341 100644 --- a/homeassistant/components/cert_expiry/sensor.py +++ b/homeassistant/components/cert_expiry/sensor.py @@ -12,7 +12,7 @@ from homeassistant.components.sensor import ( SensorDeviceClass, SensorEntity, ) -from homeassistant.config_entries import SOURCE_IMPORT, ConfigEntry +from homeassistant.config_entries import SOURCE_IMPORT from homeassistant.const import CONF_HOST, CONF_PORT, EVENT_HOMEASSISTANT_START from homeassistant.core import Event, HomeAssistant, callback import homeassistant.helpers.config_validation as cv @@ -22,7 +22,7 @@ from homeassistant.helpers.event import async_call_later from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType from homeassistant.helpers.update_coordinator import CoordinatorEntity -from . import CertExpiryDataUpdateCoordinator +from . import CertExpiryConfigEntry, CertExpiryDataUpdateCoordinator from .const import DEFAULT_PORT, DOMAIN SCAN_INTERVAL = timedelta(hours=12) @@ -62,15 +62,13 @@ async def async_setup_platform( async def async_setup_entry( hass: HomeAssistant, - entry: ConfigEntry, + entry: CertExpiryConfigEntry, async_add_entities: AddEntitiesCallback, ) -> None: """Add cert-expiry entry.""" - coordinator: CertExpiryDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id] + coordinator = entry.runtime_data - sensors = [ - SSLCertificateTimestamp(coordinator), - ] + sensors = [SSLCertificateTimestamp(coordinator)] async_add_entities(sensors, True)