from datetime import datetime
from pydantic import BaseModel, Field
from typing import Optional


class ProjectTrainOut(BaseModel):
    """项目训练版本信息表"""
    id: Optional[int] = Field(None, description="训练id")
    train_version: Optional[str] = Field(None, description="训练版本号")
    create_time: Optional[datetime] = Field(None, description="训练时间")

    class Config:
        orm_mode = True
        json_encoders = {
            datetime: lambda v: v.strftime("%Y-%m-%d %H:%M:%S")
        }