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

Use runtime data in Yale Smart Alarm (#116548)

Browse Source
This commit is contained in:
G Johansson 2024-05-01 21:13:29 +02:00 committed by GitHub
parent ae6a497cd1
commit ef2ae7b600
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
7 changed files with 22 additions and 42 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
homeassistant/components/yale_smart_alarm/__init__.py
View File

@ -9,11 +9,13 @@ from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed from homeassistant.exceptions import ConfigEntryAuthFailed
from homeassistant.helpers import entity_registry as er from homeassistant.helpers import entity_registry as er
from .const import COORDINATOR, DOMAIN, LOGGER, PLATFORMS from .const import LOGGER, PLATFORMS
from .coordinator import YaleDataUpdateCoordinator from .coordinator import YaleDataUpdateCoordinator
YaleConfigEntry = ConfigEntry["YaleDataUpdateCoordinator"]
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
async def async_setup_entry(hass: HomeAssistant, entry: YaleConfigEntry) -> bool:
"""Set up Yale from a config entry.""" """Set up Yale from a config entry."""
coordinator = YaleDataUpdateCoordinator(hass, entry) coordinator = YaleDataUpdateCoordinator(hass, entry)
@ -21,9 +23,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
raise ConfigEntryAuthFailed raise ConfigEntryAuthFailed
await coordinator.async_config_entry_first_refresh() await coordinator.async_config_entry_first_refresh()
hass.data.setdefault(DOMAIN, {})[entry.entry_id] = { entry.runtime_data = coordinator
COORDINATOR: coordinator,
}
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS) await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
entry.async_on_unload(entry.add_update_listener(update_listener)) entry.async_on_unload(entry.add_update_listener(update_listener))
@ -38,11 +38,7 @@ async def update_listener(hass: HomeAssistant, entry: ConfigEntry) -> None:
async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
"""Unload a config entry.""" """Unload a config entry."""
return await hass.config_entries.async_unload_platforms(entry, PLATFORMS)
if await hass.config_entries.async_unload_platforms(entry, PLATFORMS):
hass.data[DOMAIN].pop(entry.entry_id)
return True
return False
async def async_migrate_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: async def async_migrate_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:

10
homeassistant/components/yale_smart_alarm/alarm_control_panel.py
View File

@ -14,26 +14,24 @@ from homeassistant.components.alarm_control_panel import (
AlarmControlPanelEntity, AlarmControlPanelEntity,
AlarmControlPanelEntityFeature, AlarmControlPanelEntityFeature,
) )
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_NAME from homeassistant.const import CONF_NAME
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.typing import StateType from homeassistant.helpers.typing import StateType
from .const import COORDINATOR, DOMAIN, STATE_MAP, YALE_ALL_ERRORS from . import YaleConfigEntry
from .const import DOMAIN, STATE_MAP, YALE_ALL_ERRORS
from .coordinator import YaleDataUpdateCoordinator from .coordinator import YaleDataUpdateCoordinator
from .entity import YaleAlarmEntity from .entity import YaleAlarmEntity
async def async_setup_entry( async def async_setup_entry(
hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback hass: HomeAssistant, entry: YaleConfigEntry, async_add_entities: AddEntitiesCallback
) -> None: ) -> None:
"""Set up the alarm entry.""" """Set up the alarm entry."""
async_add_entities( async_add_entities([YaleAlarmDevice(coordinator=entry.runtime_data)])
[YaleAlarmDevice(coordinator=hass.data[DOMAIN][entry.entry_id][COORDINATOR])]
)
class YaleAlarmDevice(YaleAlarmEntity, AlarmControlPanelEntity): class YaleAlarmDevice(YaleAlarmEntity, AlarmControlPanelEntity):

9
homeassistant/components/yale_smart_alarm/binary_sensor.py
View File

@ -7,12 +7,11 @@ from homeassistant.components.binary_sensor import (
BinarySensorEntity, BinarySensorEntity,
BinarySensorEntityDescription, BinarySensorEntityDescription,
) )
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import EntityCategory from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddEntitiesCallback
from .const import COORDINATOR, DOMAIN from . import YaleConfigEntry
from .coordinator import YaleDataUpdateCoordinator from .coordinator import YaleDataUpdateCoordinator
from .entity import YaleAlarmEntity, YaleEntity from .entity import YaleAlarmEntity, YaleEntity
@ -45,13 +44,11 @@ SENSOR_TYPES = (
async def async_setup_entry( async def async_setup_entry(
hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback hass: HomeAssistant, entry: YaleConfigEntry, async_add_entities: AddEntitiesCallback
) -> None: ) -> None:
"""Set up the Yale binary sensor entry.""" """Set up the Yale binary sensor entry."""
coordinator: YaleDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id][ coordinator = entry.runtime_data
COORDINATOR
]
sensors: list[YaleDoorSensor | YaleProblemSensor] = [ sensors: list[YaleDoorSensor | YaleProblemSensor] = [
YaleDoorSensor(coordinator, data) for data in coordinator.data["door_windows"] YaleDoorSensor(coordinator, data) for data in coordinator.data["door_windows"]
] ]

9
homeassistant/components/yale_smart_alarm/button.py
View File

@ -5,11 +5,10 @@ from __future__ import annotations
from typing import TYPE_CHECKING from typing import TYPE_CHECKING
from homeassistant.components.button import ButtonEntity, ButtonEntityDescription from homeassistant.components.button import ButtonEntity, ButtonEntityDescription
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddEntitiesCallback
from .const import COORDINATOR, DOMAIN from . import YaleConfigEntry
from .coordinator import YaleDataUpdateCoordinator from .coordinator import YaleDataUpdateCoordinator
from .entity import YaleAlarmEntity from .entity import YaleAlarmEntity
@ -23,14 +22,12 @@ BUTTON_TYPES = (
async def async_setup_entry( async def async_setup_entry(
hass: HomeAssistant, hass: HomeAssistant,
entry: ConfigEntry, entry: YaleConfigEntry,
async_add_entities: AddEntitiesCallback, async_add_entities: AddEntitiesCallback,
) -> None: ) -> None:
"""Set up the button from a config entry.""" """Set up the button from a config entry."""
coordinator: YaleDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id][ coordinator = entry.runtime_data
COORDINATOR
]
async_add_entities( async_add_entities(
[YalePanicButton(coordinator, description) for description in BUTTON_TYPES] [YalePanicButton(coordinator, description) for description in BUTTON_TYPES]

1
homeassistant/components/yale_smart_alarm/const.py
View File

@ -26,7 +26,6 @@ MANUFACTURER = "Yale"
MODEL = "main" MODEL = "main"
DOMAIN = "yale_smart_alarm" DOMAIN = "yale_smart_alarm"
COORDINATOR = "coordinator"
DEFAULT_SCAN_INTERVAL = 15 DEFAULT_SCAN_INTERVAL = 15

10
homeassistant/components/yale_smart_alarm/diagnostics.py
View File

@ -5,11 +5,9 @@ from __future__ import annotations
from typing import Any from typing import Any
from homeassistant.components.diagnostics import async_redact_data from homeassistant.components.diagnostics import async_redact_data
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from .const import COORDINATOR, DOMAIN from . import YaleConfigEntry
from .coordinator import YaleDataUpdateCoordinator
TO_REDACT = { TO_REDACT = {
"address", "address",
@ -24,12 +22,10 @@ TO_REDACT = {
async def async_get_config_entry_diagnostics( async def async_get_config_entry_diagnostics(
hass: HomeAssistant, entry: ConfigEntry hass: HomeAssistant, entry: YaleConfigEntry
) -> dict[str, Any]: ) -> dict[str, Any]:
"""Return diagnostics for a config entry.""" """Return diagnostics for a config entry."""
coordinator: YaleDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id][ coordinator = entry.runtime_data
COORDINATOR
]
assert coordinator.yale assert coordinator.yale
get_all_data = await hass.async_add_executor_job(coordinator.yale.get_all) get_all_data = await hass.async_add_executor_job(coordinator.yale.get_all)

9
homeassistant/components/yale_smart_alarm/lock.py
View File

@ -5,15 +5,14 @@ from __future__ import annotations
from typing import TYPE_CHECKING, Any from typing import TYPE_CHECKING, Any
from homeassistant.components.lock import LockEntity from homeassistant.components.lock import LockEntity
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_CODE from homeassistant.const import ATTR_CODE
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.exceptions import HomeAssistantError from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddEntitiesCallback
from . import YaleConfigEntry
from .const import ( from .const import (
CONF_LOCK_CODE_DIGITS, CONF_LOCK_CODE_DIGITS,
COORDINATOR,
DEFAULT_LOCK_CODE_DIGITS, DEFAULT_LOCK_CODE_DIGITS,
DOMAIN, DOMAIN,
YALE_ALL_ERRORS, YALE_ALL_ERRORS,
@ -23,13 +22,11 @@ from .entity import YaleEntity
async def async_setup_entry( async def async_setup_entry(
hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback hass: HomeAssistant, entry: YaleConfigEntry, async_add_entities: AddEntitiesCallback
) -> None: ) -> None:
"""Set up the Yale lock entry.""" """Set up the Yale lock entry."""
coordinator: YaleDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id][ coordinator = entry.runtime_data
COORDINATOR
]
code_format = entry.options.get(CONF_LOCK_CODE_DIGITS, DEFAULT_LOCK_CODE_DIGITS) code_format = entry.options.get(CONF_LOCK_CODE_DIGITS, DEFAULT_LOCK_CODE_DIGITS)
async_add_entities( async_add_entities(