Update EventBus listen type signatures (#112760)

This commit is contained in:
Marc Mueller 2024-03-08 21:57:23 +01:00 committed by GitHub
parent cef20506dc
commit de886d8c49
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
11 changed files with 27 additions and 31 deletions

View file

@ -133,7 +133,7 @@ class KafkaManager:
async def start(self) -> None: async def start(self) -> None:
"""Start the Kafka manager.""" """Start the Kafka manager."""
self._hass.bus.async_listen(EVENT_STATE_CHANGED, self.write) # type: ignore[arg-type] self._hass.bus.async_listen(EVENT_STATE_CHANGED, self.write)
await self._producer.start() await self._producer.start()
async def shutdown(self, _: Event) -> None: async def shutdown(self, _: Event) -> None:

View file

@ -397,7 +397,7 @@ class APIDomainServicesView(HomeAssistantView):
cancel_listen = hass.bus.async_listen( cancel_listen = hass.bus.async_listen(
EVENT_STATE_CHANGED, EVENT_STATE_CHANGED,
_async_save_changed_entities, # type: ignore[arg-type] _async_save_changed_entities,
run_immediately=True, run_immediately=True,
) )

View file

@ -161,17 +161,17 @@ class DefaultAgent(AbstractConversationAgent):
self.hass.bus.async_listen( self.hass.bus.async_listen(
ar.EVENT_AREA_REGISTRY_UPDATED, ar.EVENT_AREA_REGISTRY_UPDATED,
self._async_handle_area_registry_changed, # type: ignore[arg-type] self._async_handle_area_registry_changed,
run_immediately=True, run_immediately=True,
) )
self.hass.bus.async_listen( self.hass.bus.async_listen(
er.EVENT_ENTITY_REGISTRY_UPDATED, er.EVENT_ENTITY_REGISTRY_UPDATED,
self._async_handle_entity_registry_changed, # type: ignore[arg-type] self._async_handle_entity_registry_changed,
run_immediately=True, run_immediately=True,
) )
self.hass.bus.async_listen( self.hass.bus.async_listen(
EVENT_STATE_CHANGED, EVENT_STATE_CHANGED,
self._async_handle_state_changed, # type: ignore[arg-type] self._async_handle_state_changed,
run_immediately=True, run_immediately=True,
) )
async_listen_entity_updates( async_listen_entity_updates(

View file

@ -211,7 +211,7 @@ def async_subscribe_events(
subscriptions.append( subscriptions.append(
hass.bus.async_listen( hass.bus.async_listen(
EVENT_STATE_CHANGED, EVENT_STATE_CHANGED,
_forward_state_events_filtered, # type: ignore[arg-type] _forward_state_events_filtered,
run_immediately=True, run_immediately=True,
) )
) )

View file

@ -132,10 +132,10 @@ def setup(hass: HomeAssistant, config: ConfigType) -> bool:
default_metric, default_metric,
) )
hass.bus.listen(EVENT_STATE_CHANGED, metrics.handle_state_changed_event) # type: ignore[arg-type] hass.bus.listen(EVENT_STATE_CHANGED, metrics.handle_state_changed_event)
hass.bus.listen( hass.bus.listen(
EVENT_ENTITY_REGISTRY_UPDATED, EVENT_ENTITY_REGISTRY_UPDATED,
metrics.handle_entity_registry_updated, # type: ignore[arg-type] metrics.handle_entity_registry_updated,
) )
for state in hass.states.all(): for state in hass.states.all():

View file

@ -1406,7 +1406,7 @@ class EventBus:
def listen( def listen(
self, self,
event_type: str, event_type: str,
listener: Callable[[Event], Coroutine[Any, Any, None] | None], listener: Callable[[Event[Any]], Coroutine[Any, Any, None] | None],
) -> CALLBACK_TYPE: ) -> CALLBACK_TYPE:
"""Listen for all events or events of a specific type. """Listen for all events or events of a specific type.
@ -1427,8 +1427,8 @@ class EventBus:
def async_listen( def async_listen(
self, self,
event_type: str, event_type: str,
listener: Callable[[Event], Coroutine[Any, Any, None] | None], listener: Callable[[Event[_DataT]], Coroutine[Any, Any, None] | None],
event_filter: Callable[[Event], bool] | None = None, event_filter: Callable[[Event[_DataT]], bool] | None = None,
run_immediately: bool = False, run_immediately: bool = False,
) -> CALLBACK_TYPE: ) -> CALLBACK_TYPE:
"""Listen for all events or events of a specific type. """Listen for all events or events of a specific type.
@ -1464,7 +1464,7 @@ class EventBus:
@callback @callback
def _async_listen_filterable_job( def _async_listen_filterable_job(
self, event_type: str, filterable_job: _FilterableJobType self, event_type: str, filterable_job: _FilterableJobType[Any]
) -> CALLBACK_TYPE: ) -> CALLBACK_TYPE:
self._listeners.setdefault(event_type, []).append(filterable_job) self._listeners.setdefault(event_type, []).append(filterable_job)
return functools.partial( return functools.partial(
@ -1474,7 +1474,7 @@ class EventBus:
def listen_once( def listen_once(
self, self,
event_type: str, event_type: str,
listener: Callable[[Event], Coroutine[Any, Any, None] | None], listener: Callable[[Event[Any]], Coroutine[Any, Any, None] | None],
) -> CALLBACK_TYPE: ) -> CALLBACK_TYPE:
"""Listen once for event of a specific type. """Listen once for event of a specific type.
@ -1497,7 +1497,7 @@ class EventBus:
def async_listen_once( def async_listen_once(
self, self,
event_type: str, event_type: str,
listener: Callable[[Event], Coroutine[Any, Any, None] | None], listener: Callable[[Event[Any]], Coroutine[Any, Any, None] | None],
) -> CALLBACK_TYPE: ) -> CALLBACK_TYPE:
"""Listen once for event of a specific type. """Listen once for event of a specific type.

View file

@ -329,8 +329,8 @@ class AreaRegistry(BaseRegistry):
self.hass.bus.async_listen( self.hass.bus.async_listen(
event_type=fr.EVENT_FLOOR_REGISTRY_UPDATED, event_type=fr.EVENT_FLOOR_REGISTRY_UPDATED,
event_filter=_removed_from_registry_filter, # type: ignore[arg-type] event_filter=_removed_from_registry_filter,
listener=_handle_floor_registry_update, # type: ignore[arg-type] listener=_handle_floor_registry_update,
) )
@callback @callback
@ -345,8 +345,8 @@ class AreaRegistry(BaseRegistry):
self.hass.bus.async_listen( self.hass.bus.async_listen(
event_type=lr.EVENT_LABEL_REGISTRY_UPDATED, event_type=lr.EVENT_LABEL_REGISTRY_UPDATED,
event_filter=_removed_from_registry_filter, # type: ignore[arg-type] event_filter=_removed_from_registry_filter,
listener=_handle_label_registry_update, # type: ignore[arg-type] listener=_handle_label_registry_update,
) )

View file

@ -1141,8 +1141,8 @@ def async_setup_cleanup(hass: HomeAssistant, dev_reg: DeviceRegistry) -> None:
hass.bus.async_listen( hass.bus.async_listen(
event_type=lr.EVENT_LABEL_REGISTRY_UPDATED, event_type=lr.EVENT_LABEL_REGISTRY_UPDATED,
event_filter=_label_removed_from_registry_filter, # type: ignore[arg-type] event_filter=_label_removed_from_registry_filter,
listener=_handle_label_registry_update, # type: ignore[arg-type] listener=_handle_label_registry_update,
) )
@callback @callback

View file

@ -1380,8 +1380,8 @@ def _async_setup_cleanup(hass: HomeAssistant, registry: EntityRegistry) -> None:
hass.bus.async_listen( hass.bus.async_listen(
event_type=lr.EVENT_LABEL_REGISTRY_UPDATED, event_type=lr.EVENT_LABEL_REGISTRY_UPDATED,
event_filter=_label_removed_from_registry_filter, # type: ignore[arg-type] event_filter=_label_removed_from_registry_filter,
listener=_handle_label_registry_update, # type: ignore[arg-type] listener=_handle_label_registry_update,
) )
@callback @callback

View file

@ -287,9 +287,7 @@ def async_track_state_change(
return async_track_state_change_event(hass, entity_ids, state_change_listener) return async_track_state_change_event(hass, entity_ids, state_change_listener)
return hass.bus.async_listen( return hass.bus.async_listen(
EVENT_STATE_CHANGED, EVENT_STATE_CHANGED, state_change_dispatcher, event_filter=state_change_filter
state_change_dispatcher, # type: ignore[arg-type]
event_filter=state_change_filter, # type: ignore[arg-type]
) )
@ -806,8 +804,7 @@ class _TrackStateChangeFiltered:
@callback @callback
def _setup_all_listener(self) -> None: def _setup_all_listener(self) -> None:
self._listeners[_ALL_LISTENER] = self.hass.bus.async_listen( self._listeners[_ALL_LISTENER] = self.hass.bus.async_listen(
EVENT_STATE_CHANGED, EVENT_STATE_CHANGED, self._action
self._action, # type: ignore[arg-type]
) )
@ -1387,8 +1384,7 @@ def async_track_same_state(
if entity_ids == MATCH_ALL: if entity_ids == MATCH_ALL:
async_remove_state_for_cancel = hass.bus.async_listen( async_remove_state_for_cancel = hass.bus.async_listen(
EVENT_STATE_CHANGED, EVENT_STATE_CHANGED, state_for_cancel_listener
state_for_cancel_listener, # type: ignore[arg-type]
) )
else: else:
async_remove_state_for_cancel = async_track_state_change_event( async_remove_state_for_cancel = async_track_state_change_event(

View file

@ -585,7 +585,7 @@ def _async_when_setup(
listeners: list[CALLBACK_TYPE] = [] listeners: list[CALLBACK_TYPE] = []
async def _matched_event(event: Event) -> None: async def _matched_event(event: Event[Any]) -> None:
"""Call the callback when we matched an event.""" """Call the callback when we matched an event."""
for listener in listeners: for listener in listeners:
listener() listener()
@ -600,7 +600,7 @@ def _async_when_setup(
hass.bus.async_listen( hass.bus.async_listen(
EVENT_COMPONENT_LOADED, EVENT_COMPONENT_LOADED,
_matched_event, _matched_event,
event_filter=_async_is_component_filter, # type: ignore[arg-type] event_filter=_async_is_component_filter,
) )
) )
if start_event: if start_event: