evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add type hints to homekit_controller (#72155)

Browse source
This commit is contained in:
epenet 2022-05-19 18:34:51 +02:00 • committed by GitHub
parent 8ef39c3cfd
commit 0422d7f256
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
homeassistant/components/homekit_controller/device_trigger.py
View file

@ -1,6 +1,7 @@
"""Provides device automations for homekit devices.""" """Provides device automations for homekit devices."""
from __future__ import annotations from __future__ import annotations
from collections.abc import Generator
from typing import TYPE_CHECKING, Any from typing import TYPE_CHECKING, Any
from aiohomekit.model.characteristics import CharacteristicsTypes from aiohomekit.model.characteristics import CharacteristicsTypes
@ -63,7 +64,7 @@ class TriggerSource:
self._hass = connection.hass self._hass = connection.hass
self._connection = connection self._connection = connection
self._aid = aid self._aid = aid
self._triggers = {} self._triggers: dict[tuple[str, str], dict[str, Any]] = {}
for trigger in triggers: for trigger in triggers:
self._triggers[(trigger["type"], trigger["subtype"])] = trigger self._triggers[(trigger["type"], trigger["subtype"])] = trigger
self._callbacks = {} self._callbacks = {}
@ -73,7 +74,7 @@ class TriggerSource:
for event_handler in self._callbacks.get(iid, []): for event_handler in self._callbacks.get(iid, []):
event_handler(value) event_handler(value)
def async_get_triggers(self): def async_get_triggers(self) -> Generator[tuple[str, str], None, None]:
"""List device triggers for homekit devices.""" """List device triggers for homekit devices."""
yield from self._triggers yield from self._triggers
@ -246,7 +247,7 @@ async def async_get_triggers(
if device_id not in hass.data.get(TRIGGERS, {}): if device_id not in hass.data.get(TRIGGERS, {}):
return [] return []
device = hass.data[TRIGGERS][device_id] device: TriggerSource = hass.data[TRIGGERS][device_id]
return [ return [
{ {