Retries and simplification

src/lib/homeassistant.py

@@ -1,25 +1,37 @@
-#!/usr/bin/env python3
-import json
-from urllib.request import Request, urlopen
-
-
-class HomeAssistantClient:
-    def __init__(self, base_url: str, token: str):
-        self.base_url = base_url.rstrip("/")
-        self.token = token
-
-    def get_state(self, entity_id: str) -> dict:
-        url = f"{self.base_url}/api/states/{entity_id}"
-        req = Request(url, headers={
-            "Authorization": f"Bearer {self.token}",
-            "Content-Type": "application/json",
-        })
-        with urlopen(req, timeout=30) as resp:
-            return json.loads(resp.read().decode())
-
-    def is_person_home(self, entity_id: str) -> bool:
-        try:
-            return self.get_state(entity_id).get("state") == "home"
-        except Exception as e:
-            print(f"Failed to check {entity_id}: {e}")
-            return False
+#!/usr/bin/env python3
+import json
+import time
+from urllib.error import URLError
+from urllib.request import Request, urlopen
+
+RETRY_DELAYS = (3, 10)
+
+
+class HomeAssistantClient:
+    def __init__(self, base_url: str, token: str):
+        self.base_url = base_url.rstrip("/")
+        self.token = token
+
+    def get_state(self, entity_id: str) -> dict:
+        url = f"{self.base_url}/api/states/{entity_id}"
+        req = Request(url, headers={
+            "Authorization": f"Bearer {self.token}",
+            "Content-Type": "application/json",
+        })
+        last_err: Exception | None = None
+        for attempt in range(len(RETRY_DELAYS) + 1):
+            try:
+                with urlopen(req, timeout=30) as resp:
+                    return json.loads(resp.read().decode())
+            except (URLError, TimeoutError) as e:
+                last_err = e
+                if attempt < len(RETRY_DELAYS):
+                    time.sleep(RETRY_DELAYS[attempt])
+        raise last_err
+
+    def is_person_home(self, entity_id: str) -> bool:
+        try:
+            return self.get_state(entity_id).get("state") == "home"
+        except Exception as e:
+            print(f"Failed to check {entity_id}: {e}")
+            return False