from pydantic import BaseModel as PydanticBaseModel, ConfigDict

from app.enums import AdReportingPlatform


class BaseModel(PydanticBaseModel):
    model_config = ConfigDict(
        alias_generator=lambda field_name: field_name.upper(),
        populate_by_name=True,
    )


class Creative(BaseModel):
    platform: AdReportingPlatform
    creative_id: str
    source_url: str
    fails_count: int


class AdReportingCreativeThumbnailUrl(BaseModel):
    creative_id: str
    platform: str
    source_url: str
    img_s3_obj_key: str | None = None
    timg64x64_s3_obj_key: str | None = None
    fails_count: int