jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-21 12:17:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improve type hints in utility_meter select entity (#76447)

Browse Source
This commit is contained in:
epenet 2022-08-17 16:38:41 +02:00 committed by GitHub
parent 79ab13d118
commit 8619df5294
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
homeassistant/components/utility_meter/select.py
View File

@ -23,7 +23,7 @@ async def async_setup_entry(
) -> None: ) -> None:
"""Initialize Utility Meter config entry.""" """Initialize Utility Meter config entry."""
name = config_entry.title name = config_entry.title
tariffs = config_entry.options[CONF_TARIFFS] tariffs: list[str] = config_entry.options[CONF_TARIFFS]
unique_id = config_entry.entry_id unique_id = config_entry.entry_id
tariff_select = TariffSelect(name, tariffs, unique_id) tariff_select = TariffSelect(name, tariffs, unique_id)
@ -52,7 +52,7 @@ async def async_setup_platform(
async_add_entities( async_add_entities(
[ [
TariffSelect( TariffSelect(
discovery_info[CONF_METER], meter,
discovery_info[CONF_TARIFFS], discovery_info[CONF_TARIFFS],
conf_meter_unique_id, conf_meter_unique_id,
) )
@ -67,22 +67,22 @@ class TariffSelect(SelectEntity, RestoreEntity):
"""Initialize a tariff selector.""" """Initialize a tariff selector."""
self._attr_name = name self._attr_name = name
self._attr_unique_id = unique_id self._attr_unique_id = unique_id
self._current_tariff = None self._current_tariff: str | None = None
self._tariffs = tariffs self._tariffs = tariffs
self._attr_icon = TARIFF_ICON self._attr_icon = TARIFF_ICON
self._attr_should_poll = False self._attr_should_poll = False
@property @property
def options(self): def options(self) -> list[str]:
"""Return the available tariffs.""" """Return the available tariffs."""
return self._tariffs return self._tariffs
@property @property
def current_option(self): def current_option(self) -> str | None:
"""Return current tariff.""" """Return current tariff."""
return self._current_tariff return self._current_tariff
async def async_added_to_hass(self): async def async_added_to_hass(self) -> None:
"""Run when entity about to be added.""" """Run when entity about to be added."""
await super().async_added_to_hass() await super().async_added_to_hass()