Odysseus v1.0

scripts/claim_ownerless.py

@@ -0,0 +1,88 @@
+#!/usr/bin/env python3
+"""Claim all ownerless data for a specific user.
+
+Run once after enabling multi-user auth to assign existing data to the admin.
+
+Usage:
+    python scripts/claim_ownerless.py admin@example.com
+"""
+
+import sys
+import os
+import json
+
+sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+
+def main():
+    if len(sys.argv) < 2:
+        print("Usage: python scripts/claim_ownerless.py <username>")
+        sys.exit(1)
+
+    owner = sys.argv[1]
+    print(f"Claiming all ownerless data for: {owner}\n")
+
+    # 1. Memories (JSON files)
+    for label, path in [
+        ("memory.json", "data/memory.json"),
+        ("skills.json", "data/skills.json"),
+    ]:
+        if not os.path.exists(path):
+            print(f"  {label}: not found, skipping")
+            continue
+        with open(path, "r") as f:
+            entries = json.load(f)
+        count = 0
+        for e in entries:
+            if not e.get("owner"):
+                e["owner"] = owner
+                count += 1
+        if count:
+            with open(path, "w") as f:
+                json.dump(entries, f, ensure_ascii=False, indent=2)
+        print(f"  {label}: claimed {count} entries")
+
+    # 2. Database tables (sessions, gallery, comparisons, documents)
+    from core.database import SessionLocal, Session, Document
+    try:
+        from core.database import GalleryImage
+    except ImportError:
+        GalleryImage = None
+    try:
+        from core.database import Comparison
+    except ImportError:
+        Comparison = None
+
+    db = SessionLocal()
+    try:
+        # Sessions
+        count = db.query(Session).filter(Session.owner == None).update({"owner": owner})
+        print(f"  sessions: claimed {count}")
+
+        # Documents
+        count = db.query(Document).filter(Document.session_id.in_(
+            db.query(Session.id).filter(Session.owner == owner)
+        )).update({"session_id": Document.session_id}, synchronize_session=False)
+
+        # Gallery
+        if GalleryImage:
+            count = db.query(GalleryImage).filter(GalleryImage.owner == None).update({"owner": owner})
+            print(f"  gallery: claimed {count}")
+
+        # Comparisons
+        if Comparison:
+            count = db.query(Comparison).filter(Comparison.owner == None).update({"owner": owner})
+            print(f"  comparisons: claimed {count}")
+
+        db.commit()
+    except Exception as e:
+        db.rollback()
+        print(f"  ERROR: {e}")
+    finally:
+        db.close()
+
+    print(f"\nDone! All ownerless data now belongs to {owner}")
+    print("Restart the server: sudo systemctl restart odysseus-ui")
+
+
+if __name__ == "__main__":
+    main()