ea807974cf
- Profile API: GET/PUT /profile + PUT /profile/password - Messages API: 模型/迁移(005)/服务/路由 + 任务操作自动创建消息 - SSE 推送集成: tasks.py 中 6 个操作触发 SSE 通知 - Alembic 迁移: 004 audit_logs + 005 messages - env.py 导入所有模型确保迁移正确 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
98 lines
2.8 KiB
Python
98 lines
2.8 KiB
Python
"""
|
|
消息/通知 API
|
|
"""
|
|
from typing import Optional
|
|
from fastapi import APIRouter, Depends, HTTPException, Query, status
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.database import get_db
|
|
from app.models.user import User
|
|
from app.api.deps import get_current_user
|
|
from app.schemas.message import MessageResponse, MessageListResponse, UnreadCountResponse
|
|
from app.services.message_service import (
|
|
list_messages,
|
|
get_unread_count,
|
|
mark_as_read,
|
|
mark_all_as_read,
|
|
)
|
|
|
|
router = APIRouter(prefix="/messages", tags=["消息"])
|
|
|
|
|
|
@router.get("", response_model=MessageListResponse)
|
|
async def get_messages(
|
|
page: int = Query(1, ge=1),
|
|
page_size: int = Query(20, ge=1, le=100),
|
|
is_read: Optional[bool] = Query(None),
|
|
type: Optional[str] = Query(None),
|
|
current_user: User = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
"""获取消息列表"""
|
|
messages, total = await list_messages(
|
|
db=db,
|
|
user_id=current_user.id,
|
|
page=page,
|
|
page_size=page_size,
|
|
is_read=is_read,
|
|
type=type,
|
|
)
|
|
|
|
return MessageListResponse(
|
|
items=[
|
|
MessageResponse(
|
|
id=m.id,
|
|
type=m.type,
|
|
title=m.title,
|
|
content=m.content,
|
|
is_read=m.is_read,
|
|
related_task_id=m.related_task_id,
|
|
related_project_id=m.related_project_id,
|
|
sender_name=m.sender_name,
|
|
created_at=m.created_at,
|
|
)
|
|
for m in messages
|
|
],
|
|
total=total,
|
|
page=page,
|
|
page_size=page_size,
|
|
)
|
|
|
|
|
|
@router.get("/unread-count", response_model=UnreadCountResponse)
|
|
async def get_message_unread_count(
|
|
current_user: User = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
"""获取未读消息数"""
|
|
count = await get_unread_count(db, current_user.id)
|
|
return UnreadCountResponse(count=count)
|
|
|
|
|
|
@router.put("/{message_id}/read")
|
|
async def mark_message_as_read(
|
|
message_id: str,
|
|
current_user: User = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
"""标记消息已读"""
|
|
success = await mark_as_read(db, message_id, current_user.id)
|
|
if not success:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_404_NOT_FOUND,
|
|
detail="消息不存在",
|
|
)
|
|
await db.commit()
|
|
return {"message": "已标记为已读"}
|
|
|
|
|
|
@router.put("/read-all")
|
|
async def mark_all_messages_as_read(
|
|
current_user: User = Depends(get_current_user),
|
|
db: AsyncSession = Depends(get_db),
|
|
):
|
|
"""标记所有消息已读"""
|
|
count = await mark_all_as_read(db, current_user.id)
|
|
await db.commit()
|
|
return {"message": f"已标记 {count} 条消息为已读", "count": count}
|