import uuid
from typing import Optional

import pydantic
from pydantic import Field

from core.models.base import BaseModel


class NewPriceModel(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='بهای روز')
    fee_unit: Optional[int] = Field(description='بهای واحد مبنای پیمان')
    total_day: Optional[int] = Field(description=' بهای کل روز = تعداد * بهای واحد روز')
    total_unit: 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=" شماره آیتم")