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

Type plex session decorator (#70991)

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

17
homeassistant/components/plex/media_player.py
View File

@ -1,11 +1,14 @@
"""Support to interface with the Plex API.""" """Support to interface with the Plex API."""
from __future__ import annotations from __future__ import annotations
from collections.abc import Callable
from functools import wraps from functools import wraps
import logging import logging
from typing import TypeVar
import plexapi.exceptions import plexapi.exceptions
import requests.exceptions import requests.exceptions
from typing_extensions import Concatenate, ParamSpec
from homeassistant.components.media_player import ( from homeassistant.components.media_player import (
DOMAIN as MP_DOMAIN, DOMAIN as MP_DOMAIN,
@ -43,17 +46,25 @@ from .const import (
from .media_browser import browse_media from .media_browser import browse_media
from .services import process_plex_payload from .services import process_plex_payload
_PlexMediaPlayerT = TypeVar("_PlexMediaPlayerT", bound="PlexMediaPlayer")
_R = TypeVar("_R")
_P = ParamSpec("_P")
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
def needs_session(func): def needs_session(
func: Callable[Concatenate[_PlexMediaPlayerT, _P], _R]
) -> Callable[Concatenate[_PlexMediaPlayerT, _P], _R | None]:
"""Ensure session is available for certain attributes.""" """Ensure session is available for certain attributes."""
@wraps(func) @wraps(func)
def get_session_attribute(self, *args): def get_session_attribute(
self: _PlexMediaPlayerT, *args: _P.args, **kwargs: _P.kwargs
) -> _R | None:
if self.session is None: if self.session is None:
return None return None
return func(self, *args) return func(self, *args, **kwargs)
return get_session_attribute return get_session_attribute