feat: initial commit
This commit is contained in:
@@ -0,0 +1,25 @@
|
||||
from typing import Any, Optional, TypeVar, Generic
|
||||
from pydantic import Field
|
||||
from pydantic import BaseModel
|
||||
|
||||
from exceptions import ApiException
|
||||
|
||||
|
||||
T = TypeVar('T')
|
||||
|
||||
class ApiResponse(BaseModel, Generic[T]):
|
||||
"""Standard API response model matching Node.js implementation."""
|
||||
|
||||
data: Optional[T] = Field(default=None, description="Response data")
|
||||
error: Optional[Any] = Field(default=None, description="Error details")
|
||||
message: Optional[str] = Field(default=None, description="Response message")
|
||||
|
||||
@classmethod
|
||||
def from_api_exception(cls, exception: ApiException) -> dict:
|
||||
"""Create an API response from an API exception."""
|
||||
import traceback
|
||||
return cls(
|
||||
data=None,
|
||||
message=exception.message,
|
||||
error=traceback.format_exc() if exception else None
|
||||
).model_dump(exclude_none=True)
|
||||
@@ -0,0 +1,52 @@
|
||||
# schemas.py
|
||||
from datetime import datetime
|
||||
from typing import List, Optional
|
||||
from pydantic import BaseModel, EmailStr
|
||||
from enums.enums import AppointmentStatus, ClinicUserRoles, UserType
|
||||
|
||||
|
||||
# Base schemas (shared attributes for create/read operations)
|
||||
class ClinicBase(BaseModel):
|
||||
name: str
|
||||
address: Optional[str] = None
|
||||
phone: str
|
||||
email: Optional[EmailStr] = None
|
||||
|
||||
|
||||
class DoctorBase(BaseModel):
|
||||
name: str
|
||||
age: Optional[int] = None
|
||||
email: Optional[EmailStr] = None
|
||||
phone: str
|
||||
address: Optional[str] = None
|
||||
clinic_id: int
|
||||
|
||||
|
||||
class PatientBase(BaseModel):
|
||||
name: str
|
||||
age: Optional[int] = None
|
||||
email: Optional[EmailStr] = None
|
||||
phone: Optional[str] = None
|
||||
address: Optional[str] = None
|
||||
dob: Optional[str] = None
|
||||
|
||||
|
||||
class AppointmentBase(BaseModel):
|
||||
doctor_id: int
|
||||
patient_id: int
|
||||
appointment_time: datetime
|
||||
status: AppointmentStatus = AppointmentStatus.CONFIRMED
|
||||
|
||||
|
||||
class CalendarBase(BaseModel):
|
||||
doc_id: int
|
||||
# rrule: str # Recurrence rule in iCalendar format
|
||||
time: str
|
||||
|
||||
|
||||
class UserBase(BaseModel):
|
||||
username: str
|
||||
email: EmailStr
|
||||
password: str
|
||||
clinicRole: Optional[ClinicUserRoles] = None
|
||||
userType: Optional[UserType] = None
|
||||
@@ -0,0 +1,36 @@
|
||||
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(UserBase):
|
||||
pass
|
||||
@@ -0,0 +1,126 @@
|
||||
from datetime import datetime
|
||||
from typing import List
|
||||
from .BaseSchemas import *
|
||||
from pydantic import Field
|
||||
|
||||
# Response schemas (used for API responses)
|
||||
class Clinic(ClinicBase):
|
||||
id: int
|
||||
create_time: datetime
|
||||
update_time: datetime
|
||||
|
||||
class Config:
|
||||
orm_mode = True
|
||||
|
||||
class UserResponse(UserBase):
|
||||
id: int
|
||||
create_time: datetime
|
||||
update_time: datetime
|
||||
password: str = Field(exclude=True)
|
||||
|
||||
class Config:
|
||||
orm_mode = True
|
||||
allow_population_by_field_name = True
|
||||
|
||||
class Doctor(DoctorBase):
|
||||
id: int
|
||||
create_time: datetime
|
||||
update_time: datetime
|
||||
|
||||
class Config:
|
||||
orm_mode = True
|
||||
|
||||
|
||||
class Patient(PatientBase):
|
||||
id: int
|
||||
create_time: datetime
|
||||
update_time: datetime
|
||||
|
||||
class Config:
|
||||
orm_mode = True
|
||||
|
||||
|
||||
class AppointmentSchema(AppointmentBase):
|
||||
id: int
|
||||
create_time: datetime
|
||||
update_time: datetime
|
||||
|
||||
class Config:
|
||||
orm_mode = True
|
||||
|
||||
|
||||
class Calendar(CalendarBase):
|
||||
id: int
|
||||
create_time: datetime
|
||||
update_time: datetime
|
||||
|
||||
class Config:
|
||||
orm_mode = True
|
||||
|
||||
|
||||
# custom schema for response
|
||||
class CalendarTimeSchema(BaseModel):
|
||||
time: str
|
||||
|
||||
class Config:
|
||||
orm_mode = True
|
||||
|
||||
|
||||
class ClinicSchema(BaseModel):
|
||||
id: int
|
||||
name: str
|
||||
address: str
|
||||
phone: str
|
||||
email: str
|
||||
|
||||
class Config:
|
||||
orm_mode = True
|
||||
|
||||
|
||||
# Detailed response schemas with nested relationships
|
||||
class ClinicWithDoctors(Clinic):
|
||||
doctors: List[Doctor] = []
|
||||
|
||||
|
||||
class DoctorWithAppointments(Doctor):
|
||||
appointments: List[AppointmentSchema] = []
|
||||
calendars: List[CalendarTimeSchema] = []
|
||||
clinic: ClinicSchema
|
||||
|
||||
|
||||
class DoctorWithCalendar(Doctor):
|
||||
calendars: List[CalendarTimeSchema] = []
|
||||
clinic: ClinicSchema
|
||||
|
||||
|
||||
class PatientWithAppointments(Patient):
|
||||
appointments: List[AppointmentSchema] = []
|
||||
|
||||
|
||||
class AppointmentDetailed(AppointmentSchema):
|
||||
|
||||
class Doctor(BaseModel):
|
||||
id: int
|
||||
name: str
|
||||
age: int
|
||||
email: str
|
||||
phone: str
|
||||
address: str
|
||||
|
||||
class Config:
|
||||
orm_mode = True
|
||||
|
||||
class Patient(BaseModel):
|
||||
id: int
|
||||
name: str
|
||||
age: int
|
||||
email: str
|
||||
phone: str
|
||||
address: str
|
||||
dob: str
|
||||
|
||||
class Config:
|
||||
orm_mode = True
|
||||
|
||||
doctor: Doctor
|
||||
patient: Patient
|
||||
@@ -0,0 +1,38 @@
|
||||
from .BaseSchemas import *
|
||||
|
||||
|
||||
# Update schemas (all fields optional for partial updates)
|
||||
class ClinicUpdate(BaseModel):
|
||||
name: Optional[str] = None
|
||||
address: Optional[str] = None
|
||||
phone: Optional[str] = None
|
||||
email: Optional[EmailStr] = None
|
||||
|
||||
|
||||
class DoctorUpdate(BaseModel):
|
||||
name: Optional[str] = None
|
||||
age: Optional[int] = None
|
||||
email: Optional[EmailStr] = None
|
||||
phone: Optional[str] = None
|
||||
address: Optional[str] = None
|
||||
clinic_id: Optional[int] = None
|
||||
|
||||
|
||||
class PatientUpdate(BaseModel):
|
||||
name: Optional[str] = None
|
||||
age: Optional[int] = None
|
||||
email: Optional[EmailStr] = None
|
||||
phone: Optional[str] = None
|
||||
address: Optional[str] = None
|
||||
|
||||
|
||||
class AppointmentUpdate(BaseModel):
|
||||
doctor_id: Optional[int] = None
|
||||
patient_id: Optional[int] = None
|
||||
appointment_time: Optional[datetime] = None
|
||||
status: Optional[AppointmentStatus] = None
|
||||
|
||||
|
||||
class CalendarUpdate(BaseModel):
|
||||
doc_id: Optional[int] = None
|
||||
rrule: Optional[str] = None
|
||||
Reference in New Issue
Block a user