add fix auto delete

db/mongodb.py

@@ -28,8 +28,7 @@ def save_file_to_mongodb(uploaded_file, db_name="pptx", collection_name="root_fi
 
     try:
         # Kiểm tra kích thước dữ liệu hiện tại trong DB (bytes → MB)
-        stats = db.command("dbstats")
-        db_size_mb = stats.get("storageSize", 0) / (1024 * 1024)
+        db_size_mb = get_total_cluster_size(client)
 
         print(f"📦 Database size: {db_size_mb:.2f} MB")
         if db_size_mb > max_db_size_mb:
@@ -134,7 +133,7 @@ def delete_all_files_in_collection(collection_name, db_name="ppt"):
 
         for file_id in file_ids:
             fs.delete(file_id)
-        print(f"✅ Đã xóa {len(file_ids)} file trong collection '{collection_name}'")
+        print(f"✅ Đã xóa {len(file_ids)} file trong collection '{collection_name}' của db '{db_name}'")
     except Exception as e:
         print(f"❌ Lỗi khi xóa file: {str(e)}")
     finally:
@@ -143,4 +142,18 @@ def delete_all_files_in_collection(collection_name, db_name="ppt"):
 def delete_all():
     for i in ['root_file', 'final_file']:
         for j in ['word', 'exce', 'pptx', 'csv']:
-            delete_all_files_in_collection(i, db_name=j)
+            delete_all_files_in_collection(i, db_name=j)
+
+
+def get_total_cluster_size(client):
+    total_size = 0
+    try: 
+        for db_name in ['word', 'exce', 'pptx', 'csv']:
+            db = client[db_name]
+            stats = db.command("dbstats")
+            db_size = stats.get("dataSize", 0)
+            total_size += db_size
+    except Exception as e:
+        print(f"❌ Lỗi khi tính dung lượng cluster: {e}")
+        return -1
+    return total_size / (1024 ** 2)