Spaces:
Sleeping
Sleeping
File size: 2,432 Bytes
abd17e2 d37f3cd 32ddb83 d37f3cd abd17e2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 |
from pydantic import BaseModel, validator
from datetime import datetime, timezone
# ------------------------------------------------------------------------------
class Attendee(BaseModel, extra="forbid"):
email: str
name: str
# ------------------------------------------------------------------------------
class Appointment(BaseModel, extra="forbid"):
start_time: datetime
end_time: datetime
# To compare two Appointment objects
def __hash__(self):
return hash((self.start_time, self.end_time))
# ------------------------------------------------------------------------------
class CreateParams(BaseModel, extra="forbid"):
appointment: Appointment
subject: str
recruiter: Attendee
client: Attendee
candidate: Attendee
# ------------------------------------------------------------------------------
class SendClientParams(BaseModel, extra="forbid"):
client_email: str
candidate_name: str
job_title: str
# ------------------------------------------------------------------------------
class AppointmentDBItem(BaseModel, extra="forbid"):
id: str
zoom_meeting_id: str
process_id: str
job_title: str
start_time: str
end_time: str
meeting_url: str
environment: str
process_status: str
recruiter: Attendee
client: Attendee
candidate: Attendee
summary_recruiter: str
summary_client: str
summary_candidate: str
# ------------------------------------------------------------------------------
class SendCandidateParams(BaseModel, extra="forbid"):
appointment_suggestions: list[Appointment]
candidate_email: str
job_title: str
@validator("appointment_suggestions")
def check_length(cls, appointment_list):
if len(appointment_list) < 3:
raise ValueError(
"appointment_suggestions must contain at least 3 items"
)
if len(appointment_list) != len(set(appointment_list)):
raise ValueError("All suggestions must be unique.")
current_datetime = datetime.now(timezone.utc)
if not all(
appointment.start_time > current_datetime
and appointment.end_time > current_datetime
for appointment in appointment_list
):
raise ValueError(
"All suggestions must refer to appointments in the future."
)
return list
|