reorder

tour-comfy/database.py

@@ -174,6 +174,21 @@ def set_archived(filename, archived: bool):
         cur.close()
         _put_db_connection(conn)
 
+def set_filenames_archived(filenames, archived: bool):
+    if not filenames:
+        return []
+    conn = get_db_connection()
+    cur = conn.cursor()
+    try:
+        cur.execute("UPDATE person SET archived = %s WHERE filename = ANY(%s) RETURNING filename", (archived, list(filenames)))
+        rows = cur.fetchall()
+        updated = [r[0] for r in rows]
+        conn.commit()
+        return updated
+    finally:
+        cur.close()
+        _put_db_connection(conn)
+
 def set_hidden(filename, hidden: bool):
     conn = get_db_connection()
     cur = conn.cursor()