CardScoreDataServer/app/api/images.py

import os
import uuid
import json
from typing import Optional, Dict, Any
from enum import Enum
from fastapi import APIRouter, File, UploadFile, Depends, HTTPException, Form, Path
from fastapi.responses import JSONResponse, FileResponse
from mysql.connector.pooling import PooledMySQLConnection
from mysql.connector import IntegrityError

from app.core.config import settings
from app.core.logger import get_logger
from app.utils.scheme import CardImageResponse, ImageJsonPairResponse
from app.core.database_loader import get_db_connection

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


class ImageType(str, Enum):
    front_face = "front_face"
    back_face = "back_face"
    front_edge = "front_edge"
    back_edge = "back_edge"


@router.post("/insert/{card_id}", response_model=CardImageResponse, status_code=201,
             summary="为卡牌上传并关联一张图片")
async def upload_image_for_card(
        card_id: int = Path(..., description="要关联的卡牌ID"),
        image_type: ImageType = Form(..., description="图片类型 (front_face, back_face, etc.)"),
        image: UploadFile = File(..., description="图片文件"),
        image_name: Optional[str] = Form(None, description="图片的可选名称"),
        json_data_str: str = Form(..., description="与图片关联的JSON字符串"),
        db_conn: PooledMySQLConnection = db_dependency
):
    """
    上传一张图片，并将其作为一条新记录存入 card_images 表。
    这是一个事务性操作，并会检查是否存在重复的 (card_id, image_type) 组合。
    """
    try:
        detection_json = json.loads(json_data_str)
    except json.JSONDecodeError:
        raise HTTPException(status_code=400, detail="JSON格式无效。")

    file_extension = os.path.splitext(image.filename)[1]
    unique_filename = f"{uuid.uuid4()}{file_extension}"
    image_path = settings.DATA_DIR / unique_filename
    relative_path = f"/{image_path.parent.name}/{image_path.name}"
    try:
        with open(image_path, "wb") as buffer:
            buffer.write(await image.read())
    except Exception as e:
        logger.error(f"保存图片失败: {e}")
        raise HTTPException(status_code=500, detail="无法保存图片文件。")

    cursor = None
    try:
        cursor = db_conn.cursor()

        # 检查 card_id 是否存在
        cursor.execute(f"SELECT id FROM {settings.DB_CARD_TABLE_NAME} WHERE id = %s", (card_id,))
        if not cursor.fetchone():
            raise HTTPException(status_code=404, detail=f"ID为 {card_id} 的卡牌不存在。")

        query_insert_image = (
            f"INSERT INTO {settings.DB_IMAGE_TABLE_NAME} "
            "(card_id, image_type, image_name, image_path, detection_json) "
            "VALUES (%s, %s, %s, %s, %s)"
        )
        params = (
            card_id, image_type.value, image_name, relative_path, json.dumps(detection_json, ensure_ascii=False))

        cursor.execute(query_insert_image, params)
        new_id = cursor.lastrowid

        db_conn.commit()
        logger.info(f"图片 {new_id} 已成功关联到卡牌 {card_id}，类型为 {image_type.value}。")

        cursor.execute(f"SELECT * FROM {settings.DB_IMAGE_TABLE_NAME} WHERE id = %s", (new_id,))
        new_image_data = cursor.fetchone()
        columns = [desc[0] for desc in cursor.description]
        return CardImageResponse.model_validate(dict(zip(columns, new_image_data)))

    except IntegrityError as e:
        db_conn.rollback()
        if os.path.exists(image_path): os.remove(image_path)
        if e.errno == 1062:
            raise HTTPException(
                status_code=409,
                detail=f"卡牌ID {card_id} 已存在类型为 '{image_type.value}' 的图片，请勿重复添加。"
            )
        logger.error(f"数据库完整性错误: {e}")
        raise HTTPException(status_code=500, detail="数据库操作失败。")

    except Exception as e:
        db_conn.rollback()
        if os.path.exists(image_path):
            os.remove(image_path)
        logger.error(f"关联图片到卡牌失败: {e}")
        if isinstance(e, HTTPException): raise e
        raise HTTPException(status_code=500, detail="数据库操作失败，所有更改已回滚。")
    finally:
        if cursor:
            cursor.close()


@router.get("/jsons/{id}", response_model=ImageJsonPairResponse, summary="获取图片的原始和修改后JSON")
def get_image_jsons(id: int, db_conn: PooledMySQLConnection = db_dependency):
    """获取指定图片ID的 detection_json 和 modified_json。"""
    cursor = None
    try:
        cursor = db_conn.cursor(dictionary=True)
        query = f"SELECT id, detection_json, modified_json FROM {settings.DB_IMAGE_TABLE_NAME} WHERE id = %s"
        cursor.execute(query, (id,))
        result = cursor.fetchone()
        if not result:
            raise HTTPException(status_code=404, detail=f"ID为 {id} 的图片未找到。")

        return ImageJsonPairResponse.model_validate(result)
    except Exception as e:
        logger.error(f"获取JSON对失败 ({id}): {e}")
        if isinstance(e, HTTPException): raise e
        raise HTTPException(status_code=500, detail="数据库查询失败。")
    finally:
        if cursor:
            cursor.close()


@router.put("/update/json/{id}", status_code=200, summary="修改图片的 modified_json")
def update_image_modified_json(
        id: int,
        new_json_data: Dict[str, Any],
        db_conn: PooledMySQLConnection = db_dependency
):
    """根据 id 更新 modified_json 字段。updated_at 会自动更新。"""
    cursor = None
    try:
        cursor = db_conn.cursor()
        new_json_str = json.dumps(new_json_data, ensure_ascii=False)
        query = f"UPDATE {settings.DB_IMAGE_TABLE_NAME} SET modified_json = %s WHERE id = %s"
        cursor.execute(query, (new_json_str, id))

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

        db_conn.commit()
        logger.info(f"图片ID {id} 的 modified_json 已更新。")
        return {"message": f"成功更新图片ID {id} 的JSON数据"}
    except Exception as e:
        db_conn.rollback()
        logger.error(f"更新JSON失败 ({id}): {e}")
        if isinstance(e, HTTPException): raise e
        raise HTTPException(status_code=500, detail="更新JSON数据失败。")
    finally:
        if cursor:
            cursor.close()


@router.get("/image_file/{id}", summary="获取指定ID的图片文件")
def get_image_file(id: int, db_conn: PooledMySQLConnection = db_dependency):
    """根据 id 查找记录，并返回对应的图片文件。"""
    cursor = None
    try:
        cursor = db_conn.cursor()
        query = f"SELECT image_path FROM {settings.DB_IMAGE_TABLE_NAME} WHERE id = %s"
        cursor.execute(query, (id,))
        result = cursor.fetchone()

        if not result:
            raise HTTPException(status_code=404, detail=f"ID为 {id} 的记录未找到。")

        image_path = result[0]
        absolute_path = settings.BASE_PATH / image_path.lstrip('/\\')

        if not os.path.exists(absolute_path):
            logger.error(f"文件在服务器上未找到: {absolute_path} (数据库ID: {id})")
            raise HTTPException(status_code=404, detail="图片文件在服务器上不存在。")

        return FileResponse(absolute_path)
    except Exception as e:
        logger.error(f"获取图片失败 ({id}): {e}")
        if isinstance(e, HTTPException): raise e
        raise HTTPException(status_code=500, detail="获取图片文件失败。")
    finally:
        if cursor:
            cursor.close()