import uuid
from typing import Optional

import pydantic
from pydantic import Field

from core.models.base import BaseModel


class FactorModel(BaseModel):
    id: Optional[uuid.UUID] = Field(alias='_id')
    project_id: uuid.UUID = Field(description='کد شناسایی پروژ')
    number: Optional[str] = Field(description=" شماره آیتم")
    description: Optional[str] = Field(description='موضوع فاکتور')
    unit: Optional[str] = Field(description='واحد')
    count: Optional[int] = Field(description='تعداد')
    fee_day: Optional[int] = Field(description='بهای روز')
    total_day: Optional[int] = Field(description=' بهای کل روز = تعداد * بهای واحد روز')
    field: Optional[str] = Field(description="""
                'electrical',
                'mechanic',
                'building',
                'other'
            """)
    location: Optional[str] = Field(description='موقعیت')

class CheckModel(pydantic.BaseModel):
    project_id: uuid.UUID = Field(description='کد شناسایی پروژ')
    number: Optional[str] = Field(description=" شماره آیتم")