from .BaseSchemas import *
from datetime import datetime
from typing import Optional
from enums.enums import AppointmentStatus


# Create schemas (used for creating new records)
class ClinicCreate(ClinicBase):
    pass


class DoctorCreate(DoctorBase):
    pass


class PatientCreate(PatientBase):
    pass


class AppointmentCreate(AppointmentBase):
    pass


class CalendarCreate(CalendarBase):
    pass


class AppointmentCreateWithNames(BaseModel):
    doctor_name: str
    patient_name: str
    appointment_time: datetime
    status: AppointmentStatus = AppointmentStatus.CONFIRMED


class UserCreate(BaseModel):
    # User data sent from frontend
    user: UserBase
    # Clinic data sent from frontend
    clinic: ClinicBase