39 lines
1018 B
Python
39 lines
1018 B
Python
from __future__ import annotations
|
|
|
|
import logging
|
|
from dataclasses import dataclass
|
|
from typing import Any
|
|
|
|
import requests
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class HaClientSettings:
|
|
url: str
|
|
token: str
|
|
timeout_seconds: int = 10
|
|
|
|
|
|
class HaClient:
|
|
def __init__(self, settings: HaClientSettings) -> None:
|
|
self._settings = settings
|
|
self._session = requests.Session()
|
|
self._session.headers.update({
|
|
"Authorization": f"Bearer {settings.token}",
|
|
"Content-Type": "application/json",
|
|
})
|
|
|
|
def list_entities(self) -> list[dict[str, Any]]:
|
|
response = self._session.get(
|
|
f"{self._settings.url}/api/states",
|
|
timeout=self._settings.timeout_seconds,
|
|
)
|
|
response.raise_for_status()
|
|
payload = response.json()
|
|
if not isinstance(payload, list):
|
|
msg = "Home Assistant states response must be a list."
|
|
raise TypeError(msg)
|
|
return payload
|