sqxp/backend/app/api/product_banner.py

"""API - 商品轮播图管理"""
from typing import Optional, List

from fastapi import APIRouter, Depends, HTTPException, Query
from sqlalchemy.orm import Session

from app.core.database import get_db
from app.core.auth import get_current_user
from app.core.response import success_response, error_response
from app.crud.product_banner import product_banner
from app.schemas.product_banner import (
    ProductBannerCreate,
    ProductBannerUpdate,
    ProductBannerSortUpdate,
    ProductBannerResponse,
    ProductBannerListResponse
)

router = APIRouter(prefix="/web/admin/product-banners", tags=["商品轮播图管理"])


def require_admin(current_user: dict = Depends(get_current_user)):
    """验证运营人员权限"""
    if current_user.get("role") != "admin":
        raise HTTPException(status_code=403, detail="需要运营人员权限")
    return current_user


@router.get("/product/{product_id}", response_model=dict)
def list_banners(
    product_id: str,
    db: Session = Depends(get_db),
    current_user: dict = Depends(require_admin)
):
    """获取商品的所有轮播图"""
    banners = product_banner.get_by_product(db, product_id)
    return success_response({
        "list": [{
            "id": b.id,
            "productId": b.product_id,
            "mediaType": b.media_type,
            "fileUrl": b.file_url,
            "thumbnailUrl": b.thumbnail_url,
            "sortOrder": b.sort_order,
            "createAt": str(b.create_at) if b.create_at else None
        } for b in banners],
        "total": len(banners)
    })


@router.post("/product/{product_id}", response_model=dict)
def create_banners(
    product_id: str,
    items: List[ProductBannerCreate],
    db: Session = Depends(get_db),
    current_user: dict = Depends(require_admin)
):
    """批量添加轮播图"""
    creator = {"user_id": current_user.get("user_id"), "real_name": current_user.get("real_name")}
    
    items_data = [item.model_dump() for item in items]
    banners = product_banner.create_batch(db, product_id, items_data, creator)
    
    return success_response({
        "list": [{
            "id": b.id,
            "productId": b.product_id,
            "mediaType": b.media_type,
            "fileUrl": b.file_url,
            "thumbnailUrl": b.thumbnail_url,
            "sortOrder": b.sort_order
        } for b in banners],
        "total": len(banners)
    })


@router.put("/product/{product_id}/sort", response_model=dict)
def update_sort(
    product_id: str,
    sort_data: ProductBannerSortUpdate,
    db: Session = Depends(get_db),
    current_user: dict = Depends(require_admin)
):
    """更新轮播图排序"""
    items_data = [item.model_dump() for item in sort_data.items]
    product_banner.update_sort(db, items_data)
    return success_response({"message": "排序更新成功"})


@router.put("/{banner_id}", response_model=dict)
def update_banner(
    banner_id: str,
    update_data: ProductBannerUpdate,
    db: Session = Depends(get_db),
    current_user: dict = Depends(require_admin)
):
    """更新轮播图"""
    banner = product_banner.update(db, banner_id, update_data.model_dump(exclude_unset=True))
    if not banner:
        return error_response(404, "轮播图不存在")
    return success_response({
        "id": banner.id,
        "productId": banner.product_id,
        "mediaType": banner.media_type,
        "fileUrl": banner.file_url,
        "thumbnailUrl": banner.thumbnail_url,
        "sortOrder": banner.sort_order
    })


@router.delete("/{banner_id}", response_model=dict)
def delete_banner(
    banner_id: str,
    db: Session = Depends(get_db),
    current_user: dict = Depends(require_admin)
):
    """删除轮播图"""
    result = product_banner.delete(db, banner_id)
    if not result:
        return error_response(404, "轮播图不存在")
    return success_response({"message": "删除成功"})