from pydantic import Field

from ...common.sync_node import SyncNode
from schemas.project.project_base import ProjectResponseBase


class ProjectSyncSchema(ProjectResponseBase):
    """项目同步内部 schema，补充 all_info 扩展字段。"""

    implementation_estimate: float | None = Field(None, description="执行概算")
    static_total_investment: float | None = Field(None, description="静态总投资")
    completed_investment: float | None = Field(None, description="已完成投资")
    total_contract_quantity: float | None = Field(None, description="合同总量")
    completed_settlement_quantity: float | None = Field(None, description="已结算量")


class ProjectSyncNode(SyncNode[ProjectSyncSchema]):
    """项目同步节点"""
    node_type = "project"
    schema = ProjectSyncSchema
    
    def __init__(self, **kwargs):
        super().__init__(**kwargs)