feat: add ImmichClient

2026-04-06 07:41:53 +00:00
parent 4967f0e699
commit daccfeb46f
1 changed files with 45 additions and 0 deletions
@@ -0,0 +1,45 @@
+from __future__ import annotations
+
+import logging
+
+from app.clients.base import BaseHTTPClient
+from app.config import Settings
+from app.models.sources import ImmichSnapshot
+
+
+logger = logging.getLogger(__name__)
+
+
+class ImmichClient(BaseHTTPClient):
+    def __init__(self, settings: Settings) -> None:
+        super().__init__(settings, "immich", settings.immich_base_url)
+
+    async def fetch_stats(self) -> ImmichSnapshot:
+        snapshot = ImmichSnapshot()
+        if not self.base_url or not self.settings.immich_api_key:
+            logger.info("immich skipped: base URL or API key missing")
+            return snapshot
+
+        headers = {"x-api-key": self.settings.immich_api_key}
+
+        try:
+            data = await self._request_json("GET", "/api/server/statistics", headers=headers)
+        except Exception as exc:
+            logger.warning("immich fetch_stats failed: %s", exc)
+            return snapshot
+
+        if not isinstance(data, dict):
+            logger.warning("immich unexpected response type: %s", type(data))
+            return snapshot
+
+        photos = int(data.get("photos", 0))
+        videos = int(data.get("videos", 0))
+        usage_bytes = int(data.get("usage", 0))
+        usage_gb = round(usage_bytes / (1024 ** 3), 1)
+
+        return ImmichSnapshot(
+            source_status="online",
+            photos=photos,
+            videos=videos,
+            storage_gb=usage_gb,
+        )