first commit

2026-03-09 16:31:42 +08:00
commit 4a9c444b10
486 changed files with 52479 additions and 0 deletions
@@ -0,0 +1,122 @@
+from pydantic import BaseModel, Field,field_validator
+from typing import List, Optional, Literal
+from ..common.datetime import datetime_date,get_current_date
+from fastapi import HTTPException
+from ..common.validators import DateStr
+from ..common.base import BaseResponseWithID
+
+class ContractResponse(BaseResponseWithID):
+    """
+    合同返回信息
+    """
+    id : str = Field(..., description="合同ID")
+    project_id: str = Field("", description="所属项目ID")
+    name: str = Field("", description="合同名称")
+    short_name: str = Field("", description="合同简称")
+    code: str = Field("", description="合同编码")
+    contract_type: str = Field("", description="合同类型")
+
+    currency: str = Field("CNY", description="币种")
+    manager_name: str = Field("", description="负责人姓名")
+    manager_phone: str = Field("", description="负责人电话")
+
+    sign_company: str = Field("", description="签订公司")
+    company_id: str = Field("", description="签订公司id")
+    sign_date: DateStr = Field("", description="签订日期", examples=["2023-10-01"])
+    credit_code: Optional[str] = Field("", description="签订公司信用代码")
+
+    total_price: float = Field(0.0, description="合同金额")
+    sub_name: Optional[str] = Field(None, description="合同子标题")
+
+
+class ContractCreate(BaseModel):
+    """
+    创建合同
+    CONTRACT = "contract"
+    """
+    name: str|None = Field(None, description="合同名称")
+    sub_name:str|None= Field(None, description="合同子标题")
+    code: str = Field("", description="合同编码")
+    contract_type: str = Field("", description="合同类型")
+
+    project_id: str = Field("", description="所属项目ID")
+
+    currency: str = Field("CNY", description="币种")
+    manager_name: str = Field("", description="负责人姓名")
+    manager_phone: str = Field("", description="负责人电话")
+
+    sign_company: str = Field("", description="签订公司")
+    company_id: Optional[str] = Field(None, description="签订公司id")
+    sign_date: DateStr = Field("", description="签订日期", examples=["2023-10-01"])
+    credit_code: Optional[str] = Field(None, description="签订公司信用代码")
+
+    total_price: float = Field(0.0, description="合同金额")
+
+    @field_validator("total_price")
+    def validate_price(cls, v):
+        """验证金额必须为非负数"""
+        if v < 0:
+            raise HTTPException(status_code=400, detail='合同金额不能小于0')
+        return v
+
+    @field_validator('sign_date')
+    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
+
+class ContractUpdate(BaseModel):
+    """
+    更新合同
+    CONTRACT = "contract"
+    """
+    id : str = Field(..., description="合同ID")
+    sub_name:str|None= Field(None, description="合同子标题")
+
+    currency: str|None = Field(None, description="币种")
+
+    manager_name: str|None= Field(None, description="负责人姓名")
+    manager_phone: str|None= Field(None, description="负责人电话")
+
+    sign_company: str|None= Field(None, description="签订公司")
+    company_id: Optional[str] = Field(None, description="签订公司id")
+    sign_date: str|None = Field(None, description="签订日期", examples=["2023-10-01"])
+    credit_code: Optional[str] = Field(None, description="签订公司信用代码")
+
+    total_price: float|None= Field(None, description="合同金额")
+
+    @field_validator("total_price")
+    def validate_price(cls, v):
+        """验证金额必须为非负数"""
+        if v is None:
+            return None
+        if v < 0:
+            raise HTTPException(status_code=400, detail='合同金额不能小于0')
+        return v
+
+    @field_validator('sign_date')
+    def validate_actual_date_not_future(cls, v: str):
+        if v is None:
+            return None
+        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
+
+
+