32 lines
1014 B
Python
32 lines
1014 B
Python
|
"""Decorators for the Home Assistant API."""
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from collections.abc import Awaitable, Callable
|
||
|
|
from typing import Concatenate, ParamSpec, TypeVar
|
||
|
|
|
||
|
|
from aiohttp.web import Request, Response
|
||
|
|
|
||
|
|
from homeassistant.exceptions import Unauthorized
|
||
|
|
|
||
|
|
from .view import HomeAssistantView
|
||
|
|
|
||
|
|
_HomeAssistantViewT = TypeVar("_HomeAssistantViewT", bound=HomeAssistantView)
|
||
|
|
_P = ParamSpec("_P")
|
||
|
|
|
||
|
|
|
||
|
|
def require_admin(
|
||
|
|
func: Callable[Concatenate[_HomeAssistantViewT, Request, _P], Awaitable[Response]]
|
||
|
|
) -> Callable[Concatenate[_HomeAssistantViewT, Request, _P], Awaitable[Response]]:
|
||
|
|
"""Home Assistant API decorator to require user to be an admin."""
|
||
|
|
|
||
|
|
async def with_admin(
|
||
|
|
self: _HomeAssistantViewT, request: Request, *args: _P.args, **kwargs: _P.kwargs
|
||
|
|
) -> Response:
|
||
|
|
"""Check admin and call function."""
|
||
|
|
if not request["hass_user"].is_admin:
|
||
|
|
raise Unauthorized()
|
||
|
|
|
||
|
|
return await func(self, request, *args, **kwargs)
|
||
|
|
|
||
|
|
return with_admin
|