feat(immich): ImmichClient read — search_assets, list_tags, resolve_tag_id, download_thumbnail

2026-06-27 17:11:49 +02:00
parent 01ac95059e
commit 862480916a
3 changed files with 127 additions and 0 deletions
@@ -0,0 +1,50 @@
+from werkzeug.wrappers import Response
+from photoflow.immich import ImmichClient
+
+
+def test_resolve_tag_id_matches_name_or_value(httpserver):
+    httpserver.expect_request("/api/tags").respond_with_json([
+        {"id": "t1", "name": "Italy 2019", "value": "Italy 2019"},
+        {"id": "t2", "name": "processed", "value": "_pipeline/processed"},
+    ])
+    c = ImmichClient(httpserver.url_for(""), "k")
+    assert c.resolve_tag_id("Italy 2019") == "t1"
+    assert c.resolve_tag_id("_pipeline/processed") == "t2"
+    assert c.resolve_tag_id("nope") is None
+
+
+def test_search_assets_normalizes_and_paginates(httpserver):
+    def handler(request):
+        page = request.json.get("page", 1)
+        assert request.json.get("withExif") is True
+        if page == 1:
+            return Response(
+                '{"assets": {"items": [{"id": "a", "originalFileName": "a.jpg",'
+                ' "localDateTime": "2019-06-01T10:00:00.000Z", "type": "IMAGE",'
+                ' "updatedAt": "2026-01-01T00:00:00Z",'
+                ' "exifInfo": {"latitude": 45.4, "longitude": 12.3, "city": "Venezia",'
+                ' "country": "Italy", "rating": 4},'
+                ' "tags": [{"name": "Italy 2019"}]}], "nextPage": 2}}',
+                content_type="application/json")
+        return Response(
+            '{"assets": {"items": [{"id": "b", "originalFileName": "b.jpg",'
+            ' "localDateTime": "2019-06-02T11:00:00.000Z", "type": "IMAGE",'
+            ' "updatedAt": "2026-01-02T00:00:00Z", "exifInfo": {}, "tags": []}],'
+            ' "nextPage": null}}', content_type="application/json")
+
+    httpserver.expect_request("/api/search/metadata", method="POST").respond_with_handler(handler)
+    c = ImmichClient(httpserver.url_for(""), "k")
+    assets = c.search_assets(taken_after="2019-01-01", taken_before="2020-01-01")
+    assert [a["id"] for a in assets] == ["a", "b"]
+    a = assets[0]
+    assert a["gps_lat"] == 45.4 and a["place_city"] == "Venezia"
+    assert a["tags"] == ["Italy 2019"] and a["rating"] == 4
+    assert a["taken_at"] == "2019-06-01T10:00:00.000Z"
+    assert assets[1]["gps_lat"] is None and assets[1]["tags"] == []
+
+
+def test_download_thumbnail(httpserver):
+    httpserver.expect_request("/api/assets/a/thumbnail").respond_with_data(
+        b"\xff\xd8\xffjpegbytes", content_type="image/jpeg")
+    c = ImmichClient(httpserver.url_for(""), "k")
+    assert c.download_thumbnail("a").startswith(b"\xff\xd8\xff")