Đang Code

.gitignore

@@ -0,0 +1,2 @@
+__pycache__/
+*.pyc

backend/app/api/routes/alerts.py

@@ -1,48 +1,52 @@
-"""
-Alert routes - now using async MongoDB operations
-"""
 from fastapi import APIRouter, Query
 from app.services.alert_service import alert_service
+from fastapi import APIRouter, Depends, Query, Response
+from typing import Optional, List
+from datetime import datetime
+import io, csv
+from app.services.alert_service import AlertService
+from app.db import get_database
 
 router = APIRouter(tags=["Alerts"])
 
+# Hàm service alert
+def get_alert_service() -> AlertService:
+    db = get_database()
+    return alert_service
+
+# Hàm hiển thị tất cả danh sách alert
 @router.get("/alerts")
 async def get_alerts(
     severity: str = Query(None),
-    limit: int = Query(10, ge=1, le=10000),  # Increased from 100 to 10000 for pagination
+    limit: int = Query(10, ge=1, le=10000), 
     offset: int = Query(0, ge=0)
 ):
-    """Get all alerts with optional severity filter"""
     if severity:
         return await alert_service.get_alerts_by_severity(severity, limit)
     return await alert_service.get_all_alerts(limit, offset)
 
 @router.get("/alerts/stats")
 async def get_alert_stats():
-    """Get alert statistics"""
     return await alert_service.get_alerts_statistics()
 
 @router.get("/alerts/{alert_id}")
 async def get_alert(alert_id: str):
-    """Get specific alert by ID"""
     alert = await alert_service.get_alert_by_id(alert_id)
     if not alert:
         return {"error": "Alert not found"}
     return alert
 
 @router.post("/alerts/generate")
 async def generate_alert():
-    """Generate random alert for testing"""
     alert = alert_service.generate_random_alert()
     saved_alert = await alert_service.add_alert(alert)
     return saved_alert
 
 @router.post("/alerts")
 async def create_alert(alert_data: dict):
-    """Create new alert"""
     return await alert_service.add_alert(alert_data)
 
 @router.patch("/alerts/{alert_id}")
 async def update_alert(alert_id: str, update_data: dict):
-    """Update alert (e.g., acknowledge, change status)"""
     return await alert_service.update_alert(alert_id, update_data)
+

backend/app/db.py

@@ -14,29 +14,19 @@ class Database:
 
     @classmethod
     async def connect_db(cls):
-        """Connect to MongoDB"""
         mongodb_uri = os.getenv("MONGODB_URI")
         db_name = os.getenv("DB_NAME", "ids_ips_db")
 
         if not mongodb_uri:
             raise ValueError("MONGODB_URI not found in environment variables")
-
-        # Replace <db_password> with actual password
-        # For now, we assume the URI is complete or password is set in env
         cls.client = AsyncIOMotorClient(mongodb_uri)
-
-        # Test connection
         try:
             await cls.client.admin.command('ping')
             print(f"[OK] Connected to MongoDB: {db_name}")
         except Exception as e:
             print(f"[FAIL] Failed to connect to MongoDB: {e}")
             raise
-
-        # Get database
         cls.db = cls.client[db_name]
-
-        # Setup indexes
         await cls.setup_indexes()
 
     @classmethod
@@ -73,11 +63,8 @@ async def setup_indexes(cls):
             print("[OK] Database indexes created successfully")
         except Exception as e:
             print(f"[WARN] Could not create indexes: {e}")
-
-# Create global database instance
 db = Database()
 
 def get_database():
-    """Get database instance"""
     return db.db