Replace EventType with Event [t-z] (#112742)

This commit is contained in:
Marc Mueller 2024-03-08 19:37:26 +01:00 committed by GitHub
parent 84c44c1835
commit 3db28d46b2
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
10 changed files with 31 additions and 34 deletions

View file

@ -21,6 +21,7 @@ from homeassistant.const import (
from homeassistant.core import ( from homeassistant.core import (
CALLBACK_TYPE, CALLBACK_TYPE,
Context, Context,
Event,
HomeAssistant, HomeAssistant,
State, State,
callback, callback,
@ -47,7 +48,7 @@ from homeassistant.helpers.trigger_template_entity import (
TEMPLATE_ENTITY_BASE_SCHEMA, TEMPLATE_ENTITY_BASE_SCHEMA,
make_template_entity_base_schema, make_template_entity_base_schema,
) )
from homeassistant.helpers.typing import ConfigType, EventType from homeassistant.helpers.typing import ConfigType
from .const import ( from .const import (
CONF_ATTRIBUTE_TEMPLATES, CONF_ATTRIBUTE_TEMPLATES,
@ -190,7 +191,7 @@ class _TemplateAttribute:
@callback @callback
def handle_result( def handle_result(
self, self,
event: EventType[EventStateChangedData] | None, event: Event[EventStateChangedData] | None,
template: Template, template: Template,
last_result: str | None | TemplateError, last_result: str | None | TemplateError,
result: str | TemplateError, result: str | TemplateError,
@ -399,7 +400,7 @@ class TemplateEntity(Entity):
@callback @callback
def _handle_results( def _handle_results(
self, self,
event: EventType[EventStateChangedData] | None, event: Event[EventStateChangedData] | None,
updates: list[TrackTemplateResult], updates: list[TrackTemplateResult],
) -> None: ) -> None:
"""Call back the results to the attributes.""" """Call back the results to the attributes."""

View file

@ -8,7 +8,7 @@ import voluptuous as vol
from homeassistant import exceptions from homeassistant import exceptions
from homeassistant.const import CONF_FOR, CONF_PLATFORM, CONF_VALUE_TEMPLATE from homeassistant.const import CONF_FOR, CONF_PLATFORM, CONF_VALUE_TEMPLATE
from homeassistant.core import CALLBACK_TYPE, HassJob, HomeAssistant, callback from homeassistant.core import CALLBACK_TYPE, Event, HassJob, HomeAssistant, callback
from homeassistant.helpers import config_validation as cv, template from homeassistant.helpers import config_validation as cv, template
from homeassistant.helpers.event import ( from homeassistant.helpers.event import (
EventStateChangedData, EventStateChangedData,
@ -19,7 +19,7 @@ from homeassistant.helpers.event import (
) )
from homeassistant.helpers.template import Template, result_as_boolean from homeassistant.helpers.template import Template, result_as_boolean
from homeassistant.helpers.trigger import TriggerActionType, TriggerInfo from homeassistant.helpers.trigger import TriggerActionType, TriggerInfo
from homeassistant.helpers.typing import ConfigType, EventType from homeassistant.helpers.typing import ConfigType
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
@ -65,7 +65,7 @@ async def async_attach_trigger(
@callback @callback
def template_listener( def template_listener(
event: EventType[EventStateChangedData] | None, event: Event[EventStateChangedData] | None,
updates: list[TrackTemplateResult], updates: list[TrackTemplateResult],
) -> None: ) -> None:
"""Listen for state changes and calls action.""" """Listen for state changes and calls action."""

View file

@ -22,7 +22,7 @@ from homeassistant.const import (
STATE_UNAVAILABLE, STATE_UNAVAILABLE,
STATE_UNKNOWN, STATE_UNKNOWN,
) )
from homeassistant.core import HomeAssistant, callback from homeassistant.core import Event, HomeAssistant, callback
from homeassistant.helpers import ( from homeassistant.helpers import (
config_validation as cv, config_validation as cv,
device_registry as dr, device_registry as dr,
@ -34,7 +34,7 @@ from homeassistant.helpers.event import (
EventStateChangedData, EventStateChangedData,
async_track_state_change_event, async_track_state_change_event,
) )
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType, EventType from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from .const import CONF_HYSTERESIS, CONF_LOWER, CONF_UPPER from .const import CONF_HYSTERESIS, CONF_LOWER, CONF_UPPER
@ -215,7 +215,7 @@ class ThresholdSensor(BinarySensorEntity):
@callback @callback
def async_threshold_sensor_state_listener( def async_threshold_sensor_state_listener(
event: EventType[EventStateChangedData], event: Event[EventStateChangedData],
) -> None: ) -> None:
"""Handle sensor state changes.""" """Handle sensor state changes."""
_update_sensor_state() _update_sensor_state()

View file

@ -31,7 +31,7 @@ from homeassistant.const import (
STATE_UNAVAILABLE, STATE_UNAVAILABLE,
STATE_UNKNOWN, STATE_UNKNOWN,
) )
from homeassistant.core import HomeAssistant, callback from homeassistant.core import Event, HomeAssistant, callback
import homeassistant.helpers.config_validation as cv import homeassistant.helpers.config_validation as cv
from homeassistant.helpers.entity import generate_entity_id from homeassistant.helpers.entity import generate_entity_id
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddEntitiesCallback
@ -41,7 +41,7 @@ from homeassistant.helpers.event import (
) )
from homeassistant.helpers.reload import async_setup_reload_service from homeassistant.helpers.reload import async_setup_reload_service
from homeassistant.helpers.restore_state import RestoreEntity from homeassistant.helpers.restore_state import RestoreEntity
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType, EventType from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from homeassistant.util.dt import utcnow from homeassistant.util.dt import utcnow
from . import PLATFORMS from . import PLATFORMS
@ -215,7 +215,7 @@ class SensorTrend(BinarySensorEntity, RestoreEntity):
@callback @callback
def trend_sensor_state_listener( def trend_sensor_state_listener(
event: EventType[EventStateChangedData], event: Event[EventStateChangedData],
) -> None: ) -> None:
"""Handle state changes on the observed device.""" """Handle state changes on the observed device."""
if (new_state := event.data["new_state"]) is None: if (new_state := event.data["new_state"]) is None:

View file

@ -79,7 +79,7 @@ from homeassistant.const import (
STATE_UNAVAILABLE, STATE_UNAVAILABLE,
STATE_UNKNOWN, STATE_UNKNOWN,
) )
from homeassistant.core import HomeAssistant, callback from homeassistant.core import Event, HomeAssistant, callback
from homeassistant.exceptions import TemplateError from homeassistant.exceptions import TemplateError
from homeassistant.helpers import config_validation as cv from homeassistant.helpers import config_validation as cv
from homeassistant.helpers.entity_component import EntityComponent from homeassistant.helpers.entity_component import EntityComponent
@ -93,7 +93,7 @@ from homeassistant.helpers.event import (
) )
from homeassistant.helpers.reload import async_setup_reload_service from homeassistant.helpers.reload import async_setup_reload_service
from homeassistant.helpers.service import async_call_from_config from homeassistant.helpers.service import async_call_from_config
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType, EventType from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
ATTR_ACTIVE_CHILD = "active_child" ATTR_ACTIVE_CHILD = "active_child"
@ -186,7 +186,7 @@ class UniversalMediaPlayer(MediaPlayerEntity):
@callback @callback
def _async_on_dependency_update( def _async_on_dependency_update(
event: EventType[EventStateChangedData], event: Event[EventStateChangedData],
) -> None: ) -> None:
"""Update ha state when dependencies update.""" """Update ha state when dependencies update."""
self.async_set_context(event.context) self.async_set_context(event.context)
@ -194,7 +194,7 @@ class UniversalMediaPlayer(MediaPlayerEntity):
@callback @callback
def _async_on_template_update( def _async_on_template_update(
event: EventType[EventStateChangedData] | None, event: Event[EventStateChangedData] | None,
updates: list[TrackTemplateResult], updates: list[TrackTemplateResult],
) -> None: ) -> None:
"""Update state when template state changes.""" """Update state when template state changes."""

View file

@ -28,7 +28,7 @@ from homeassistant.const import (
STATE_UNKNOWN, STATE_UNKNOWN,
UnitOfEnergy, UnitOfEnergy,
) )
from homeassistant.core import HomeAssistant, State, callback from homeassistant.core import Event, HomeAssistant, State, callback
from homeassistant.helpers import ( from homeassistant.helpers import (
device_registry as dr, device_registry as dr,
entity_platform, entity_platform,
@ -44,7 +44,7 @@ from homeassistant.helpers.event import (
) )
from homeassistant.helpers.start import async_at_started from homeassistant.helpers.start import async_at_started
from homeassistant.helpers.template import is_number from homeassistant.helpers.template import is_number
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType, EventType from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from homeassistant.util import slugify from homeassistant.util import slugify
import homeassistant.util.dt as dt_util import homeassistant.util.dt as dt_util
@ -466,7 +466,7 @@ class UtilityMeterSensor(RestoreSensor):
return None return None
@callback @callback
def async_reading(self, event: EventType[EventStateChangedData]) -> None: def async_reading(self, event: Event[EventStateChangedData]) -> None:
"""Handle the sensor state changes.""" """Handle the sensor state changes."""
if ( if (
source_state := self.hass.states.get(self._sensor_source_id) source_state := self.hass.states.get(self._sensor_source_id)
@ -517,7 +517,7 @@ class UtilityMeterSensor(RestoreSensor):
self.async_write_ha_state() self.async_write_ha_state()
@callback @callback
def async_tariff_change(self, event: EventType[EventStateChangedData]) -> None: def async_tariff_change(self, event: Event[EventStateChangedData]) -> None:
"""Handle tariff changes.""" """Handle tariff changes."""
if (new_state := event.data["new_state"]) is None: if (new_state := event.data["new_state"]) is None:
return return

View file

@ -48,7 +48,6 @@ from homeassistant.helpers.json import (
json_bytes, json_bytes,
) )
from homeassistant.helpers.service import async_get_all_descriptions from homeassistant.helpers.service import async_get_all_descriptions
from homeassistant.helpers.typing import EventType
from homeassistant.loader import ( from homeassistant.loader import (
Integration, Integration,
IntegrationNotFound, IntegrationNotFound,
@ -622,7 +621,7 @@ async def handle_render_template(
@callback @callback
def _template_listener( def _template_listener(
event: EventType[EventStateChangedData] | None, event: Event[EventStateChangedData] | None,
updates: list[TrackTemplateResult], updates: list[TrackTemplateResult],
) -> None: ) -> None:
track_template_result = updates.pop() track_template_result = updates.pop()

View file

@ -11,7 +11,7 @@ from typing import TYPE_CHECKING, Any, Self
from zigpy.quirks.v2 import EntityMetadata, EntityType from zigpy.quirks.v2 import EntityMetadata, EntityType
from homeassistant.const import ATTR_NAME, EntityCategory from homeassistant.const import ATTR_NAME, EntityCategory
from homeassistant.core import CALLBACK_TYPE, callback from homeassistant.core import CALLBACK_TYPE, Event, callback
from homeassistant.helpers import entity from homeassistant.helpers import entity
from homeassistant.helpers.debounce import Debouncer from homeassistant.helpers.debounce import Debouncer
from homeassistant.helpers.device_registry import CONNECTION_ZIGBEE, DeviceInfo from homeassistant.helpers.device_registry import CONNECTION_ZIGBEE, DeviceInfo
@ -24,7 +24,6 @@ from homeassistant.helpers.event import (
async_track_state_change_event, async_track_state_change_event,
) )
from homeassistant.helpers.restore_state import RestoreEntity from homeassistant.helpers.restore_state import RestoreEntity
from homeassistant.helpers.typing import EventType
from .core.const import ( from .core.const import (
ATTR_MANUFACTURER, ATTR_MANUFACTURER,
@ -346,9 +345,7 @@ class ZhaGroupEntity(BaseZhaEntity):
self.async_on_remove(send_removed_signal) self.async_on_remove(send_removed_signal)
@callback @callback
def async_state_changed_listener( def async_state_changed_listener(self, event: Event[EventStateChangedData]) -> None:
self, event: EventType[EventStateChangedData]
) -> None:
"""Handle child updates.""" """Handle child updates."""
# Delay to ensure that we get updates from all members before updating the group # Delay to ensure that we get updates from all members before updating the group
assert self._change_listener_debouncer assert self._change_listener_debouncer

View file

@ -38,7 +38,7 @@ from homeassistant.helpers import (
service, service,
storage, storage,
) )
from homeassistant.helpers.typing import ConfigType, EventType from homeassistant.helpers.typing import ConfigType
from homeassistant.loader import bind_hass from homeassistant.loader import bind_hass
from homeassistant.util.location import distance from homeassistant.util.location import distance
@ -166,7 +166,7 @@ def async_setup_track_zone_entity_ids(hass: HomeAssistant) -> None:
@callback @callback
def _async_add_zone_entity_id( def _async_add_zone_entity_id(
event_: EventType[event.EventStateChangedData], event_: Event[event.EventStateChangedData],
) -> None: ) -> None:
"""Add zone entity ID.""" """Add zone entity ID."""
zone_entity_ids.append(event_.data["entity_id"]) zone_entity_ids.append(event_.data["entity_id"])
@ -174,7 +174,7 @@ def async_setup_track_zone_entity_ids(hass: HomeAssistant) -> None:
@callback @callback
def _async_remove_zone_entity_id( def _async_remove_zone_entity_id(
event_: EventType[event.EventStateChangedData], event_: Event[event.EventStateChangedData],
) -> None: ) -> None:
"""Remove zone entity ID.""" """Remove zone entity ID."""
zone_entity_ids.remove(event_.data["entity_id"]) zone_entity_ids.remove(event_.data["entity_id"])
@ -389,7 +389,7 @@ class Zone(collection.CollectionEntity):
@callback @callback
def _person_state_change_listener( def _person_state_change_listener(
self, evt: EventType[event.EventStateChangedData] self, evt: Event[event.EventStateChangedData]
) -> None: ) -> None:
person_entity_id = evt.data["entity_id"] person_entity_id = evt.data["entity_id"]
cur_count = len(self._persons_in_zone) cur_count = len(self._persons_in_zone)

View file

@ -13,7 +13,7 @@ from homeassistant.const import (
CONF_PLATFORM, CONF_PLATFORM,
CONF_ZONE, CONF_ZONE,
) )
from homeassistant.core import CALLBACK_TYPE, HassJob, HomeAssistant, callback from homeassistant.core import CALLBACK_TYPE, Event, HassJob, HomeAssistant, callback
from homeassistant.helpers import ( from homeassistant.helpers import (
condition, condition,
config_validation as cv, config_validation as cv,
@ -25,7 +25,7 @@ from homeassistant.helpers.event import (
async_track_state_change_event, async_track_state_change_event,
) )
from homeassistant.helpers.trigger import TriggerActionType, TriggerInfo from homeassistant.helpers.trigger import TriggerActionType, TriggerInfo
from homeassistant.helpers.typing import ConfigType, EventType from homeassistant.helpers.typing import ConfigType
EVENT_ENTER = "enter" EVENT_ENTER = "enter"
EVENT_LEAVE = "leave" EVENT_LEAVE = "leave"
@ -75,7 +75,7 @@ async def async_attach_trigger(
job = HassJob(action) job = HassJob(action)
@callback @callback
def zone_automation_listener(zone_event: EventType[EventStateChangedData]) -> None: def zone_automation_listener(zone_event: Event[EventStateChangedData]) -> None:
"""Listen for state changes and calls action.""" """Listen for state changes and calls action."""
entity = zone_event.data["entity_id"] entity = zone_event.data["entity_id"]
from_s = zone_event.data["old_state"] from_s = zone_event.data["old_state"]