from pydantic import BaseModel


class EquipmentCreate(BaseModel):
    plan_year: str
    purchasing_authority: str
    department: str
    equipment_name: str
    quantity: int
    budget_unit_price: float
    final_unit_price: float
    purchase_item_no: str
    progress: str
    delete_flag: int = 0


class Equipment(EquipmentCreate):
    id: int

    class Config:
        orm_mode = True