from datetime import datetime

from pydantic import BaseModel


class FeedWaitingTimeIn(BaseModel):
    id: int
    vn: int | None = None
    txn: int | None = None
    hn: str | None = None
    name: str | None = None
    doctor_code: str | None = None
    doctor_name: str | None = None
    location_code: str | None = None
    location_name: str | None = None
    step_name: str | None = None
    time: datetime


class FeedCheckpointIn(BaseModel):
    id: int | None = None
    hn: int
    vn: int
    location: str
    type: str
    timestamp_in: datetime
    timestamp_out: datetime | None = None
    waiting_time: int | None = None
    bu: str | None = None