IntelligentRailwayCosting/SourceCode/IntelligentRailwayCosting/app/core/dtos/project_quota.py

from pydantic import BaseModel
from typing import Optional
from datetime import datetime

from core.models import ProjectQuotaModel


class ProjectQuotaDto(BaseModel):
    """项目定额DTO"""

    id: Optional[int] = None
    task_id: Optional[int] = None
    project_id: str
    budget_id: int
    budget_code: Optional[str] = None
    item_id: int
    item_code: str
    quota_id: Optional[int] = 0
    quota_code: Optional[str] = None
    quota_adjustment: Optional[str] = None
    entry_name: Optional[str] = None
    units: Optional[str] = None
    amount: Optional[float] = None
    ex_file: Optional[str] = None
    ex_cell: Optional[str] = None
    ex_row: Optional[str] = None
    ex_unit: Optional[str] = None
    ex_amount: Optional[float] = None
    note: Optional[str] = None
    is_change: int = 1
    send_status: int = 0
    send_time: Optional[datetime] = None
    send_error: Optional[str] = None
    created_at: Optional[datetime] = None
    created_by: Optional[str] = None
    updated_at: Optional[datetime] = None
    updated_by: Optional[str] = None
    sequence_number: Optional[int] = None

    @classmethod
    def from_model(cls, model: ProjectQuotaModel) -> "ProjectQuotaDto":
        """从数据库模型创建DTO对象"""
        return cls(
            id=model.id,
            task_id=model.task_id,
            project_id=model.project_id,
            budget_id=model.budget_id,
            budget_code=model.budget_code,
            item_id=model.item_id,
            item_code=model.item_code,
            quota_id=model.quota_id,
            quota_code=model.quota_code,
            quota_adjustment=model.quota_adjustment,
            entry_name=model.entry_name,
            units=model.units,
            amount=model.amount,
            ex_file=model.ex_file,
            ex_cell=model.ex_cell,
            ex_row=model.ex_row,
            ex_unit=model.ex_unit,
            ex_amount=model.ex_amount,
            note=model.note,
            is_change=model.is_change,
            send_status=model.send_status,
            send_time=model.send_time,
            send_error=model.send_error,
            created_at=model.created_at,
            created_by=model.created_by,
            updated_at=model.updated_at,
            updated_by=model.updated_by,
        )

    def to_dict(self) -> dict:
        """转换为字典格式"""
        return self.model_dump()

    @classmethod
    def ai_prompt_struct(cls):
        return """```typescript
        export interface input_struct {  //输入数据结构体
            i: // ID
            n: string; //工程或费用项目名称
            u：string; //单位
            q: float; //数量
        }
        export interface item { //需返回的结构体
            i: int; // ID 与提供的ID保持一致
            q: string; // 定额编号
            u：string; // 定额单位
            p: float; // 定额单价
            tp: float; // 总价
        }
        ```
        """

    @classmethod
    def from_ai_dict(cls, data: dict):
        return cls(
            project_id="",
            budget_id=0,
            item_id=0,
            item_code="",
            id=data.get("i", 0),
            quota_code=data.get("q", ""),
            units=data.get("u", ""),
        )

    def to_ai_dict(self):
        return {
            "i": self.id,
            "n": self.entry_name,
            "u": self.unit,
            "q": self.project_quantity,
        }

    class Config:
        from_attributes = True