修改schema

2026-03-24 10:27:14 +08:00
parent 65bf2346a3
commit 8ee0131d47
12 changed files with 350 additions and 15 deletions
@@ -74,3 +74,12 @@ class ProjectInfoResponse(BaseModel):
    # 里程碑节点
    preparation_list: list[PreparationDetail] = Field(default_factory=list, description="里程碑节点列表")

+
+class ProjectListResponse(BaseModel):
+    """
+    项目列表模型:分页
+    """
+    page: int
+    page_size: int
+    total: int
+    list: list[ProjectInfoResponse]
@@ -0,0 +1,150 @@
+"""
+一个接口按 domain 返回项目聚合数据。
+"""
+
+from typing import Literal
+
+from pydantic import BaseModel, Field
+
+from ..construction.construction import ConstructionResponse
+from ..construction.construction_detail import ConstructionDetailResponse
+from ..contract.contract import ContractResponse
+from ..contract_settlement.contract_settlement import (
+	ContractSettlementResponse,
+)
+from ..contract_settlement.contract_settlement_detail import (
+	ContractSettlementDetailResponse,
+)
+from ..material.material import MaterialResponse
+from ..material.material_detail import MaterialDetailResponse
+from ..power.power import PowerResponse
+from ..power.power_detail import PowerDetail
+from ..project.project_base import ProjectResponseBase
+from ..project_extensions.contract_change import ContractChangeResponse
+from ..project_extensions.generator_unit import GeneratorUnitProject
+from ..project_extensions.lar import (
+	LarChangeDetailSchema,
+	LarDetailSchema,
+)
+from ..project_extensions.preparation import PreparationDetail
+from ..project_extensions.production import ProductionDetail
+from ..project_extensions.project_detail import ProjectDetailProject
+from ..project_multi_type import (
+	GasTurbineExtension,
+	HydropowerExtension,
+	OffShoreWindExtension,
+	OnShoreWindExtension,
+	PhotovoltaicExtension,
+	PumpedStorageExtension,
+	ThermalPowerExtension,
+)
+
+
+ProjectAggregateDomain = Literal[
+	"project",
+	"project_detail",
+	"generator_units",
+	"lar",
+	"lar_change",
+	"contract_changes",
+	"photovoltaic_extension",
+	"onshore_wind_extension",
+	"offshore_wind_extension",
+	"hydropower_extension",
+	"pumped_storage_extension",
+	"gas_turbine_extension",
+	"thermal_power_extension",
+	"production_list",
+	"preparation_list",
+	"contracts",
+	"materials",
+	"material_details",
+	"constructions",
+	"construction_details",
+	"contract_settlements",
+	"contract_settlement_details",
+	"power",
+	"power_details",
+]
+
+
+ALL_PROJECT_AGGREGATE_DOMAINS: list[ProjectAggregateDomain] = [
+	"project",
+	"project_detail",
+	"generator_units",
+	"lar",
+	"lar_change",
+	"contract_changes",
+	"photovoltaic_extension",
+	"onshore_wind_extension",
+	"offshore_wind_extension",
+	"hydropower_extension",
+	"pumped_storage_extension",
+	"gas_turbine_extension",
+	"thermal_power_extension",
+	"production_list",
+	"preparation_list",
+	"contracts",
+	"materials",
+	"material_details",
+	"constructions",
+	"construction_details",
+	"contract_settlements",
+	"contract_settlement_details",
+	"power",
+	"power_details",
+]
+
+
+class ProjectAggregateData(BaseModel):
+	"""按 domain 组织的项目聚合数据。"""
+
+	project: ProjectResponseBase | None = Field(None, description="项目基本信息")
+	project_detail: list[ProjectDetailProject] = Field(default_factory=list, description="项目容量信息")
+	generator_units: list[GeneratorUnitProject] = Field(default_factory=list, description="发电单元信息")
+	lar: LarDetailSchema | None = Field(None, description="移民征地信息")
+	lar_change: list[LarChangeDetailSchema] = Field(default_factory=list, description="移民变更信息")
+	contract_changes: list[ContractChangeResponse] = Field(default_factory=list, description="合同变更信息")
+
+	photovoltaic_extension: PhotovoltaicExtension | None = Field(None, description="光伏项目扩展信息")
+	onshore_wind_extension: OnShoreWindExtension | None = Field(None, description="陆风项目扩展信息")
+	offshore_wind_extension: OffShoreWindExtension | None = Field(None, description="海风项目扩展信息")
+	hydropower_extension: HydropowerExtension | None = Field(None, description="水电项目扩展信息")
+	pumped_storage_extension: PumpedStorageExtension | None = Field(None, description="抽蓄项目扩展信息")
+	gas_turbine_extension: GasTurbineExtension | None = Field(None, description="燃气轮机项目扩展信息")
+	thermal_power_extension: ThermalPowerExtension | None = Field(None, description="火电项目扩展信息")
+
+	production_list: list[ProductionDetail] = Field(default_factory=list, description="投产节点列表")
+	preparation_list: list[PreparationDetail] = Field(default_factory=list, description="里程碑节点列表")
+
+	contracts: list[ContractResponse] = Field(default_factory=list, description="合同列表")
+	materials: list[MaterialResponse] = Field(default_factory=list, description="物资主表列表")
+	material_details: list[MaterialDetailResponse] = Field(default_factory=list, description="物资明细列表")
+	constructions: list[ConstructionResponse] = Field(default_factory=list, description="施工主表列表")
+	construction_details: list[ConstructionDetailResponse] = Field(default_factory=list, description="施工明细列表")
+	contract_settlements: list[ContractSettlementResponse] = Field(default_factory=list, description="合同结算主表列表")
+	contract_settlement_details: list[ContractSettlementDetailResponse] = Field(default_factory=list, description="合同结算明细列表")
+	power: list[PowerResponse] = Field(default_factory=list, description="力能主表列表")
+	power_details: list[PowerDetail] = Field(default_factory=list, description="力能明细列表")
+
+
+class ProjectAggregateMeta(BaseModel):
+	"""聚合请求回显和截断信息。"""
+
+	project_id: str = Field(..., description="项目ID")
+	requested_domains: list[ProjectAggregateDomain] = Field(
+		default_factory=lambda: list(ALL_PROJECT_AGGREGATE_DOMAINS),
+		description="本次请求的 domain 列表",
+	)
+	domain_limit: int = Field(10000, description="每个 domain 的最大返回条数")
+	truncated_domains: list[ProjectAggregateDomain] = Field(default_factory=list, description="被截断的 domain 列表")
+
+
+class ProjectAggregateResponse(BaseModel):
+	"""项目聚合同步响应。"""
+
+	meta: ProjectAggregateMeta = Field(..., description="请求元信息")
+	data: ProjectAggregateData = Field(
+		default_factory=lambda: ProjectAggregateData.model_construct(),
+		description="按 domain 分组的数据",
+	)
@@ -78,6 +78,7 @@ class ProjectResponseBase(BaseResponseWithID):
    
    # 拓展信息
    dbtc_score_list: Optional[list] = Field(default_factory=list, description='达标投产评分列表') # 这里项目用的是list而不是List
+    approval_status: Optional[str] = Field(None, description="审核状态", examples=["pending"])