feat: initial commit

custom_components/two_gis_weather/api.py

@@ -0,0 +1,67 @@
+from __future__ import annotations
+
+import uuid
+from typing import Any
+
+import async_timeout
+from aiohttp import ClientResponse
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.aiohttp_client import async_get_clientsession
+
+from .const import API_BASE, DEFAULT_USER_AGENT, REQUEST_TIMEOUT
+
+
+class TwoGisWeatherClient:
+    def __init__(
+        self,
+        hass: HomeAssistant,
+        api_key: str,
+        latitude: float,
+        longitude: float,
+        language: str = "en-US",
+        user_agent: str = DEFAULT_USER_AGENT,
+    ) -> None:
+        self.hass = hass
+        self.api_key = api_key
+        self.latitude = latitude
+        self.longitude = longitude
+        self.language = language
+        self.user_agent = user_agent
+        self._last_modified: str | None = None
+
+    def _headers(self) -> dict[str, str]:
+        headers: dict[str, str] = {
+            "User-Agent": self.user_agent,
+            "x-request-id": uuid.uuid4().hex,
+            "Accept-Language": self.language,
+            "Accept": "application/json",
+        }
+        if self._last_modified:
+            headers["If-Modified-Since"] = self._last_modified
+        return headers
+
+    async def async_get_current(self) -> dict[str, Any] | None:
+        session = async_get_clientsession(self.hass)
+        url = f"{API_BASE}/forecast"
+        params = {
+            "lon": str(self.longitude),
+            "lat": str(self.latitude),
+            "key": self.api_key,
+        }
+        async with async_timeout.timeout(REQUEST_TIMEOUT):
+            resp: ClientResponse
+            async with session.get(url, params=params, headers=self._headers()) as resp:
+                if resp.status == 304:
+                    return None
+                resp.raise_for_status()
+                # Track Last-Modified for conditional requests
+                last_mod = resp.headers.get("Last-Modified")
+                if last_mod:
+                    self._last_modified = last_mod
+                return await resp.json(content_type=None)
+
+    @property
+    def last_modified(self) -> str | None:
+        return self._last_modified
+
+