Add setup type hints to ebusd

This commit is contained in:
epenet
2022-01-10 07:50:46 +00:00
parent 18706e6408
commit a630372912

View File

@@ -1,8 +1,13 @@
"""Support for Ebusd sensors.""" """Support for Ebusd sensors."""
from __future__ import annotations
import datetime import datetime
import logging import logging
from homeassistant.components.sensor import SensorEntity from homeassistant.components.sensor import SensorEntity
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from homeassistant.util import Throttle from homeassistant.util import Throttle
import homeassistant.util.dt as dt_util import homeassistant.util.dt as dt_util
@@ -19,8 +24,15 @@ MIN_TIME_BETWEEN_UPDATES = datetime.timedelta(seconds=15)
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
def setup_platform(hass, config, add_entities, discovery_info=None): def setup_platform(
hass: HomeAssistant,
config: ConfigType,
add_entities: AddEntitiesCallback,
discovery_info: DiscoveryInfoType | None = None,
) -> None:
"""Set up the Ebus sensor.""" """Set up the Ebus sensor."""
if not discovery_info:
return
ebusd_api = hass.data[DOMAIN] ebusd_api = hass.data[DOMAIN]
monitored_conditions = discovery_info["monitored_conditions"] monitored_conditions = discovery_info["monitored_conditions"]
name = discovery_info["client_name"] name = discovery_info["client_name"]