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

from fastapi import APIRouter, Request
from fastapi.responses import JSONResponse
from database import SessionLocal
from models.user_data import UserData
from models.total import User
from models.points import PointsConsumptionLog
from utils.logger import logger

router = APIRouter()

REQUIRED_POINTS = 10


@router.get("/points/balance")
async def get_balance(request: Request):
    """获取用户积分余额（支持本地用户和外部用户）"""
    user_info = request.state.user
    if not user_info:
        return JSONResponse(status_code=401, content={"statusCode": 401, "msg": "未认证"})

    db = SessionLocal()
    try:
        # 优先查询本地用户
        user = db.query(User).filter(
            User.id == user_info.user_id,
            User.is_deleted == 0
        ).first()

        if user:
            return JSONResponse(content={
                "statusCode": 200,
                "msg": "success",
                "data": {"points": user.points or 0}
            })

        # 查询外部用户
        user_data = db.query(UserData).filter(
            UserData.accountID == user_info.account
        ).first()

        if not user_data:
            return JSONResponse(content={"statusCode": 404, "msg": "未找到用户数据"})

        return JSONResponse(content={
            "statusCode": 200,
            "msg": "success",
            "data": {"points": user_data.points or 0}
        })
    except Exception as e:
        logger.error(f"获取积分余额失败: {e}")
        return JSONResponse(content={"statusCode": 500, "msg": f"获取积分余额失败: {str(e)}"})
    finally:
        db.close()


@router.post("/points/consume")
async def consume_points(request: Request):
    """消费积分下载文件（每次消耗10积分，支持本地用户和外部用户）"""
    user_info = request.state.user
    if not user_info:
        return JSONResponse(status_code=401, content={"statusCode": 401, "msg": "未认证"})

    try:
        body = await request.json()
    except Exception:
        return JSONResponse(content={"statusCode": 400, "msg": "JSON解析失败"})

    file_name = body.get("file_name", "")
    file_url = body.get("file_url", "")

    db = SessionLocal()
    try:
        # 优先查询本地用户
        user = db.query(User).filter(
            User.id == user_info.user_id,
            User.is_deleted == 0
        ).first()

        if user:
            current_points = user.points or 0
            if current_points < REQUIRED_POINTS:
                return JSONResponse(content={
                    "statusCode": 400,
                    "msg": "积分不足，下载需要10积分",
                    "data": {
                        "current_points": current_points,
                        "required_points": REQUIRED_POINTS
                    }
                })

            new_balance = current_points - REQUIRED_POINTS
            user.points = new_balance

            log = PointsConsumptionLog(
                user_id=str(user.id),
                file_name=file_name,
                file_url=file_url,
                points_consumed=REQUIRED_POINTS,
                balance_after=new_balance
            )
            db.add(log)
            db.commit()

            return JSONResponse(content={
                "statusCode": 200,
                "msg": "success",
                "data": {
                    "new_balance": new_balance,
                    "points_consumed": REQUIRED_POINTS
                }
            })

        # 查询外部用户
        user_data = db.query(UserData).filter(
            UserData.accountID == user_info.account
        ).first()

        if not user_data:
            return JSONResponse(content={"statusCode": 404, "msg": "未找到用户数据"})

        current_points = user_data.points or 0
        if current_points < REQUIRED_POINTS:
            return JSONResponse(content={
                "statusCode": 400,
                "msg": "积分不足，下载需要10积分",
                "data": {
                    "current_points": current_points,
                    "required_points": REQUIRED_POINTS
                }
            })

        new_balance = current_points - REQUIRED_POINTS
        user_data.points = new_balance

        log = PointsConsumptionLog(
            user_id=user_info.account,
            file_name=file_name,
            file_url=file_url,
            points_consumed=REQUIRED_POINTS,
            balance_after=new_balance
        )
        db.add(log)
        db.commit()

        return JSONResponse(content={
            "statusCode": 200,
            "msg": "success",
            "data": {
                "new_balance": new_balance,
                "points_consumed": REQUIRED_POINTS
            }
        })
    except Exception as e:
        db.rollback()
        logger.error(f"消费积分失败: {e}")
        return JSONResponse(content={"statusCode": 500, "msg": f"消费积分失败: {str(e)}"})
    finally:
        db.close()


@router.post("/points/add")
async def add_points(request: Request):
    """增加积分（仅管理员）"""
    user_info = request.state.user
    if not user_info:
        return JSONResponse(status_code=401, content={"statusCode": 401, "msg": "未认证"})
    
    # 检查管理员权限
    if user_info.role != "admin":
        return JSONResponse(status_code=403, content={"statusCode": 403, "msg": "权限不足"})
    
    try:
        body = await request.json()
    except Exception:
        return JSONResponse(content={"statusCode": 400, "msg": "JSON解析失败"})
    
    user_id = body.get("user_id")
    points = body.get("points", 0)
    reason = body.get("reason", "")
    
    if not user_id or points <= 0:
        return JSONResponse(content={"statusCode": 400, "msg": "参数错误"})
    
    db = SessionLocal()
    try:
        # 优先查询本地用户
        user = db.query(User).filter(
            User.id == user_id,
            User.is_deleted == 0
        ).first()
        
        if user:
            user.points = (user.points or 0) + points
            db.commit()
            
            logger.info(f"管理员 {user_info.username} 为用户 {user_id} 添加 {points} 积分，原因: {reason}")
            return JSONResponse(content={
                "statusCode": 200,
                "msg": "积分添加成功",
                "data": {"new_balance": user.points}
            })
        
        # 查询外部用户
        user_data = db.query(UserData).filter(UserData.id == user_id).first()
        if not user_data:
            return JSONResponse(content={"statusCode": 404, "msg": "用户不存在"})
        
        user_data.points = (user_data.points or 0) + points
        db.commit()
        
        logger.info(f"管理员 {user_info.username} 为外部用户 {user_id} 添加 {points} 积分，原因: {reason}")
        return JSONResponse(content={
            "statusCode": 200,
            "msg": "积分添加成功",
            "data": {"new_balance": user_data.points}
        })
        
    except Exception as e:
        db.rollback()
        logger.error(f"添加积分失败: {e}")
        return JSONResponse(content={"statusCode": 500, "msg": f"添加积分失败: {str(e)}"})
    finally:
        db.close()


@router.get("/points/logs")
@router.get("/points/history")
async def get_consumption_history(request: Request, page: int = 1, pageSize: int = 10):
    """获取积分消费记录（支持本地用户和外部用户）"""
    user_info = request.state.user
    if not user_info:
        return JSONResponse(status_code=401, content={"statusCode": 401, "msg": "未认证"})

    db = SessionLocal()
    try:
        # 确定用户ID（本地用户用user_id，外部用户用account）
        user = db.query(User).filter(
            User.id == user_info.user_id,
            User.is_deleted == 0
        ).first()

        user_identifier = str(user.id) if user else user_info.account

        query = db.query(PointsConsumptionLog).filter(
            PointsConsumptionLog.user_id == user_identifier
        )
        total = query.count()
        logs = query.order_by(PointsConsumptionLog.id.desc()) \
                    .offset((page - 1) * pageSize) \
                    .limit(pageSize) \
                    .all()

        items = [
            {
                "id": log.id,
                "file_name": log.file_name,
                "file_url": log.file_url,
                "points_consumed": log.points_consumed,
                "balance_after": log.balance_after,
                "created_at": log.created_at,
            }
            for log in logs
        ]

        return JSONResponse(content={
            "statusCode": 200,
            "msg": "success",
            "data": {
                "list": items,
                "total": total,
                "page": page,
                "pageSize": pageSize,
            }
        })
    except Exception as e:
        logger.error(f"获取消费记录失败: {e}")
        return JSONResponse(content={"statusCode": 500, "msg": f"获取消费记录失败: {str(e)}"})
    finally:
        db.close()