""" 消息通知管理API路由 """ from typing import Optional, List, Dict, Any from datetime import datetime from fastapi import APIRouter, Depends, HTTPException, status, Query from sqlalchemy.ext.asyncio import AsyncSession from sqlalchemy import and_ from pydantic import BaseModel, Field from app.core.deps import get_db, get_current_user from app.schemas.notification import ( NotificationCreate, NotificationUpdate, NotificationResponse, NotificationQueryParams, NotificationBatchCreate, NotificationBatchUpdate, NotificationStatistics, NotificationSendFromTemplate ) from app.services.notification_service import notification_service router = APIRouter() class NotificationIdsPayload(BaseModel): ids: List[int] = Field(..., min_items=1, description="通知ID列表") @router.get("/", response_model=Dict[str, Any]) async def get_notifications( skip: int = Query(0, ge=0, description="跳过条数"), limit: int = Query(20, ge=1, le=100, description="返回条数"), notification_type: Optional[str] = Query(None, description="通知类型"), priority: Optional[str] = Query(None, description="优先级"), is_read: Optional[bool] = Query(None, description="是否已读"), start_time: Optional[datetime] = Query(None, description="开始时间"), end_time: Optional[datetime] = Query(None, description="结束时间"), keyword: Optional[str] = Query(None, description="关键词"), db: AsyncSession = Depends(get_db), current_user = Depends(get_current_user) ): """ 获取消息通知列表 - **skip**: 跳过条数 - **limit**: 返回条数（最大100） - **notification_type**: 通知类型筛选 - **priority**: 优先级筛选 - **is_read**: 是否已读筛选 - **start_time**: 开始时间筛选 - **end_time**: 结束时间筛选 - **keyword**: 关键词搜索 注意：普通用户只能查看自己的通知，管理员可以查看所有通知 """ recipient_id = None if current_user.is_superuser else current_user.id data = await notification_service.get_notifications( db, skip=skip, limit=limit, recipient_id=recipient_id, notification_type=notification_type, priority=priority, is_read=is_read, start_time=start_time, end_time=end_time, keyword=keyword ) # Append unread count for current user unread = await notification_service.get_unread_count(db, current_user.id) data["unread_count"] = unread.get("unread_count", 0) return data @router.get("/unread-count", response_model=Dict[str, Any]) async def get_unread_count( db: AsyncSession = Depends(get_db), current_user = Depends(get_current_user) ): """ 获取当前用户未读通知数量 返回未读通知数量 """ return await notification_service.get_unread_count(db, current_user.id) @router.get("/statistics", response_model=Dict[str, Any]) async def get_notification_statistics( db: AsyncSession = Depends(get_db), current_user = Depends(get_current_user) ): """ 获取当前用户通知统计信息 返回通知总数、未读数、已读数、高优先级数、紧急通知数、类型分布等统计信息 """ return await notification_service.get_statistics(db, current_user.id) @router.get("/{notification_id}", response_model=Dict[str, Any]) async def get_notification( notification_id: int, db: AsyncSession = Depends(get_db), current_user = Depends(get_current_user) ): """ 获取消息通知详情 - **notification_id**: 通知ID 注意：只能查看自己的通知，管理员可以查看所有通知 """ notification = await notification_service.get_notification(db, notification_id) if not notification: raise HTTPException( status_code=status.HTTP_404_NOT_FOUND, detail="通知不存在" ) # 检查权限 if not current_user.is_superuser and notification["recipient_id"] != current_user.id: raise HTTPException( status_code=status.HTTP_403_FORBIDDEN, detail="无权查看此通知" ) return notification @router.post("/", response_model=Dict[str, Any], status_code=status.HTTP_201_CREATED) async def create_notification( obj_in: NotificationCreate, db: AsyncSession = Depends(get_db), current_user = Depends(get_current_user) ): """ 创建消息通知 - **recipient_id**: 接收人ID - **title**: 通知标题 - **content**: 通知内容 - **notification_type**: 通知类型 - **priority**: 优先级（low/normal/high/urgent） - **related_entity_type**: 关联实体类型 - **related_entity_id**: 关联实体ID - **action_url**: 操作链接 - **extra_data**: 额外数据 - **send_email**: 是否发送邮件 - **send_sms**: 是否发送短信 - **expire_at**: 过期时间 """ try: return await notification_service.create_notification(db, obj_in=obj_in) except ValueError as e: raise HTTPException( status_code=status.HTTP_400_BAD_REQUEST, detail=str(e) ) @router.post("/batch", response_model=Dict[str, Any]) async def batch_create_notifications( batch_in: NotificationBatchCreate, db: AsyncSession = Depends(get_db), current_user = Depends(get_current_user) ): """ 批量创建消息通知 - **recipient_ids**: 接收人ID列表 - **title**: 通知标题 - **content**: 通知内容 - **notification_type**: 通知类型 - **priority**: 优先级 - **action_url**: 操作链接 - **extra_data**: 额外数据 """ return await notification_service.batch_create_notifications(db, batch_in=batch_in) @router.post("/from-template", response_model=Dict[str, Any]) async def send_from_template( template_in: NotificationSendFromTemplate, db: AsyncSession = Depends(get_db), current_user = Depends(get_current_user) ): """ 从模板发送通知 - **template_code**: 模板编码 - **recipient_ids**: 接收人ID列表 - **variables**: 模板变量 - **related_entity_type**: 关联实体类型 - **related_entity_id**: 关联实体ID - **action_url**: 操作链接 """ try: return await notification_service.send_from_template(db, template_in=template_in) except ValueError as e: raise HTTPException( status_code=status.HTTP_400_BAD_REQUEST, detail=str(e) ) @router.put("/{notification_id}/read", response_model=Dict[str, Any]) async def mark_notification_as_read( notification_id: int, db: AsyncSession = Depends(get_db), current_user = Depends(get_current_user) ): """ 标记通知为已读 - **notification_id**: 通知ID """ try: notification = await notification_service.get_notification(db, notification_id) if not notification: raise HTTPException( status_code=status.HTTP_404_NOT_FOUND, detail="通知不存在" ) # 检查权限 if not current_user.is_superuser and notification["recipient_id"] != current_user.id: raise HTTPException( status_code=status.HTTP_403_FORBIDDEN, detail="无权操作此通知" ) return await notification_service.mark_as_read(db, notification_id) except ValueError as e: raise HTTPException( status_code=status.HTTP_400_BAD_REQUEST, detail=str(e) ) @router.put("/read", response_model=Dict[str, Any]) async def batch_mark_notifications_as_read( payload: NotificationIdsPayload, db: AsyncSession = Depends(get_db), current_user = Depends(get_current_user) ): """ 批量标记通知为已读 """ recipient_id = None if current_user.is_superuser else current_user.id return await notification_service.batch_mark_as_read( db, notification_ids=payload.ids, recipient_id=recipient_id ) @router.put("/read-all", response_model=Dict[str, Any]) async def mark_all_as_read( db: AsyncSession = Depends(get_db), current_user = Depends(get_current_user) ): """ 标记所有未读通知为已读 将当前用户的所有未读通知标记为已读 """ return await notification_service.mark_all_as_read(db, current_user.id) @router.post("/mark-all-read", response_model=Dict[str, Any]) async def mark_all_as_read_alias( db: AsyncSession = Depends(get_db), current_user = Depends(get_current_user) ): """ 标记所有未读通知为已读（别名） """ return await notification_service.mark_all_as_read(db, current_user.id) @router.delete("/{notification_id}", status_code=status.HTTP_204_NO_CONTENT) async def delete_notification( notification_id: int, db: AsyncSession = Depends(get_db), current_user = Depends(get_current_user) ): """ 删除消息通知 - **notification_id**: 通知ID """ notification = await notification_service.get_notification(db, notification_id) if not notification: raise HTTPException( status_code=status.HTTP_404_NOT_FOUND, detail="通知不存在" ) # 检查权限 if not current_user.is_superuser and notification["recipient_id"] != current_user.id: raise HTTPException( status_code=status.HTTP_403_FORBIDDEN, detail="无权删除此通知" ) await notification_service.delete_notification(db, notification_id) return None @router.delete("/", response_model=Dict[str, Any]) async def delete_notifications( payload: NotificationIdsPayload, db: AsyncSession = Depends(get_db), current_user = Depends(get_current_user) ): """ 批量删除消息通知（兼容前端） """ notification_ids = payload.ids # 检查权限 if not current_user.is_superuser: notifications = await notification_service.get_notifications( db, skip=0, limit=len(notification_ids) * 2 ) valid_ids = [ n["id"] for n in notifications["items"] if n["recipient_id"] == current_user.id and n["id"] in notification_ids ] if not valid_ids: raise HTTPException( status_code=status.HTTP_400_BAD_REQUEST, detail="没有有效的通知ID" ) notification_ids = valid_ids return await notification_service.batch_delete_notifications(db, notification_ids) @router.post("/batch-delete", response_model=Dict[str, Any]) async def batch_delete_notifications( notification_ids: List[int], db: AsyncSession = Depends(get_db), current_user = Depends(get_current_user) ): """ 批量删除消息通知 - **notification_ids**: 通知ID列表 """ # 检查权限 if not current_user.is_superuser: # 普通用户只能删除自己的通知 notifications = await notification_service.get_notifications( db, skip=0, limit=len(notification_ids) * 2 ) valid_ids = [ n["id"] for n in notifications["items"] if n["recipient_id"] == current_user.id and n["id"] in notification_ids ] if not valid_ids: raise HTTPException( status_code=status.HTTP_400_BAD_REQUEST, detail="没有有效的通知ID" ) notification_ids = valid_ids return await notification_service.batch_delete_notifications(db, notification_ids) @router.post("/mark-unread", response_model=Dict[str, Any]) async def batch_mark_notifications_as_unread( payload: NotificationIdsPayload, db: AsyncSession = Depends(get_db), current_user = Depends(get_current_user) ): """ 批量标记通知为未读 """ recipient_id = None if current_user.is_superuser else current_user.id return await notification_service.batch_mark_as_unread( db, notification_ids=payload.ids, recipient_id=recipient_id )