1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
|
import requests
from io import BytesIO
from texture import ImageTexture
class ImmichConnector:
def __init__(self, server_url, api_key):
self.server_url = server_url.removesuffix("/")
self.api_key = api_key
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
data = response.json()
return data["assets"]
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_texture(pd, asset["id"], asset["exifInfo"])
|
