from pydantic import Field
from app.utils.schemas_utils import CustomModel
from enum import Enum as PyEnum

class EquipmentTypeEnum(PyEnum):
    SPECIALISED = "Specialised"
    UNIVERSAL = "Universal"

class CommercialEquipmentCreate(CustomModel):
    format_type: str
    type_of_equipment: str
    #store_format_type: StoreFormat
    number_of_units: int
    type_of_equipment:str
    cost_per_unit: float
    equipment_category: EquipmentTypeEnum
    remarks: str
    group_id: int

class CommercialEquipmentResponse(CustomModel):
    equipment_id: int | None = Field(default=None)
    group_id: int | None = Field(default=None)
    #store_format_type: str | None = Field(default=None) 
    format_type: str | None = Field(default=None)
    type_of_equipment: str | None = Field(default=None)
    number_of_units: int | None = Field(default=None)
    cost_per_unit: float | None = Field(default=None)
    equipment_category:  str | None = Field(default=None)
    total: float | None = Field(default=None)
    remarks: str | None = Field(default=None)