update mongodb

db/mongodb.py

@@ -162,33 +162,37 @@ def fetch_file_from_mongodb(db_name, collection_name, file_id):
 def detect_file_type(uploaded_file):
     if uploaded_file is not None:
         try:
-            file_bytes = uploaded_file.read(4096)  # Đọc nhiều bytes hơn để nhận diện MIME
+            # Ưu tiên kiểm tra phần mở rộng trước
+            ext = os.path.splitext(uploaded_file.name)[1].lower()
+            ext_mapping = {
+                ".csv": "CSV", ".docx": "Word", ".doc": "Word",
+                ".xlsx": "Excel", ".pptx": "PPTX", ".pdf": "PDF"
+            }
+            detected_type = ext_mapping.get(ext)
+
+            if detected_type:
+                return detected_type  # Nếu có trong danh sách, trả về ngay
+            
+            # Nếu không có phần mở rộng hợp lệ, fallback vào MIME type
+            file_bytes = uploaded_file.read(4096)
             mime = magic.Magic(mime=True)
             file_type = mime.from_buffer(file_bytes)
+
+            mime_types = {
+                "application/pdf": "PDF",
+                "application/vnd.ms-powerpoint": "PPTX",
+                "application/vnd.openxmlformats-officedocument.presentationml.presentation": "PPTX",
+                "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet": "Excel",
+                "application/vnd.ms-excel": "Excel",
+                "application/vnd.openxmlformats-officedocument.wordprocessingml.document": "Word",
+                "application/msword": "Word",
+                "text/csv": "CSV",
+                "text/plain": "CSV"
+            }
+            return mime_types.get(file_type, "Unknown")
+        
         except Exception as e:
             print(f"Error detecting file type: {e}")
-            file_type = "Unknown"
-        
-        # Danh sách MIME types phổ biến
-        mime_types = {
-            "application/pdf": "PDF",
-            "application/vnd.openxmlformats-officedocument.presentationml.presentation": "PPTX",
-            "application/vnd.ms-powerpoint": "PPTX",
-            "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet": "Excel",
-            "application/vnd.ms-excel": "Excel",
-            "application/vnd.openxmlformats-officedocument.wordprocessingml.document": "Word",
-            "application/msword": "Word",
-            "text/csv": "CSV",
-            "text/plain": "CSV"  # Một số file CSV có thể nhận diện là text/plain
-        }
-        
-        detected_type = mime_types.get(file_type, "Unknown")
-        
-        # Nếu vẫn không chắc, kiểm tra phần mở rộng file
-        if detected_type == "Unknown":
-            ext = os.path.splitext(uploaded_file.name)[1].lower()
-            ext_mapping = {".csv": "CSV", ".docx": "Word", ".doc": "Word", ".xlsx": "Excel", ".pptx": "PPTX", ".pdf": "PDF"}
-            detected_type = ext_mapping.get(ext, "Unknown")
-        
-        return detected_type
+            return "Unknown"
+
     return None