shudao-main/shudao-chat-py/routers/total.py

from fastapi import APIRouter, Depends, Request
from fastapi.responses import StreamingResponse
from sqlalchemy.orm import Session
from pydantic import BaseModel
from typing import Optional
from database import get_db
from models.total import RecommendQuestion, PolicyFile, FunctionCard, HotQuestion, FeedbackQuestion
from models.chat import AIMessage
from models.user_data import UserData
from services.oss_service import oss_service
from utils.crypto import decrypt_url
import time
import httpx

router = APIRouter()


@router.get("/recommend_question")
async def get_recommend_question(db: Session = Depends(get_db)):
    """获取推荐问题"""
    questions = db.query(RecommendQuestion).limit(10).all()
    return {
        "statusCode": 200,
        "msg": "success",
        "data": [{"id": q.id, "question": q.question} for q in questions]
    }


@router.get("/get_policy_file")
async def get_policy_file(
    policy_type: Optional[int] = None,
    page: int = 1,
    page_size: int = 20,
    db: Session = Depends(get_db)
):
    """获取策略文件列表"""
    query = db.query(PolicyFile).filter(PolicyFile.is_deleted == 0)

    if policy_type is not None and policy_type != 0:
        query = query.filter(PolicyFile.policy_type == policy_type)

    total = query.count()

    offset = (page - 1) * page_size
    files = query.order_by(PolicyFile.updated_at.desc()).offset(
        offset).limit(page_size).all()

    return {
        "statusCode": 200,
        "msg": "success",
        "data": {
            "total": total,
            "items": [
                {
                    "id": f.id,
                    "policy_name": f.policy_name,
                    "policy_file_url": f.policy_file_url,
                    "policy_type": f.policy_type,
                    "file_type": f.file_type,
                    "view_count": f.view_count,
                    "created_at": f.created_at
                }
                for f in files
            ]
        }
    }


@router.get("/get_function_card")
async def get_function_card(db: Session = Depends(get_db)):
    """获取功能卡片"""
    cards = db.query(FunctionCard).limit(4).all()
    return {
        "statusCode": 200,
        "msg": "success",
        "data": [
            {
                "id": c.id,
                "function_title": c.function_title,
                "function_icon": c.function_icon,
                "function_content": c.function_content,
                "function_type": c.function_type
            }
            for c in cards
        ]
    }


@router.get("/get_hot_question")
async def get_hot_question(db: Session = Depends(get_db)):
    """获取热点问题（按点击量排序）"""
    questions = db.query(HotQuestion).order_by(
        HotQuestion.click_count.desc()).limit(3).all()
    return {
        "statusCode": 200,
        "msg": "success",
        "data": [
            {
                "id": q.id,
                "question": q.question,
                "click_count": q.click_count or 0
            }
            for q in questions
        ]
    }


class SubmitFeedbackRequest(BaseModel):
    feedback_type: str
    content: str
    contact: str = ""


@router.post("/submit_feedback")
async def submit_feedback(request: SubmitFeedbackRequest, req: Request, db: Session = Depends(get_db)):
    """提交意见反馈（对齐Go版本）"""
    # 从token获取user_id
    user = req.state.user
    user_id = user.id if user else 0

    # 映射反馈类型：支持中文描述和英文标识
    type_map = {
        "功能建议": 1, "bug": 1, "问题反馈": 1,
        "ui": 2, "界面优化": 2,
        "experience": 3, "体验问题": 3,
        "other": 4, "其他": 4
    }
    feedback_type_id = type_map.get(request.feedback_type, 4)

    feedback = FeedbackQuestion(
        feedback_type=feedback_type_id,
        feedback_content=request.content,
        feedback_user_phone=request.contact,
        user_id=user_id,
        created_at=int(time.time()),
        updated_at=int(time.time())
    )
    db.add(feedback)
    db.commit()

    return {
        "statusCode": 200,
        "msg": "感谢您的反馈！"
    }


class LikeDislikeRequest(BaseModel):
    ai_message_id: int
    action: str  # "like" 或 "dislike"


@router.post("/like_and_dislike")
async def like_and_dislike(request: LikeDislikeRequest, db: Session = Depends(get_db)):
    """点赞/踩（对齐Go版本）"""
    message = db.query(AIMessage).filter(
        AIMessage.id == request.ai_message_id).first()
    if not message:
        return {"statusCode": 404, "msg": "消息不存在"}

    # 将action转换为user_feedback：like=2(满意/赞), dislike=3(不满意/踩)
    user_feedback = 2 if request.action == "like" else 3
    message.user_feedback = user_feedback
    message.updated_at = int(time.time())
    db.commit()

    return {"statusCode": 200, "msg": "success"}


@router.get("/get_user_data_id")
async def get_user_data_id(request: Request, db: Session = Depends(get_db)):
    """通过 token 的 account_id 查询 UserData 主键"""
    user = request.state.user
    if not user:
        return {"statusCode": 401, "msg": "未认证"}

    user_data = db.query(UserData).filter(
        UserData.accountID == user.account).first()
    if not user_data:
        return {"statusCode": 404, "msg": "用户数据不存在"}

    return {
        "statusCode": 200,
        "msg": "success",
        "data": {"user_data_id": user_data.id}
    }


class PolicyFileCountRequest(BaseModel):
    policy_file_id: int


@router.post("/policy_file_count")
async def get_policy_file_view_and_download_count(request: PolicyFileCountRequest, db: Session = Depends(get_db)):
    """更新政策文件查看计数 - 对齐Go版本接口名"""
    policy_file = db.query(PolicyFile).filter(
        PolicyFile.id == request.policy_file_id).first()
    if not policy_file:
        return {"statusCode": 404, "msg": "文件不存在"}

    policy_file.view_count = (policy_file.view_count or 0) + 1
    policy_file.updated_at = int(time.time())
    db.commit()

    return {"statusCode": 200, "msg": "success"}


@router.get("/download_file")
async def get_pdf_oss_download_link(pdf_oss_download_link: str):
    """流式代理下载 OSS 文件（解密代理 URL）- 对齐Go版本接口名"""
    try:
        # 解密代理 URL 获取真实 OSS URL
        real_url = decrypt_url(pdf_oss_download_link)

        # 流式代理下载
        async with httpx.AsyncClient() as client:
            async with client.stream("GET", real_url) as response:
                if response.status_code != 200:
                    return {"statusCode": response.status_code, "msg": "文件下载失败"}

                # 获取文件名和内容类型
                content_type = response.headers.get(
                    "content-type", "application/octet-stream")
                content_disposition = response.headers.get(
                    "content-disposition", "")

                async def generate():
                    async for chunk in response.aiter_bytes():
                        yield chunk

                return StreamingResponse(
                    generate(),
                    media_type=content_type,
                    headers={
                        "Content-Disposition": content_disposition} if content_disposition else {}
                )
    except Exception as e:
        return {"statusCode": 500, "msg": f"下载失败: {str(e)}"}