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

Type heos error decorator (#70978)

Browse Source
This commit is contained in:
Marc Mueller 2022-04-28 20:53:39 +02:00 committed by GitHub
parent 319ae8b0b7
commit 64fc93ba5f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
homeassistant/components/heos/media_player.py
View File

@ -1,11 +1,14 @@
"""Denon HEOS Media Player."""
from __future__ import annotations
from collections.abc import Awaitable, Callable, Coroutine
from functools import reduce, wraps
import logging
from operator import ior
from typing import Any
from pyheos import HeosError, const as heos_const
from typing_extensions import ParamSpec
from homeassistant.components import media_source
from homeassistant.components.media_player import (
@ -42,6 +45,8 @@ from .const import (
SIGNAL_HEOS_UPDATED,
)
_P = ParamSpec("_P")
BASE_SUPPORTED_FEATURES = (
MediaPlayerEntityFeature.VOLUME_MUTE
| MediaPlayerEntityFeature.VOLUME_SET
@ -80,12 +85,16 @@ async def async_setup_entry(
async_add_entities(devices, True)
def log_command_error(command: str):
def log_command_error(
command: str,
) -> Callable[[Callable[_P, Awaitable[Any]]], Callable[_P, Coroutine[Any, Any, None]]]:
"""Return decorator that logs command failure."""
def decorator(func):
def decorator(
func: Callable[_P, Awaitable[Any]]
) -> Callable[_P, Coroutine[Any, Any, None]]:
@wraps(func)
async def wrapper(*args, **kwargs):
async def wrapper(*args: _P.args, **kwargs: _P.kwargs) -> None:
try:
await func(*args, **kwargs)
except (HeosError, ValueError) as ex: