153 lines
4.8 KiB
Python
153 lines
4.8 KiB
Python
"""
|
|
资产分配相关的Pydantic Schema
|
|
"""
|
|
from typing import Optional, List, Dict, Any
|
|
from datetime import datetime, date
|
|
from decimal import Decimal
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
# ===== 分配单Schema =====
|
|
|
|
class AllocationOrderBase(BaseModel):
|
|
"""分配单基础Schema"""
|
|
order_type: str = Field(..., description="单据类型(allocation/transfer/recovery/maintenance/scrap)")
|
|
title: str = Field(..., min_length=1, max_length=200, description="标题")
|
|
source_organization_id: Optional[int] = Field(None, gt=0, description="调出网点ID")
|
|
target_organization_id: int = Field(..., gt=0, description="调入网点ID")
|
|
expect_execute_date: Optional[date] = Field(None, description="预计执行日期")
|
|
remark: Optional[str] = Field(None, description="备注")
|
|
|
|
|
|
class AllocationOrderCreate(AllocationOrderBase):
|
|
"""创建分配单Schema"""
|
|
asset_ids: List[int] = Field(..., min_items=1, description="资产ID列表")
|
|
|
|
|
|
class AllocationOrderUpdate(BaseModel):
|
|
"""更新分配单Schema"""
|
|
title: Optional[str] = Field(None, min_length=1, max_length=200)
|
|
expect_execute_date: Optional[date] = None
|
|
remark: Optional[str] = None
|
|
|
|
|
|
class AllocationOrderApproval(BaseModel):
|
|
"""分配单审批Schema"""
|
|
approval_status: str = Field(..., description="审批状态(approved/rejected)")
|
|
approval_remark: Optional[str] = Field(None, description="审批备注")
|
|
|
|
|
|
class AllocationOrderExecute(BaseModel):
|
|
"""分配单执行Schema"""
|
|
remark: Optional[str] = Field(None, description="执行备注")
|
|
|
|
|
|
class AllocationOrderInDB(BaseModel):
|
|
"""数据库中的分配单Schema"""
|
|
id: int
|
|
order_code: str
|
|
order_type: str
|
|
title: str
|
|
source_organization_id: Optional[int]
|
|
target_organization_id: int
|
|
applicant_id: int
|
|
approver_id: Optional[int]
|
|
approval_status: str
|
|
approval_time: Optional[datetime]
|
|
approval_remark: Optional[str]
|
|
expect_execute_date: Optional[date]
|
|
actual_execute_date: Optional[date]
|
|
executor_id: Optional[int]
|
|
execute_status: str
|
|
remark: Optional[str]
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class AllocationOrderResponse(AllocationOrderInDB):
|
|
"""分配单响应Schema"""
|
|
pass
|
|
|
|
|
|
class AllocationOrderWithRelations(AllocationOrderResponse):
|
|
"""带关联信息的分配单响应Schema"""
|
|
source_organization: Optional[Dict[str, Any]] = None
|
|
target_organization: Optional[Dict[str, Any]] = None
|
|
applicant: Optional[Dict[str, Any]] = None
|
|
approver: Optional[Dict[str, Any]] = None
|
|
executor: Optional[Dict[str, Any]] = None
|
|
items: Optional[List[Dict[str, Any]]] = None
|
|
|
|
|
|
class AllocationOrderListResponse(BaseModel):
|
|
"""分配单列表响应Schema"""
|
|
total: int
|
|
page: int
|
|
page_size: int
|
|
total_pages: int
|
|
items: List[AllocationOrderWithRelations]
|
|
|
|
|
|
# ===== 分配单明细Schema =====
|
|
|
|
class AllocationItemBase(BaseModel):
|
|
"""分配单明细基础Schema"""
|
|
asset_id: int = Field(..., gt=0, description="资产ID")
|
|
remark: Optional[str] = Field(None, description="备注")
|
|
|
|
|
|
class AllocationItemInDB(BaseModel):
|
|
"""数据库中的分配单明细Schema"""
|
|
id: int
|
|
order_id: int
|
|
asset_id: int
|
|
asset_code: str
|
|
asset_name: str
|
|
from_organization_id: Optional[int]
|
|
to_organization_id: Optional[int]
|
|
from_status: Optional[str]
|
|
to_status: Optional[str]
|
|
execute_status: str
|
|
execute_time: Optional[datetime]
|
|
failure_reason: Optional[str]
|
|
remark: Optional[str]
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class AllocationItemResponse(AllocationItemInDB):
|
|
"""分配单明细响应Schema"""
|
|
pass
|
|
|
|
|
|
# ===== 查询参数Schema =====
|
|
|
|
class AllocationOrderQueryParams(BaseModel):
|
|
"""分配单查询参数"""
|
|
order_type: Optional[str] = Field(None, description="单据类型")
|
|
approval_status: Optional[str] = Field(None, description="审批状态")
|
|
execute_status: Optional[str] = Field(None, description="执行状态")
|
|
applicant_id: Optional[int] = Field(None, gt=0, description="申请人ID")
|
|
target_organization_id: Optional[int] = Field(None, gt=0, description="目标网点ID")
|
|
keyword: Optional[str] = Field(None, description="搜索关键词")
|
|
page: int = Field(default=1, ge=1, description="页码")
|
|
page_size: int = Field(default=20, ge=1, le=100, description="每页数量")
|
|
|
|
|
|
# ===== 统计Schema =====
|
|
|
|
class AllocationOrderStatistics(BaseModel):
|
|
"""分配单统计Schema"""
|
|
total: int = Field(..., description="总数")
|
|
pending: int = Field(..., description="待审批数")
|
|
approved: int = Field(..., description="已审批数")
|
|
rejected: int = Field(..., description="已拒绝数")
|
|
executing: int = Field(..., description="执行中数")
|
|
completed: int = Field(..., description="已完成数")
|