CardScoreDataServer/app/api/cards.py

from datetime import datetime
import os
from typing import Optional, List
from fastapi import APIRouter, Depends, HTTPException, Query

from mysql.connector.pooling import PooledMySQLConnection

from app.core.config import settings
from app.core.logger import get_logger
from app.core.database_loader import get_db_connection
from app.utils.scheme import (
    CardDetailResponse, CardListDetailResponse, CardType, SortBy, SortOrder
)
from app.crud import crud_card

logger = get_logger(__name__)
router = APIRouter()
db_dependency = Depends(get_db_connection)


@router.post("/created", response_model=CardDetailResponse, status_code=201, summary="创建一个新的卡牌记录")
def create_card(
        card_name: Optional[str] = Query(None, summary="卡牌的名称"),
        card_type: CardType = Query(CardType.pokemon, summary="卡牌类型"),
        db_conn: PooledMySQLConnection = db_dependency
):
    """创建一个新的卡牌实体，此时它不关联任何图片。"""
    try:
        with db_conn.cursor(dictionary=True) as cursor:

            query = f"INSERT INTO {settings.DB_CARD_TABLE_NAME} (card_name, card_type) VALUES (%s, %s)"
            cursor.execute(query, (card_name, card_type.value))
            db_conn.commit()
            new_id = cursor.lastrowid
            logger.info(f"新卡牌已创建, ID: {new_id}, 类型: {card_type.value}")

            # 返回刚创建的空卡牌信息
            cursor.execute(f"SELECT * FROM {settings.DB_CARD_TABLE_NAME} WHERE id = %s", (new_id,))
            new_card_data = cursor.fetchone()

            response_data = {**new_card_data, "images": []}
            return CardDetailResponse.model_validate(response_data)

    except Exception as e:
        db_conn.rollback()
        logger.error(f"创建卡牌失败: {e}")
        raise HTTPException(status_code=500, detail="数据库插入失败。")


@router.get("/query/{id}", response_model=CardDetailResponse, summary="获取指定卡牌的详细信息")
def get_card_details(id: int, db_conn: PooledMySQLConnection = db_dependency):
    """获取卡牌元数据以及所有与之关联的图片信息。分数是预先计算好的。"""
    # REFACTORED: Use CRUD function
    card_data = crud_card.get_card_with_details(db_conn, id)
    if not card_data:
        raise HTTPException(status_code=404, detail=f"ID为 {id} 的卡牌未找到。")
    return CardDetailResponse.model_validate(card_data)


@router.get("/query_next/{id}", response_model=CardDetailResponse, summary="获取指定卡牌id的下一个卡的详细信息")
def get_next_card_details(id: int, db_conn: PooledMySQLConnection = db_dependency):  # Renamed function
    """获取指定ID的下一张卡牌的元数据以及所有与之关联的图片信息。"""
    try:
        with db_conn.cursor(dictionary=True) as cursor:
            query_next_card = (
                f"SELECT id FROM {settings.DB_CARD_TABLE_NAME} "
                f"WHERE id > %s ORDER BY id ASC LIMIT 1"
            )
            cursor.execute(query_next_card, (id,))
            next_card_row = cursor.fetchone()

            if not next_card_row:
                raise HTTPException(status_code=200, detail=f"没有下一张")

            next_card_id = next_card_row['id']
            # 获取单个卡牌的完整信息
            card_data = crud_card.get_card_with_details(db_conn, next_card_id)
            if not card_data:
                raise HTTPException(status_code=404, detail=f"下一个卡牌ID {next_card_id} 未找到详细信息。")

            return CardDetailResponse.model_validate(card_data)

    except Exception as e:
        logger.error(f"查询下一个卡牌详情失败 (基准ID: {id}): {e}")
        if isinstance(e, HTTPException): raise e
        raise HTTPException(status_code=500, detail="服务器内部错误，查询数据库失败。")


@router.get("/card_list", response_model=List[CardListDetailResponse], summary="获取卡牌列表（支持筛选和排序）")
def list_cards_detailed(
        card_id: Optional[int] = Query(None, description="筛选条件：卡牌ID (精确匹配)"),
        card_name: Optional[str] = Query(None, description="筛选条件：卡牌名称 (模糊匹配)"),
        card_type: Optional[CardType] = Query(None, description="筛选条件：卡牌类型"),
        sort_by: SortBy = Query(SortBy.updated_at, description="排序字段"),
        sort_order: SortOrder = Query(SortOrder.desc, description="排序顺序"),
        skip: int = Query(0, ge=0, description="分页：跳过的记录数"),
        limit: int = Query(100, ge=1, le=1000, description="分页：每页的记录数"),
        db_conn: PooledMySQLConnection = db_dependency
):
    """获取卡牌的基础信息列表，支持按名称、类型筛选，以及多字段排序和分页。"""
    try:
        cards_with_images = crud_card.get_card_list_with_images(
            db_conn, card_id, card_name, card_type, sort_by, sort_order, skip, limit
        )
        card_list = [CardListDetailResponse.model_validate(c) for c in cards_with_images]
        logger.info(card_list)
        return card_list
    except Exception as e:
        logger.error(f"查询卡牌列表失败: {e}")
        raise HTTPException(status_code=500, detail="获取数据列表失败。")


@router.delete("/delete/{id}", status_code=200, summary="删除卡牌及其所有关联图片")
def delete_card(id: int, db_conn: PooledMySQLConnection = db_dependency):
    """
    删除一张卡牌及其所有关联的图片记录和物理文件。
    利用了数据库的 ON DELETE CASCADE 特性。
    """
    try:
        with db_conn.cursor() as cursor:
            # 1. 查询所有关联图片的所有物理文件路径
            query_paths = (f"SELECT image_path, detection_image_path, modified_image_path "
                           f"FROM {settings.DB_IMAGE_TABLE_NAME} WHERE card_id = %s")
            cursor.execute(query_paths, (id,))
            image_paths_to_delete = [path for row in cursor.fetchall() for path in row if path]

            # 2. 删除卡牌记录
            query_delete_card = f"DELETE FROM {settings.DB_CARD_TABLE_NAME} WHERE id = %s"
            cursor.execute(query_delete_card, (id,))

            if cursor.rowcount == 0:
                raise HTTPException(status_code=404, detail=f"ID为 {id} 的卡牌未找到。")

            db_conn.commit()
            logger.info(f"ID {id} 的卡牌和关联数据已成功删除。")

            # 3. 删除物理文件
            for path in image_paths_to_delete:
                absolute_path = settings.BASE_PATH / path.lstrip('/\\')
                if os.path.exists(absolute_path):
                    try:
                        os.remove(absolute_path)
                        logger.info(f"图片文件已删除: {absolute_path}")
                    except OSError as e:
                        logger.error(f"删除文件失败 {absolute_path}: {e}")

            return {"message": f"成功删除卡牌 ID {id} 及其所有关联数据"}

    except Exception as e:
        db_conn.rollback()
        logger.error(f"删除卡牌失败 ({id}): {e}")
        if isinstance(e, HTTPException): raise e
        raise HTTPException(status_code=500, detail="删除卡牌失败。")