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

Extract lametric device from coordinator in notify (#78027)

Browse Source
This commit is contained in:
Joakim Sørensen 2022-09-08 10:50:14 +02:00 committed by GitHub
parent 5276d849ec
commit 11e897a5e8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
homeassistant/components/lametric/notify.py
View File

@ -21,6 +21,7 @@ from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from .const import CONF_CYCLES, CONF_ICON_TYPE, CONF_PRIORITY, CONF_SOUND, DOMAIN from .const import CONF_CYCLES, CONF_ICON_TYPE, CONF_PRIORITY, CONF_SOUND, DOMAIN
from .coordinator import LaMetricDataUpdateCoordinator
async def async_get_service( async def async_get_service(
@ -31,8 +32,10 @@ async def async_get_service(
"""Get the LaMetric notification service.""" """Get the LaMetric notification service."""
if discovery_info is None: if discovery_info is None:
return None return None
lametric: LaMetricDevice = hass.data[DOMAIN][discovery_info["entry_id"]] coordinator: LaMetricDataUpdateCoordinator = hass.data[DOMAIN][
return LaMetricNotificationService(lametric) discovery_info["entry_id"]
]
return LaMetricNotificationService(coordinator.lametric)
class LaMetricNotificationService(BaseNotificationService): class LaMetricNotificationService(BaseNotificationService):