diff --git a/homeassistant/components/fully_kiosk/__init__.py b/homeassistant/components/fully_kiosk/__init__.py index 074ec3feaa0..772e7f79242 100644 --- a/homeassistant/components/fully_kiosk/__init__.py +++ b/homeassistant/components/fully_kiosk/__init__.py @@ -1,17 +1,14 @@ """The Fully Kiosk Browser integration.""" -from homeassistant.config_entries import ConfigEntry from homeassistant.const import Platform from homeassistant.core import HomeAssistant from homeassistant.helpers import config_validation as cv from homeassistant.helpers.typing import ConfigType from .const import DOMAIN -from .coordinator import FullyKioskDataUpdateCoordinator +from .coordinator import FullyKioskConfigEntry, FullyKioskDataUpdateCoordinator from .services import async_setup_services -type FullyKioskConfigEntry = ConfigEntry[FullyKioskDataUpdateCoordinator] - PLATFORMS = [ Platform.BINARY_SENSOR, Platform.BUTTON, diff --git a/homeassistant/components/fully_kiosk/coordinator.py b/homeassistant/components/fully_kiosk/coordinator.py index 405f0746437..dc3640020be 100644 --- a/homeassistant/components/fully_kiosk/coordinator.py +++ b/homeassistant/components/fully_kiosk/coordinator.py @@ -14,11 +14,15 @@ from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, Upda from .const import DEFAULT_PORT, LOGGER, UPDATE_INTERVAL +type FullyKioskConfigEntry = ConfigEntry[FullyKioskDataUpdateCoordinator] + class FullyKioskDataUpdateCoordinator(DataUpdateCoordinator): """Define an object to hold Fully Kiosk Browser data.""" - def __init__(self, hass: HomeAssistant, entry: ConfigEntry) -> None: + config_entry: FullyKioskConfigEntry + + def __init__(self, hass: HomeAssistant, entry: FullyKioskConfigEntry) -> None: """Initialize.""" self.use_ssl = entry.data.get(CONF_SSL, False) self.fully = FullyKiosk( @@ -32,6 +36,7 @@ class FullyKioskDataUpdateCoordinator(DataUpdateCoordinator): super().__init__( hass, LOGGER, + config_entry=entry, name=entry.data[CONF_HOST], update_interval=UPDATE_INTERVAL, )