Home-Assistant
/
core
mirror of https://github.com/home-assistant/core.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Use PEP 695 for decorator typing with type aliases in esphome (#124234)

pull/124253/head
Erik Montnemery 2024-08-19 21:13:48 +02:00 committed by GitHub
parent 89728f41e1
commit 69652ca2ca
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
homeassistant/components/esphome/entity.py
View File

@ -5,7 +5,7 @@ from __future__ import annotations
from collections.abc import Awaitable, Callable, Coroutine
import functools
import math
from typing import TYPE_CHECKING, Any, Concatenate, Generic, ParamSpec, TypeVar, cast
from typing import TYPE_CHECKING, Any, Concatenate, Generic, TypeVar, cast
from aioesphomeapi import (
APIConnectionError,
@ -30,8 +30,6 @@ from homeassistant.helpers.entity_platform import AddEntitiesCallback
from .entry_data import ESPHomeConfigEntry, RuntimeEntryData
from .enum_mapper import EsphomeEnumMapper
_R = TypeVar("_R")
_P = ParamSpec("_P")
_InfoT = TypeVar("_InfoT", bound=EntityInfo)
_EntityT = TypeVar("_EntityT", bound="EsphomeEntity[Any,Any]")
_StateT = TypeVar("_StateT", bound=EntityState)
@ -116,7 +114,7 @@ async def platform_async_setup_entry(
)
def esphome_state_property(
def esphome_state_property[_R, _EntityT: EsphomeEntity[Any, Any]](
func: Callable[[_EntityT], _R],
) -> Callable[[_EntityT], _R | None]:
"""Wrap a state property of an esphome entity.
@ -139,7 +137,7 @@ def esphome_state_property(
return _wrapper
def convert_api_error_ha_error(
def convert_api_error_ha_error[**_P, _R, _EntityT: EsphomeEntity[Any, Any]](
func: Callable[Concatenate[_EntityT, _P], Awaitable[None]],
) -> Callable[Concatenate[_EntityT, _P], Coroutine[Any, Any, None]]:
"""Decorate ESPHome command calls that send commands/make changes to the device.