albums thumbnails endpoint and albums endpoint

1 files changed, 16 insertions, 4 deletions

diff --git a/immich.py b/immich.py
index 1f1ccb8..63e6f63 100644
--- a/immich.py
+++ b/immich.py
@@ -12,6 +12,13 @@ class ImmichConnector:
     def _request(self, endpoint):
         return requests.get(f"{self.server_url}/api/{endpoint}", headers={ "x-api-key": self.api_key })
 
+    def load_all_albums(self):
+        response = self._request("albums")
+        if response.status_code != 200: return
+
+        data = response.json()
+        return data
+
     def load_album_assets(self, key):
         response = self._request(f"albums/{key}")
         if response.status_code != 200: return
@@ -19,15 +26,20 @@ class ImmichConnector:
         data = response.json()
         return data["assets"]
 
-    def load_image(self, pd, key, exif=None):
-        response = self._request(f"assets/{key}/thumbnail?size=fullsize")
-        if response.status_code != 200: return
+    def load_image(self, key, size="preview"):
+        response = self._request(f"assets/{key}/thumbnail?size={size}")
+        if response.status_code != 200: return None, None
 
         image_data = BytesIO(response.content)
+        mimetype = response.headers.get("Content-Type")
+        return image_data, mimetype
+
+    def load_texture(self, pd, key, exif=None, size="preview"):
+        image_data, _ = self.load_image(key, size)
         it = ImageTexture(image_data, exif)
         pd.textures.append(it)
         print(f"Loaded image {key}")
 
     def idle(self, pd):
         for asset in self.load_album_assets("bac029a5-972b-4519-bce0-a0d74add3969"):
-            self.load_image(pd, asset["id"], asset["exifInfo"])
+            self.load_texture(pd, asset["id"], asset["exifInfo"])