bangbang-aigc-server/api_v2/routers/data.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

"""
Data Management Router
数据查询路由 - API v2
对应原系统的数据库查询功能
"""

import logging
from typing import Dict, Any, List, Optional
from fastapi import APIRouter, Depends, HTTPException, Query
from fastapi.responses import JSONResponse

from ..models import ApiResponse
from ..services import DatabaseService

logger = logging.getLogger(__name__)

router = APIRouter()

# 全局数据库服务实例
_db_service = None


def get_database_service() -> DatabaseService:
    """获取数据库服务实例（单例模式）"""
    global _db_service
    if _db_service is None:
        _db_service = DatabaseService()
    return _db_service


@router.get("/scenic-spots", summary="获取景区列表")
async def get_scenic_spots(
    limit: int = Query(10, ge=1, le=100, description="返回数量限制"),
    offset: int = Query(0, ge=0, description="偏移量"),
    search: Optional[str] = Query(None, description="搜索关键词"),
    db_service: DatabaseService = Depends(get_database_service)
):
    """获取景区列表"""
    try:
        spots, total = db_service.get_scenic_spots(limit, offset, search)
        
        return ApiResponse(
            success=True,
            message="获取景区列表成功",
            data={
                "scenic_spots": spots,
                "total": total,
                "limit": limit,
                "offset": offset,
                "has_more": offset + limit < total
            }
        )
        
    except Exception as e:
        logger.error(f"获取景区列表失败: {e}")
        return JSONResponse(
            status_code=500,
            content=ApiResponse(
                success=False,
                message="获取景区列表失败",
                error=str(e)
            ).dict()
        )


@router.get("/scenic-spots/{spot_id}", summary="获取景区详情")
async def get_scenic_spot(
    spot_id: int,
    db_service: DatabaseService = Depends(get_database_service)
):
    """根据ID获取景区详情"""
    try:
        spot = db_service.get_scenic_spot_by_id(spot_id)
        
        if not spot:
            raise HTTPException(status_code=404, detail="景区不存在")
        
        return ApiResponse(
            success=True,
            message="获取景区详情成功",
            data=spot
        )
        
    except HTTPException:
        raise
    except Exception as e:
        logger.error(f"获取景区详情失败: {e}")
        return JSONResponse(
            status_code=500,
            content=ApiResponse(
                success=False,
                message="获取景区详情失败",
                error=str(e)
            ).dict()
        )


@router.get("/products", summary="获取产品列表")
async def get_products(
    limit: int = Query(10, ge=1, le=100, description="返回数量限制"),
    offset: int = Query(0, ge=0, description="偏移量"),
    search: Optional[str] = Query(None, description="搜索关键词"),
    db_service: DatabaseService = Depends(get_database_service)
):
    """获取产品列表"""
    try:
        products, total = db_service.get_products(limit, offset, search)
        
        return ApiResponse(
            success=True,
            message="获取产品列表成功",
            data={
                "products": products,
                "total": total,
                "limit": limit,
                "offset": offset,
                "has_more": offset + limit < total
            }
        )
        
    except Exception as e:
        logger.error(f"获取产品列表失败: {e}")
        return JSONResponse(
            status_code=500,
            content=ApiResponse(
                success=False,
                message="获取产品列表失败",
                error=str(e)
            ).dict()
        )


@router.get("/products/{product_id}", summary="获取产品详情")
async def get_product(
    product_id: int,
    db_service: DatabaseService = Depends(get_database_service)
):
    """根据ID获取产品详情"""
    try:
        product = db_service.get_product_by_id(product_id)
        
        if not product:
            raise HTTPException(status_code=404, detail="产品不存在")
        
        return ApiResponse(
            success=True,
            message="获取产品详情成功",
            data=product
        )
        
    except HTTPException:
        raise
    except Exception as e:
        logger.error(f"获取产品详情失败: {e}")
        return JSONResponse(
            status_code=500,
            content=ApiResponse(
                success=False,
                message="获取产品详情失败",
                error=str(e)
            ).dict()
        )


@router.get("/styles", summary="获取风格列表")
async def get_styles(
    db_service: DatabaseService = Depends(get_database_service)
):
    """获取风格列表"""
    try:
        styles = db_service.get_styles()
        
        return ApiResponse(
            success=True,
            message="获取风格列表成功",
            data={"styles": styles, "total": len(styles)}
        )
        
    except Exception as e:
        logger.error(f"获取风格列表失败: {e}")
        return JSONResponse(
            status_code=500,
            content=ApiResponse(
                success=False,
                message="获取风格列表失败",
                error=str(e)
            ).dict()
        )


@router.get("/audiences", summary="获取受众列表")
async def get_audiences(
    db_service: DatabaseService = Depends(get_database_service)
):
    """获取受众列表"""
    try:
        audiences = db_service.get_audiences()
        
        return ApiResponse(
            success=True,
            message="获取受众列表成功",
            data={"audiences": audiences, "total": len(audiences)}
        )
        
    except Exception as e:
        logger.error(f"获取受众列表失败: {e}")
        return JSONResponse(
            status_code=500,
            content=ApiResponse(
                success=False,
                message="获取受众列表失败",
                error=str(e)
            ).dict()
        )


@router.post("/scenic-spots/batch", summary="批量获取景区信息")
async def get_scenic_spots_batch(
    spot_ids: List[int],
    db_service: DatabaseService = Depends(get_database_service)
):
    """批量获取景区信息"""
    try:
        spots = db_service.get_scenic_spots_by_ids(spot_ids)
        
        return ApiResponse(
            success=True,
            message=f"批量获取 {len(spots)} 个景区信息成功",
            data={
                "scenic_spots": spots, 
                "requested_count": len(spot_ids),
                "found_count": len(spots)
            }
        )
        
    except Exception as e:
        logger.error(f"批量获取景区信息失败: {e}")
        return JSONResponse(
            status_code=500,
            content=ApiResponse(
                success=False,
                message="批量获取景区信息失败",
                error=str(e)
            ).dict()
        )


@router.post("/products/batch", summary="批量获取产品信息")
async def get_products_batch(
    product_ids: List[int],
    db_service: DatabaseService = Depends(get_database_service)
):
    """批量获取产品信息"""
    try:
        products = db_service.get_products_by_ids(product_ids)
        
        return ApiResponse(
            success=True,
            message=f"批量获取 {len(products)} 个产品信息成功",
            data={
                "products": products,
                "requested_count": len(product_ids),
                "found_count": len(products)
            }
        )
        
    except Exception as e:
        logger.error(f"批量获取产品信息失败: {e}")
        return JSONResponse(
            status_code=500,
            content=ApiResponse(
                success=False,
                message="批量获取产品信息失败",
                error=str(e)
            ).dict()
        )


@router.get("/health", summary="数据库健康检查")
async def database_health_check(
    db_service: DatabaseService = Depends(get_database_service)
):
    """检查数据库连接状态"""
    try:
        is_available = db_service.is_available()
        
        return ApiResponse(
            success=True,
            message="数据库健康检查完成",
            data={
                "database_available": is_available,
                "connection_type": "mysql" if is_available else "mock_data",
                "status": "healthy" if is_available else "using_fallback"
            }
        )
        
    except Exception as e:
        logger.error(f"数据库健康检查失败: {e}")
        return JSONResponse(
            status_code=500,
            content=ApiResponse(
                success=False,
                message="数据库健康检查失败",
                error=str(e)
            ).dict()
        )