buff/backend/app/api/admin/announcements.py

"""
公告管理接口
"""
from fastapi import APIRouter, Depends
from sqlalchemy.orm import Session
from pydantic import BaseModel
from app.database import get_db
from app.api.deps import get_current_admin
from app.models.announcement import Announcement
from datetime import datetime

router = APIRouter()

class AnnouncementCreate(BaseModel):
    """创建公告请求"""
    title: str
    content: str
    priority: int = 0
    is_active: bool = True
    publish_time: str = None  # ISO格式时间字符串

@router.get("")
async def get_announcements(
    db: Session = Depends(get_db),
    current_admin = Depends(get_current_admin)
):
    """获取公告列表"""
    announcements = db.query(Announcement).all()
    
    return {
        "code": 200,
        "msg": "成功",
        "data": [
            {
                "id": a.id,
                "title": a.title,
                "content": a.content,
                "priority": a.priority,
                "is_active": bool(a.is_active),
                "publish_time": a.publish_time.isoformat() if a.publish_time else None,
                "created_at": a.created_at.isoformat() if a.created_at else None,
                "updated_at": a.updated_at.isoformat() if a.updated_at else None
            }
            for a in announcements
        ]
    }

@router.get("/{announcement_id}")
async def get_announcement_detail(
    announcement_id: int,
    db: Session = Depends(get_db),
    current_admin = Depends(get_current_admin)
):
    """获取公告详情"""
    announcement = db.query(Announcement).filter(Announcement.id == announcement_id).first()
    if not announcement:
        return {"code": 404, "msg": "公告不存在"}
    
    return {
        "code": 200,
        "msg": "成功",
        "data": {
            "id": announcement.id,
            "title": announcement.title,
            "content": announcement.content,
            "priority": announcement.priority,
            "is_active": bool(announcement.is_active),
            "publish_time": announcement.publish_time.isoformat() if announcement.publish_time else None,
            "created_at": announcement.created_at.isoformat() if announcement.created_at else None,
            "updated_at": announcement.updated_at.isoformat() if announcement.updated_at else None
        }
    }

@router.post("")
async def create_announcement(
    req: AnnouncementCreate,
    db: Session = Depends(get_db),
    current_admin = Depends(get_current_admin)
):
    """创建公告"""
    publish_time = datetime.now()
    if req.publish_time:
        try:
            publish_time = datetime.fromisoformat(req.publish_time.replace('Z', '+00:00'))
        except:
            pass
    
    announcement = Announcement(
        title=req.title,
        content=req.content,
        priority=req.priority,
        is_active=1 if req.is_active else 0,
        publish_time=publish_time,
        created_at=datetime.now(),
        updated_at=datetime.now()
    )
    db.add(announcement)
    db.commit()
    
    return {"code": 200, "msg": "创建成功", "data": {"id": announcement.id}}

@router.put("/{announcement_id}")
async def update_announcement(
    announcement_id: int,
    req: AnnouncementCreate,
    db: Session = Depends(get_db),
    current_admin = Depends(get_current_admin)
):
    """更新公告"""
    announcement = db.query(Announcement).filter(Announcement.id == announcement_id).first()
    if not announcement:
        return {"code": 404, "msg": "公告不存在"}
    
    announcement.title = req.title
    announcement.content = req.content
    announcement.priority = req.priority
    announcement.is_active = 1 if req.is_active else 0
    if req.publish_time:
        try:
            announcement.publish_time = datetime.fromisoformat(req.publish_time.replace('Z', '+00:00'))
        except:
            pass
    announcement.updated_at = datetime.now()
    db.commit()
    
    return {"code": 200, "msg": "更新成功"}

@router.put("/{announcement_id}/status")
async def toggle_announcement_status(
    announcement_id: int,
    db: Session = Depends(get_db),
    current_admin = Depends(get_current_admin)
):
    """切换公告状态"""
    announcement = db.query(Announcement).filter(Announcement.id == announcement_id).first()
    if not announcement:
        return {"code": 404, "msg": "公告不存在"}
    
    announcement.is_active = 0 if announcement.is_active else 1
    announcement.updated_at = datetime.now()
    db.commit()
    
    return {"code": 200, "msg": "状态更新成功", "data": {"is_active": bool(announcement.is_active)}}

@router.delete("/{announcement_id}")
async def delete_announcement(
    announcement_id: int,
    db: Session = Depends(get_db),
    current_admin = Depends(get_current_admin)
):
    """删除公告"""
    announcement = db.query(Announcement).filter(Announcement.id == announcement_id).first()
    if not announcement:
        return {"code": 404, "msg": "公告不存在"}
    
    db.delete(announcement)
    db.commit()
    
    return {"code": 200, "msg": "删除成功"}