首次提交by MimoCode

backend/app/api/v1/notifications.py

@@ -0,0 +1,77 @@
+from typing import Optional
+from fastapi import APIRouter, Depends, HTTPException, Query
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from app.database import get_db
+from app.core.deps import get_current_user
+from app.models.user import User
+from app.services.notification import NotificationService
+
+router = APIRouter()
+
+
+@router.get("/")
+async def list_notifications(
+    is_read: Optional[bool] = Query(None),
+    type: Optional[str] = Query(None),
+    page: int = Query(1, ge=1),
+    page_size: int = Query(20, ge=1, le=100),
+    db: AsyncSession = Depends(get_db),
+    current_user: User = Depends(get_current_user)
+):
+    service = NotificationService(db)
+    notifications = await service.get_notifications(
+        user_id=current_user.id,
+        is_read=is_read,
+        type=type,
+        page=page,
+        page_size=page_size
+    )
+    return [
+        {
+            "id": n.id,
+            "type": n.type,
+            "title": n.title,
+            "content": n.content,
+            "is_read": n.is_read,
+            "related_id": n.related_id,
+            "created_at": n.created_at
+        }
+        for n in notifications
+    ]
+
+
+@router.put("/{notification_id}/read")
+async def mark_notification_read(
+    notification_id: int,
+    db: AsyncSession = Depends(get_db),
+    current_user: User = Depends(get_current_user)
+):
+    service = NotificationService(db)
+    success = await service.mark_as_read(notification_id)
+    if not success:
+        raise HTTPException(status_code=404, detail="通知不存在")
+    return {"message": "success"}
+
+
+@router.put("/read-all")
+async def mark_all_notifications_read(
+    db: AsyncSession = Depends(get_db),
+    current_user: User = Depends(get_current_user)
+):
+    service = NotificationService(db)
+    count = await service.mark_all_as_read(current_user.id)
+    return {"message": "success", "count": count}
+
+
+@router.delete("/{notification_id}")
+async def delete_notification(
+    notification_id: int,
+    db: AsyncSession = Depends(get_db),
+    current_user: User = Depends(get_current_user)
+):
+    service = NotificationService(db)
+    success = await service.delete_notification(notification_id)
+    if not success:
+        raise HTTPException(status_code=404, detail="通知不存在")
+    return {"message": "删除成功"}