strepsiades
34 lines
1.5 KiB
Python
34 lines
1.5 KiB
Python
from pydantic import BaseModel, Field,field_validator
|
|
from typing import Optional
|
|
from ..common.base import BaseResponseWithID
|
|
from ..common.datetime import datetime_date,get_current_date
|
|
from fastapi import HTTPException
|
|
|
|
class GeneratorUnitProject(BaseResponseWithID):
|
|
project_id : Optional[str]= Field(None, description="项目ID")
|
|
serial_number: Optional[int] = Field(None, description="机组序号")
|
|
approval_status:Optional[str]=Field(None, description="审批状态")
|
|
capacity: Optional[float] = Field(None, description="单机容量(MW)")
|
|
plan_production_date: Optional[str] = Field(None, description="计划投产日期")
|
|
actual_production_date: Optional[str] = Field(None, description="实际投产日期")
|
|
|
|
class PostGeneratorUnits(BaseModel):
|
|
model_config = {"populate_by_name": True}
|
|
|
|
id:str = Field(..., description="机组ID")
|
|
project_id: str = Field(..., description="所属项目ID")
|
|
actual_production_date: str = Field(..., alias="actual_date", description="实际投产日期(格式：YYYY-MM-DD)")
|
|
|
|
@field_validator('actual_production_date', mode="before")
|
|
def validate_actual_date_not_future(cls, v: str) -> str:
|
|
if v == "":
|
|
return ""
|
|
try:
|
|
actual = datetime_date(v)
|
|
except ValueError:
|
|
raise HTTPException(status_code=400,detail='日期格式为(YYYY-MM-DD)')
|
|
|
|
today = datetime_date(get_current_date())
|
|
if actual > today:
|
|
raise HTTPException(status_code=400,detail='机组日期不能晚于今天')
|
|
return v |