data

index_immich.py

@@ -6,6 +6,7 @@ import tempfile
 
 from image_process import describe_simba_image
 from request import PaperlessNGXService
+import sqlite3
 
 logging.basicConfig(level=logging.INFO)
 
@@ -23,8 +24,16 @@ DOWNLOAD_DIR = os.getenv("DOWNLOAD_DIR", "./simba_photos")
 # Set up headers
 headers = {"x-api-key": API_KEY, "Content-Type": "application/json"}
 
+VISITED = {}
 
 if __name__ == "__main__":
+    conn = sqlite3.connect("./visited.db")
+    c = conn.cursor()
+    c.execute("select immich_id from visited")
+    rows = c.fetchall()
+    for row in rows:
+        VISITED.add(row[0])
+
     ppngx = PaperlessNGXService()
     people_url = f"{IMMICH_URL}/api/search/person?name=Simba"
     people = httpx.get(people_url, headers=headers).json()
@@ -39,7 +48,7 @@ if __name__ == "__main__":
 
     assets = results.json()["assets"]
     for asset in assets["items"]:
-        if asset["type"] == "IMAGE":
+        if asset["type"] == "IMAGE" and asset["id"] not in VISITED:
             ids[asset["id"]] = asset.get("originalFileName")
     nextPage = assets.get("nextPage")
 
@@ -58,41 +67,49 @@ if __name__ == "__main__":
     asset_search = f"{IMMICH_URL}/api/search/smart"
     request_body = {"query": "simba"}
     results = httpx.post(asset_search, headers=headers, json=request_body)
-    print(results.json()["assets"]["total"])
     for asset in results.json()["assets"]["items"]:
         if asset["type"] == "IMAGE":
             ids[asset["id"]] = asset.get("originalFileName")
 
-    immich_asset_id = list(ids.keys())[1]
-    immich_filename = ids.get(immich_asset_id)
-    response = httpx.get(
-        f"{IMMICH_URL}/api/assets/{immich_asset_id}/original", headers=headers
-    )
+    for immich_asset_id, immich_filename in ids.items():
+        try:
+            response = httpx.get(
+                f"{IMMICH_URL}/api/assets/{immich_asset_id}/original", headers=headers
+            )
 
-    path = os.path.join("/Users/ryanchen/Programs/raggr", immich_filename)
-    file = open(path, "wb+")
-    for chunk in response.iter_bytes(chunk_size=8192):
-        file.write(chunk)
+            path = os.path.join("/Users/ryanchen/Programs/raggr", immich_filename)
+            file = open(path, "wb+")
+            for chunk in response.iter_bytes(chunk_size=8192):
+                file.write(chunk)
 
-    logging.info("Processing image ...")
-    description = describe_simba_image(path)
+            logging.info("Processing image ...")
+            description = describe_simba_image(path)
 
-    image_description = description.description
-    image_date = description.image_date
+            image_description = description.description
+            image_date = description.image_date
 
-    description_filepath = os.path.join("/Users/ryanchen/Programs/raggr", f"SIMBA_DESCRIBE_001.txt")
-    file = open(description_filepath, "w+")
-    file.write(image_description)
-    file.close()
+            description_filepath = os.path.join(
+                "/Users/ryanchen/Programs/raggr", f"SIMBA_DESCRIBE_001.txt"
+            )
+            file = open(description_filepath, "w+")
+            file.write(image_description)
+            file.close()
 
-    file = open(description_filepath, 'rb')
+            file = open(description_filepath, "rb")
+            ppngx.upload_description(
+                description_filepath=description_filepath,
+                file=file,
+                title="SIMBA_DESCRIBE_001.txt",
+                exif_date=image_date,
+            )
+            file.close()
 
-    ppngx.upload_description(description_filepath=description_filepath, file=file, title="SIMBA_DESCRIBE_001.txt", exif_date=image_date)
-    
+            c.execute("INSERT INTO visited (immich_id) values (?)", (immich_asset_id,))
+            conn.commit()
+            logging.info("Processing complete. Deleting file.")
+            os.remove(file.name)
+        except Exception as e:
+            logging.info(f"something went wrong for {immich_filename}")
+            logging.info(e)
 
-    file.close()
-
-    
-
-    logging.info("Processing complete. Deleting file.")
-    os.remove(file.name)
+    conn.close()