strepsiades
123 lines
4.5 KiB
Python
123 lines
4.5 KiB
Python
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
|
|
|
|
|
|
|