Uploaded code

cleanup_expired_db.py

@@ -0,0 +1,88 @@
+#!/usr/bin/env python3
+"""
+Cleanup script for expired files using expiry database
+Works with both local storage and Backblaze B2 storage backends
+"""
+import os
+import sys
+from datetime import datetime
+from pathlib import Path
+
+# Add the project root to Python path
+project_root = Path(__file__).parent
+sys.path.append(str(project_root))
+
+from src.config import config
+from src.storage import StorageManager
+from expiry_db import ExpiryDatabase
+
+def main():
+    """Main cleanup function"""
+    print("🧹 Sharey Cleanup Script (Database-driven)")
+    print("=" * 50)
+    print(f"⏰ Started at: {datetime.utcnow().isoformat()}")
+    
+    try:
+        # Load configuration
+        print("✅ Loaded configuration from config.json")
+        
+        # Initialize storage manager
+        storage_manager = StorageManager(config)
+        backend_info = storage_manager.get_backend_info()
+        print(f"📁 Storage backend: {backend_info['type']}")
+        
+        # Initialize expiry database
+        expiry_db = ExpiryDatabase()
+        
+        # Get expired files from database
+        expired_files = expiry_db.get_expired_files()
+        
+        if not expired_files:
+            print("✅ No expired files found.")
+            return 0
+        
+        print(f"🗑️ Found {len(expired_files)} expired files to delete:")
+        
+        deleted_count = 0
+        for file_info in expired_files:
+            file_path = file_info['file_path']
+            expires_at = file_info['expires_at']
+            storage_backend = file_info['storage_backend']
+            
+            try:
+                print(f"🗑️ Deleting expired file: {file_path} (expired: {expires_at})")
+                
+                # Delete from storage backend
+                if storage_manager.delete_file(file_path):
+                    # Remove from expiry database
+                    expiry_db.remove_file(file_path)
+                    deleted_count += 1
+                    print(f"✅ Deleted: {file_path}")
+                else:
+                    print(f"❌ Failed to delete from storage: {file_path}")
+                    
+            except Exception as e:
+                print(f"❌ Error deleting {file_path}: {e}")
+        
+        print("=" * 50)
+        print(f"✅ Cleanup completed! Deleted {deleted_count} expired files.")
+        
+        # Show database stats
+        stats = expiry_db.get_stats()
+        print(f"📊 Database stats:")
+        print(f"   • Total tracked files: {stats.get('total_files', 0)}")
+        print(f"   • Active files: {stats.get('active_files', 0)}")
+        print(f"   • Expired files: {stats.get('expired_files', 0)}")
+        
+        return deleted_count
+        
+    except Exception as e:
+        print(f"❌ Cleanup failed: {e}")
+        return -1
+    
+    finally:
+        print(f"⏰ Finished at: {datetime.utcnow().isoformat()}")
+
+if __name__ == "__main__":
+    result = main()
+    sys.exit(0 if result >= 0 else 1)