add auto delete all

db/mongodb.py

@@ -13,33 +13,39 @@ def connect_mongodb(db_name, collection_name):
 
 
 
-def save_file_to_mongodb(uploaded_file, db_name="ppt", collection_name="root_file"):
-    """
-    Lưu file vào MongoDB bằng GridFS mà không kiểm tra trùng lặp.
-    
-    :param uploaded_file: đối tượng UploadedFile từ Streamlit
-    :param db_name: Tên database trong MongoDB
-    :param collection_name: Tên collection GridFS
-    :param file_tail: Phần mở rộng mặc định của file nếu không có
-    :return: file_id nếu lưu thành công
-    """
+import gridfs
+from pymongo import MongoClient
+import pprint
+
+def save_file_to_mongodb(uploaded_file, db_name="pptx", collection_name="root_file", max_db_size_mb=500):
     client = MongoClient("mongodb+srv://admin:[email protected]/?retryWrites=true&w=majority&appName=Cluster0")
     db = client[db_name]
     fs = gridfs.GridFS(db, collection=collection_name)
 
-    # Xác định tên file
     file_name = uploaded_file.name
-
-    # Đảm bảo con trỏ file đang ở đầu
     uploaded_file.seek(0)
     file_bytes = uploaded_file.read()
 
-    # Lưu file vào MongoDB (không kiểm tra trùng lặp)
-    file_id = fs.put(file_bytes, filename=file_name)
-    print(f"✅ File '{file_name}' đã được lưu vào '{collection_name}' với ID: {file_id}")
-    
-    client.close()
-    return file_id, file_name
+    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)
+
+        print(f"📦 Database size: {db_size_mb:.2f} MB")
+        if db_size_mb > max_db_size_mb:
+            delete_all()
+
+        # Sau khi dọn dẹp (nếu cần), tiến hành lưu
+        file_id = fs.put(file_bytes, filename=file_name)
+        print(f"✅ File '{file_name}' đã được lưu vào '{collection_name}' với ID: {file_id}")
+        return file_id, file_name
+
+    except Exception as e:
+        print(f"❌ Lỗi khi lưu file hoặc truy vấn MongoDB: {e}")
+        return None, None
+
+    finally:
+        client.close()
 
 
 
@@ -116,4 +122,25 @@ def detect_file_type(uploaded_file):
             print(f"Error detecting file type: {e}")
             return "Unknown"
 
-    return None
+    return None
+
+
+def delete_all_files_in_collection(collection_name, db_name="ppt"):
+    try:
+        client = MongoClient("mongodb+srv://admin:[email protected]/?retryWrites=true&w=majority&appName=Cluster0")
+        db = client[db_name]
+        fs = gridfs.GridFS(db, collection=collection_name)
+        file_ids = [file["_id"] for file in db[f"{collection_name}.files"].find({})]
+
+        for file_id in file_ids:
+            fs.delete(file_id)
+        print(f"✅ Đã xóa {len(file_ids)} file trong collection '{collection_name}'")
+    except Exception as e:
+        print(f"❌ Lỗi khi xóa file: {str(e)}")
+    finally:
+        client.close()
+
+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)