import datetime

from schemas.base import CustomModel
from schemas.client import ClientSchema


class DealChangeStatusRequest(CustomModel):
    deal_id: int
    new_status: int


class DealChangeStatusResponse(CustomModel):
    ok: bool


class DealCreateRequest(CustomModel):
    name: str


class DealQuickCreateRequest(CustomModel):
    name: str
    client_name: str
    client_address: str
    comment: str
    acceptance_date: datetime.datetime


class DealQuickCreateResponse(CustomModel):
    deal_id: int


class DealCreateResponse(CustomModel):
    ok: bool


class FastDeal(CustomModel):
    name: str
    client: ClientSchema
    comment: str
    acceptance_date: datetime.datetime