first commit

schemas/project_extensions/generator_unit.py

@@ -0,0 +1,34 @@
+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