CardScoreDataServer/app/utils/scheme.py

import json
from datetime import datetime
from pathlib import Path
from typing import Optional, Dict, Any, List
from pydantic import BaseModel, field_validator, Field, field_serializer
from enum import Enum


class ImageType(str, Enum):
    front_coaxial = "front_coaxial"
    back_coaxial = "back_coaxial"
    front_ring = "front_ring"
    back_ring = "back_ring"

    # 灰度图类型
    front_gray = "front_gray"
    back_gray = "back_gray"
    
    # 融合图类型
    front_fusion = "front_fusion"
    back_fusion = "back_fusion"


class CardType(str, Enum):
    pokemon = "pokemon"
    basketball = "basketball"
    football = "football"


class ResultImagePathType(str, Enum):
    detection = "detection"
    modified = "modified"


class SortBy(str, Enum):
    card_name = "card_name"
    created_at = "created_at"
    updated_at = "updated_at"
    detection_score = "detection_score"
    modified_score = "modified_score"


class SortOrder(str, Enum):
    asc = "ASC"
    desc = "DESC"


class CardNoList(BaseModel):
    cardNoList: List[str]


# 图片类型和推理服务 score_type 映射表
IMAGE_TYPE_TO_SCORE_TYPE = {
    "front_coaxial": "front_face",
    "back_coaxial": "back_face",
    "front_ring": "front_corner_edge",
    "back_ring": "back_corner_edge",
    "front_gray": None,
    "back_gray": None,
    "front_fusion": None,
    "back_fusion": None
}


# --- Pydantic 数据模型 ---

class CardImageResponse(BaseModel):
    """用于API响应的图片数据模型 (主键为 id)"""
    id: int
    card_id: int
    image_type: str
    image_name: Optional[str] = None
    image_path: str
    # Gray images need these fields to be Optional/None
    detection_image_path: Optional[str] = None
    modified_image_path: Optional[str] = None
    detection_json: Dict[str, Any]
    modified_json: Optional[Dict[str, Any]] = None
    is_edited: bool = False
    created_at: datetime
    updated_at: datetime

    class Config:
        from_attributes = True

    @field_validator('detection_json', 'modified_json', mode='before')
    @classmethod
    def parse_json_string(cls, v):
        if v is None:
            return None
        if isinstance(v, str):
            try:
                return json.loads(v)
            except json.JSONDecodeError:
                raise ValueError("Invalid JSON string in database")
        return v

        # 拦截序列化，格式化掉中间的 "T"

    @field_serializer('created_at', 'updated_at', check_fields=False)
    def serialize_datetime(self, dt: datetime, _info):
        if dt is not None:
            return dt.strftime('%Y-%m-%d %H:%M:%S')
        return dt


class CardDetailResponse(BaseModel):
    """用于响应单个卡牌详细信息的模型 (主键为 id)"""
    id: int
    id_prev: Optional[int] = None
    id_next: Optional[int] = None
    card_name: Optional[str] = None
    cardNo: Optional[str] = None
    created_at: datetime
    updated_at: datetime
    card_type: str
    is_edited: bool
    review_state: int = 1
    detection_score: Optional[float] = None
    modified_score: Optional[float] = None
    detection_score_detail: Optional[Dict[str, Any]] = None
    modified_score_detail: Optional[Dict[str, Any]] = None
    images: List[CardImageResponse] = []

    class Config:
        from_attributes = True

    # 拦截序列化，格式化掉中间的 "T"
    @field_serializer('created_at', 'updated_at', check_fields=False)
    def serialize_datetime(self, dt: datetime, _info):
        if dt is not None:
            return dt.strftime('%Y-%m-%d %H:%M:%S')
        return dt


class ImageJsonPairResponse(BaseModel):
    id: int
    detection_json: Dict[str, Any]
    modified_json: Optional[Dict[str, Any]] = None

    class Config:
        from_attributes = True

    @field_validator('detection_json', 'modified_json', mode='before')
    @classmethod
    def parse_json_string(cls, v):
        if v is None:
            return None
        if isinstance(v, str):
            try:
                return json.loads(v)
            except json.JSONDecodeError:
                raise ValueError("Invalid JSON string in database")
        return v


class CardImageInListResponse(BaseModel):
    id: int
    image_type: str
    image_path: str
    detection_image_path: Optional[str] = None
    modified_image_path: Optional[str] = None

    class Config:
        from_attributes = True


class CardListDetailResponse(BaseModel):
    id: int
    card_name: Optional[str] = None
    cardNo: Optional[str] = None
    card_type: str
    detection_score: Optional[float] = None
    modified_score: Optional[float] = None
    is_edited: bool
    review_state: int = 1
    created_at: datetime
    updated_at: datetime
    image_path_list: Dict[str, Optional[str]] = {}
    detection_image_path_list: Dict[str, Optional[str]] = {}
    modified_image_path_list: Dict[str, Optional[str]] = {}
    is_bound: bool = False

    class Config:
        from_attributes = True

        # 拦截序列化，格式化掉中间的 "T"

    @field_serializer('created_at', 'updated_at', check_fields=False)
    def serialize_datetime(self, dt: datetime, _info):
        if dt is not None:
            return dt.strftime('%Y-%m-%d %H:%M:%S')
        return dt


class CardListWithTotal(BaseModel):
    total: int
    list: List[CardListDetailResponse]


class CardListResponseWrapper(BaseModel):
    data: CardListWithTotal


class ReviewUpdate(BaseModel):
    review_state: int = Field(..., ge=1, le=4, description="审核状态 (1待复检, 2已复检, 3审核未通过, 4审核通过)")


class CardTable(BaseModel):
    id: int
    card_name: Optional[str] = None
    cardNo: Optional[str] = None
    card_type: str = "pokemon"
    detection_score: Optional[float] = None
    modified_score: Optional[float] = None
    is_edited: bool = False
    created_at: datetime
    updated_at: datetime
    review_state: int = 1

    class Config:
        from_attributes = True

    @field_serializer("created_at", "updated_at", check_fields=False)
    def serialize_datetime(self, dt: datetime, _info):
        if dt is not None:
            return dt.strftime("%Y-%m-%d %H:%M:%S")
        return dt


class CardImageTable(BaseModel):
    id: int
    card_id: int
    image_type: str
    image_name: Optional[str] = None
    image_path: str
    detection_image_path: Optional[str] = None
    modified_image_path: Optional[str] = None
    detection_json: Dict[str, Any]
    modified_json: Optional[Dict[str, Any]] = None
    is_edited: bool = False
    created_at: datetime
    updated_at: datetime

    class Config:
        from_attributes = True

    @field_validator("detection_json", "modified_json", mode="before")
    @classmethod
    def parse_json_string(cls, v):
        if v is None:
            return None
        if isinstance(v, str):
            try:
                return json.loads(v)
            except json.JSONDecodeError:
                raise ValueError("Invalid JSON string in database")
        return v

    @field_serializer("created_at", "updated_at", check_fields=False)
    def serialize_datetime(self, dt: datetime, _info):
        if dt is not None:
            return dt.strftime("%Y-%m-%d %H:%M:%S")
        return dt


class CardGrayImageTable(BaseModel):
    id: int
    card_id: int
    image_type: str
    image_path: str
    created_at: datetime
    updated_at: datetime

    class Config:
        from_attributes = True

    @field_serializer("created_at", "updated_at", check_fields=False)
    def serialize_datetime(self, dt: datetime, _info):
        if dt is not None:
            return dt.strftime("%Y-%m-%d %H:%M:%S")
        return dt