jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-19 11:17:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improve roku error decorator typing (#70992)

Browse Source
This commit is contained in:
Marc Mueller 2022-04-28 21:05:22 +02:00 committed by GitHub
parent cf90e34776
commit 7649b5e6c9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
homeassistant/components/roku/helpers.py
View File

@ -13,7 +13,7 @@ from .entity import RokuEntity
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
_T = TypeVar("_T", bound=RokuEntity) _RokuEntityT = TypeVar("_RokuEntityT", bound=RokuEntity)
_P = ParamSpec("_P") _P = ParamSpec("_P")
@ -25,14 +25,21 @@ def format_channel_name(channel_number: str, channel_name: str | None = None) ->
return channel_number return channel_number
def roku_exception_handler(ignore_timeout: bool = False) -> Callable[..., Callable]: def roku_exception_handler(
ignore_timeout: bool = False,
) -> Callable[
[Callable[Concatenate[_RokuEntityT, _P], Awaitable[Any]]],
Callable[Concatenate[_RokuEntityT, _P], Coroutine[Any, Any, None]],
]:
"""Decorate Roku calls to handle Roku exceptions.""" """Decorate Roku calls to handle Roku exceptions."""
def decorator( def decorator(
func: Callable[Concatenate[_T, _P], Awaitable[None]], func: Callable[Concatenate[_RokuEntityT, _P], Awaitable[Any]],
) -> Callable[Concatenate[_T, _P], Coroutine[Any, Any, None]]: ) -> Callable[Concatenate[_RokuEntityT, _P], Coroutine[Any, Any, None]]:
@wraps(func) @wraps(func)
async def wrapper(self: _T, *args: _P.args, **kwargs: _P.kwargs) -> None: async def wrapper(
self: _RokuEntityT, *args: _P.args, **kwargs: _P.kwargs
) -> None:
try: try:
await func(self, *args, **kwargs) await func(self, *args, **kwargs)
except RokuConnectionTimeoutError as error: except RokuConnectionTimeoutError as error: