yu-zhi-ran/platform/backend/api/topics.py

# 宇之然内容创作平台 - 选题管理API

from pydantic import BaseModel
from datetime import datetime
from fastapi import APIRouter, Depends, HTTPException, status, Query
from sqlalchemy.orm import Session
from typing import List, Optional
import json

from core.security import get_current_admin_user, create_audit_log
from app.database import get_db
from app.models import Topic, User

router = APIRouter()

class TopicCreateRequest(BaseModel):
    title: str
    field: Optional[str] = None
    priority_score: int = 0

class TopicUpdateRequest(BaseModel):
    title: Optional[str] = None
    field: Optional[str] = None
    priority_score: Optional[int] = None
    status: Optional[str] = None

@router.get("", response_model=List[dict])
async def get_topics(
    status: Optional[str] = Query(None),
    page: int = Query(1, ge=1),
    size: int = Query(20, ge=1, le=100),
    db: Session = Depends(get_db)
):
    """获取选题列表"""
    query = db.query(Topic)
    
    # 状态筛选
    if status:
        query = query.filter(Topic.status == status)
    
    # 分页
    offset = (page - 1) * size
    topics = query.offset(offset).limit(size).all()
    
    # 转换为字典格式
    result = []
    for topic in topics:
        result.append({
            "id": topic.id,
            "title": topic.title,
            "field": topic.field,
            "priority_score": topic.priority_score,
            "status": topic.status,
            "compliance_score": topic.compliance_score,
            "created_at": topic.created_at.isoformat() if topic.created_at else None,
            "updated_at": topic.updated_at.isoformat() if topic.updated_at else None,
            "generated_at": topic.generated_at.isoformat() if topic.generated_at else None,
            "published_at": topic.published_at.isoformat() if topic.published_at else None,
            "platform_urls": topic.platform_urls
        })
    
    return result

@router.post("/", response_model=dict)
async def create_topic(
    topic_data: TopicCreateRequest,
    current_user: User = Depends(get_current_admin_user),
    db: Session = Depends(get_db)
):
    """创建新选题（管理员功能）"""
    new_topic = Topic(
        title=topic_data.title,
        field=topic_data.field,
        priority_score=topic_data.priority_score,
        status="待处理"
    )
    db.add(new_topic)
    db.commit()
    db.refresh(new_topic)
    
    # 记录审计日志
    create_audit_log(
        db=db,
        user_id=current_user.id,
        action="create_topic",
        resource_type="topic",
        resource_id=new_topic.id,
        details=f"标题: {topic_data.title}"
    )
    
    return {
        "id": new_topic.id,
        "title": new_topic.title,
        "status": new_topic.status,
        "created_at": new_topic.created_at.isoformat() if new_topic.created_at else None
    }

@router.put("/{topic_id}", response_model=dict)
async def update_topic(
    topic_id: int,
    topic_data: TopicUpdateRequest,
    current_user: User = Depends(get_current_admin_user),
    db: Session = Depends(get_db)
):
    """更新选题信息（管理员功能）"""
    topic = db.query(Topic).filter(Topic.id == topic_id).first()
    if not topic:
        raise HTTPException(status_code=404, detail="选题不存在")
    
    # 更新字段
    if topic_data.title is not None:
        topic.title = topic_data.title
    if topic_data.field is not None:
        topic.field = topic_data.field
    if topic_data.priority_score is not None:
        topic.priority_score = topic_data.priority_score
    if topic_data.status is not None:
        topic.status = topic_data.status
    
    topic.updated_at = datetime.utcnow()
    db.commit()
    db.refresh(topic)
    
    # 记录审计日志
    create_audit_log(
        db=db,
        user_id=current_user.id,
        action="update_topic",
        resource_type="topic",
        resource_id=topic_id,
        details=f"状态更新为: {topic_data.status}"
    )
    
    return {
        "id": topic.id,
        "title": topic.title,
        "status": topic.status,
        "updated_at": topic.updated_at.isoformat() if topic.updated_at else None
    }

@router.delete("/{topic_id}")
async def delete_topic(
    topic_id: str,
    current_user: User = Depends(get_current_admin_user),
    db: Session = Depends(get_db)
):
    """删除选题（管理员功能）"""
    topic = db.query(Topic).filter(Topic.id == topic_id).first()
    if not topic:
        raise HTTPException(status_code=404, detail="选题不存在")
    
    db.delete(topic)
    db.commit()
    
    # 记录审计日志
    create_audit_log(
        db=db,
        user_id=current_user.id,
        action="delete_topic",
        resource_type="topic",
        resource_id=topic_id,
        details="选题已删除"
    )
    
    return {"message": "选题已成功删除"}

@router.get("/{topic_id}", response_model=dict)
async def get_topic(
    topic_id: int,
    db: Session = Depends(get_db)
):
    """获取单个选题详情"""
    topic = db.query(Topic).filter(Topic.id == topic_id).first()
    if not topic:
        raise HTTPException(status_code=404, detail="选题不存在")
    
    return {
        "id": topic.id,
        "title": topic.title,
        "field": topic.field,
        "priority_score": topic.priority_score,
        "status": topic.status,
        "compliance_score": topic.compliance_score,
        "created_at": topic.created_at.isoformat() if topic.created_at else None,
        "updated_at": topic.updated_at.isoformat() if topic.updated_at else None,
        "generated_at": topic.generated_at.isoformat() if topic.generated_at else None,
        "published_at": topic.published_at.isoformat() if topic.published_at else None,
        "platform_urls": topic.platform_urls
    }