From 57c1eb26056694b7ca1b25db256708fae337044f Mon Sep 17 00:00:00 2001
From: Franziska Kunsmann <hi@kunsmann.eu>
Date: Tue, 6 May 2025 18:32:20 +0200
Subject: [PATCH] bundles/docker-immich: database not existing should not error
 out the script

after all, we have monitoring to ensure the database container runs
---
 .../files/immich-auto-album-share.py          | 32 +++++++++++--------
 1 file changed, 19 insertions(+), 13 deletions(-)

diff --git a/bundles/docker-immich/files/immich-auto-album-share.py b/bundles/docker-immich/files/immich-auto-album-share.py
index 863f8b2..2cac6c2 100644
--- a/bundles/docker-immich/files/immich-auto-album-share.py
+++ b/bundles/docker-immich/files/immich-auto-album-share.py
@@ -1,5 +1,6 @@
 #!/usr/bin/env python3
 
+import logging
 from json import loads
 from os import environ
 from subprocess import check_output
@@ -12,6 +13,8 @@ PSQL_USER = environ['DB_USERNAME']
 PSQL_PASS = environ['DB_PASSWORD']
 PSQL_DB = environ['DB_DATABASE_NAME']
 
+logging.basicConfig(level=logging.INFO)
+
 docker_networks = loads(check_output(['docker', 'network', 'inspect', 'aaarghhh']))
 
 container_ip = None
@@ -26,9 +29,9 @@ for network in docker_networks:
             container_ip = container['IPv4Address'].split('/')[0]
 
 if not container_ip:
-    print(f'could not find ip address for container {PSQL_HOST=} in json')
-    print(docker_networks)
-    exit(1)
+    logging.error(f'could not find ip address for container {PSQL_HOST=} in json')
+    logging.debug(f'{docker_networks=}')
+    exit(0)
 
 print(f'{PSQL_HOST=} {container_ip=}')
 
@@ -49,6 +52,7 @@ with conn:
             }
             for i in cur.fetchall()
         }
+    logging.debug(f'{albums=}')
 
     with conn.cursor() as cur:
         cur.execute('SELECT "id","name" FROM users;')
@@ -56,25 +60,27 @@ with conn:
             i[0]: i[1]
             for i in cur.fetchall()
         }
+    logging.debug(f'{users=}')
 
 for album_id, album in albums.items():
-    print(f'----- working on album: {album["name"]}')
+    log = logging.getLogger(album["name"])
     with conn:
         with conn.cursor() as cur:
             cur.execute('SELECT "usersId" FROM albums_shared_users_users WHERE "albumsId" = %s;', (album_id,))
             album_shares = [i[0] for i in cur.fetchall()]
-        print(f' album is shared with {len(album_shares)} users: {album_shares}')
+        log.info(f'album is shared with {len(album_shares)} users: {album_shares}')
         for user_id, user_name in users.items():
             if user_id == album['owner'] or user_id in album_shares:
                 continue
 
-            print(f' sharing album with user {user_name} ... ', end='')
-            with conn.cursor() as cur:
-                cur.execute(
-                    'INSERT INTO albums_shared_users_users ("albumsId","usersId","role") VALUES (%s, %s, %s);',
-                    (album_id, user_id, 'viewer'),
-                )
-            print('done')
-    print()
+            log.info(f'sharing album with user {user_name}')
+            try:
+                with conn.cursor() as cur:
+                    cur.execute(
+                        'INSERT INTO albums_shared_users_users ("albumsId","usersId","role") VALUES (%s, %s, %s);',
+                        (album_id, user_id, 'viewer'),
+                    )
+            except Exception:
+                log.exception('failure while creating share')
 
 conn.close()