qyyfw
78ff261b5a
- 前端 React+TypeScript (源码+构建产物) - 后端 Python FastAPI - 数据库迁移 001-015 - Docker部署配置 - 微信ClawBot API频道绑定 - 项目文档 README.md
136 lines
3.5 KiB
Python
136 lines
3.5 KiB
Python
from fastapi import APIRouter, Depends, HTTPException, Query
|
|
from sqlalchemy.orm import Session
|
|
from sqlalchemy import and_, desc, asc, func
|
|
from pydantic import BaseModel, Field
|
|
from typing import Optional
|
|
import uuid
|
|
from datetime import datetime
|
|
|
|
from ..database import get_db
|
|
from ..models import (
|
|
User, Notification,
|
|
gen_uuid
|
|
)
|
|
from ..auth import get_current_user
|
|
from ..middleware.log_service import LogService
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.get("/")
|
|
def list_notifications(
|
|
type: Optional[str] = Query(None, description="通知类型"),
|
|
is_read: Optional[bool] = Query(None),
|
|
page: int = Query(default=1, ge=1),
|
|
page_size: int = Query(default=20, ge=1, le=100),
|
|
db: Session = Depends(get_db),
|
|
current_user: User = Depends(get_current_user),
|
|
):
|
|
query = db.query(Notification).filter(
|
|
Notification.user_id == current_user.id
|
|
)
|
|
|
|
if type:
|
|
query = query.filter(Notification.type == type)
|
|
if is_read is not None:
|
|
query = query.filter(Notification.is_read == is_read)
|
|
|
|
total = query.count()
|
|
notifications = (
|
|
query.order_by(desc(Notification.created_at))
|
|
.offset((page - 1) * page_size)
|
|
.limit(page_size)
|
|
.all()
|
|
)
|
|
|
|
result = []
|
|
for n in notifications:
|
|
result.append({
|
|
"id": n.id,
|
|
"type": n.type,
|
|
"title": n.title,
|
|
"content": n.content,
|
|
"is_read": n.is_read,
|
|
"related_id": n.related_id,
|
|
"related_type": n.related_type,
|
|
"created_at": str(n.created_at) if n.created_at else None,
|
|
})
|
|
|
|
return {
|
|
"code": 0,
|
|
"data": {
|
|
"total": total,
|
|
"page": page,
|
|
"page_size": page_size,
|
|
"items": result,
|
|
},
|
|
"message": "ok",
|
|
}
|
|
|
|
|
|
@router.put("/{notification_id}/read")
|
|
def mark_as_read(
|
|
notification_id: str,
|
|
db: Session = Depends(get_db),
|
|
current_user: User = Depends(get_current_user),
|
|
):
|
|
notification = db.query(Notification).filter(
|
|
and_(
|
|
Notification.id == notification_id,
|
|
Notification.user_id == current_user.id,
|
|
)
|
|
).first()
|
|
|
|
if not notification:
|
|
raise HTTPException(status_code=404, detail="通知不存在")
|
|
|
|
notification.is_read = True
|
|
notification.read_at = datetime.now()
|
|
db.commit()
|
|
|
|
return {"code": 0, "message": "已标记为已读"}
|
|
|
|
|
|
@router.put("/read-all")
|
|
def mark_all_as_read(
|
|
type: Optional[str] = Query(None),
|
|
db: Session = Depends(get_db),
|
|
current_user: User = Depends(get_current_user),
|
|
):
|
|
query = db.query(Notification).filter(
|
|
and_(
|
|
Notification.user_id == current_user.id,
|
|
Notification.is_read == False,
|
|
)
|
|
)
|
|
|
|
if type:
|
|
query = query.filter(Notification.type == type)
|
|
|
|
notifications = query.all()
|
|
now = datetime.now()
|
|
count = 0
|
|
for n in notifications:
|
|
n.is_read = True
|
|
n.read_at = now
|
|
count += 1
|
|
|
|
db.commit()
|
|
|
|
return {"code": 0, "data": {"count": count}, "message": f"已标记 {count} 条为已读"}
|
|
|
|
|
|
@router.get("/unread-count")
|
|
def get_unread_count(
|
|
db: Session = Depends(get_db),
|
|
current_user: User = Depends(get_current_user),
|
|
):
|
|
count = db.query(func.count(Notification.id)).filter(
|
|
and_(
|
|
Notification.user_id == current_user.id,
|
|
Notification.is_read == False,
|
|
)
|
|
).scalar() or 0
|
|
|
|
return {"code": 0, "data": {"count": count}, "message": "ok"}
|