from dataclasses import dataclass

from fastapi import Query
from pydantic import BaseModel, ConfigDict, Field, field_validator

from app.api.v1.module_common.file.schema import FileOutSchema
from app.common.enums import QueueEnum
from app.core.base_params import BaseQueryParam, TenantByQueryParam, UserByQueryParam
from app.core.base_schema import BaseSchema, TenantBySchema, UserBySchema


class EnergyReportTemplateCreateSchema(BaseModel):
    """能源报告模板创建模型。"""

    template_name: str = Field(..., min_length=1, max_length=128, description="模板名称")
    report_type: int = Field(default=1, ge=1, le=2, description="报告类型 1:能源报告 2:光伏报告")
    template_type: str = Field(default="", max_length=64, description="模板类型")
    version: str | None = Field(default=None, max_length=32, description="模板版本")
    file_id: int = Field(..., ge=1, description="文件记录ID")
    status: int = Field(default=0, ge=0, le=1, description="状态 0:启用 1:停用")
    order: int = Field(default=999, ge=0, description="显示排序")
    description: str | None = Field(default=None, max_length=500, description="描述")

    @field_validator("template_name")
    @classmethod
    def validate_required_text(cls, value: str) -> str:
        value = value.strip()
        if not value:
            raise ValueError("必填字段不能为空")
        return value

    @field_validator("status")
    @classmethod
    def validate_status(cls, value: int) -> int:
        if value not in {0, 1}:
            raise ValueError("状态仅支持 0(启用) 或 1(停用)")
        return value

    @field_validator("report_type")
    @classmethod
    def validate_report_type(cls, value: int) -> int:
        if value not in {1, 2}:
            raise ValueError("报告类型仅支持 1(能源报告) 或 2(光伏报告)")
        return value


class EnergyReportTemplateUpdateSchema(EnergyReportTemplateCreateSchema):
    """能源报告模板更新模型。"""


class EnergyReportTemplateOutSchema(EnergyReportTemplateCreateSchema, BaseSchema, UserBySchema, TenantBySchema):
    """能源报告模板响应模型。"""

    model_config = ConfigDict(from_attributes=True)

    file: FileOutSchema | None = Field(default=None, description="文件记录")
    file_url: str | None = Field(default=None, description="文件访问地址")
    original_name: str | None = Field(default=None, description="原始文件名")
    object_name: str | None = Field(default=None, description="MinIO对象名称")
    annotation_count: int = Field(default=0, description="标注数量")


@dataclass
class EnergyReportTemplateQueryParam(BaseQueryParam, UserByQueryParam, TenantByQueryParam):
    """能源报告模板查询参数。"""

    template_name: str | None = Query(None, description="模板名称")
    report_type: int | None = Query(None, ge=1, le=2, description="报告类型 1:能源报告 2:光伏报告")
    template_type: str | None = Query(None, description="模板类型")
    version: str | None = Query(None, description="模板版本")
    status: int | None = Query(None, ge=0, le=1, description="状态 0:启用 1:停用")

    def __post_init__(self) -> None:
        if self.template_name:
            self.template_name = (QueueEnum.like.value, self.template_name)
        if isinstance(self.report_type, int):
            self.report_type = (QueueEnum.eq.value, self.report_type)
        if self.template_type:
            self.template_type = (QueueEnum.eq.value, self.template_type)
        if self.version:
            self.version = (QueueEnum.like.value, self.version)
        if isinstance(self.status, int):
            self.status = (QueueEnum.eq.value, self.status)