CardScoreDataServer/app/api/images.py

import os
import uuid
import json
import requests
from typing import Optional, Dict, Any
from fastapi import APIRouter, File, UploadFile, Depends, HTTPException, Form, Path
from fastapi.concurrency import run_in_threadpool
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, ResultImagePathType
from app.utils.scheme import ImageType, IMAGE_TYPE_TO_SCORE_TYPE
from app.core.database_loader import get_db_connection
from app.crud import crud_card

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


@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="图片类型"),
        image: UploadFile = File(..., description="图片文件"),
        image_name: Optional[str] = Form(None, description="图片的可选名称"),
        json_data_str: str = Form(..., description="与图片关联的JSON字符串"),
        db_conn: PooledMySQLConnection = db_dependency
):
    """
    上传一张主要图片 (coaxial/ring)。
    """
    # 校验：此接口只能传主图类型
    if image_type in [ImageType.front_gray, ImageType.back_gray]:
        raise HTTPException(status_code=400, detail="此接口不支持灰度图上传，请使用 /insert/gray/{card_id}")

    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()

        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}。")

        try:
            crud_card.update_card_scores_and_status(db_conn, card_id)
        except Exception as score_update_e:
            logger.error(f"更新卡牌 {card_id} 分数失败: {score_update_e}")

        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}' 的图片。")
        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.post("/insert/gray/{card_id}", response_model=CardImageResponse, status_code=201,
             summary="[NEW] 为卡牌上传辅助灰度图")
async def upload_gray_image_for_card(
        card_id: int = Path(..., description="要关联的卡牌ID"),
        image_type: ImageType = Form(..., description="图片类型 (必须是 front_gray 或 back_gray)"),
        image: UploadFile = File(..., description="灰度图片文件"),
        db_conn: PooledMySQLConnection = db_dependency
):
    """
    上传辅助灰度图 (front_gray / back_gray)。
    不需要 JSON 数据，不参与直接计算。
    """
    if image_type not in [ImageType.front_gray, ImageType.back_gray]:
        raise HTTPException(status_code=400, detail="此接口仅支持灰度图 (front_gray, back_gray)")

    # 1. 保存文件
    file_extension = os.path.splitext(image.filename)[1]
    unique_filename = f"gray_{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()

        # 2. 检查卡牌存在
        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} 的卡牌不存在。")

        # 3. 插入数据 (card_gray_images 表)
        query_insert = (
            f"INSERT INTO {settings.DB_GRAY_IMAGE_TABLE_NAME} "
            "(card_id, image_type, image_path) VALUES (%s, %s, %s)"
        )
        cursor.execute(query_insert, (card_id, image_type.value, relative_path))
        new_id = cursor.lastrowid
        db_conn.commit()

        logger.info(f"灰度图 {new_id} 已关联到卡牌 {card_id}, 类型 {image_type.value}")

        # 4. 构造返回 (模拟 CardImageResponse 结构)
        # 因为数据库里只有基础字段，这里要补全 Pydantic 模型需要的字段
        cursor.execute(f"SELECT * FROM {settings.DB_GRAY_IMAGE_TABLE_NAME} WHERE id = %s", (new_id,))
        new_data = cursor.fetchone()  # tuple
        # 获取列名
        columns = [desc[0] for desc in cursor.description]
        row_dict = dict(zip(columns, new_data))

        # 补全空字段以符合 CardImageResponse
        from app.crud.crud_card import EMPTY_DETECTION_JSON
        response_dict = {
            **row_dict,
            "detection_json": EMPTY_DETECTION_JSON,  # 默认死值
            "modified_json": None,  # 刚上传还没有 modified
            "image_name": None,
            "detection_image_path": None,
            "modified_image_path": None,
            "is_edited": False
        }

        return CardImageResponse.model_validate(response_dict)

    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}' 的图片。")
        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):
    # 此接口主要用于主图，灰度图没有实体的 JSON 存储，暂不支持直接通过此 ID 查询 JSON
    # 如果前端通过 format_xy 里的 query 接口获取，已经能在 list 里拿到了。
    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()