Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, List
4import secrets
5from datetime import datetime, date
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11appointments = {}
12appointment_id_counter = 0
13attendance_history = {}
14 
15class UserCreate(BaseModel):
16    username: str
17    password: str
18    role: str = "patient"
19 
20class UserLogin(BaseModel):
21    username: str
22    password: str
23 
24class AppointmentCreate(BaseModel):
25    patient_id: int
26    doctor_id: int
27    time: str
28    duration: int
29    notes: str = ""
30    role: str = "patient"
31 
32class AppointmentSettingsUpdate(BaseModel):
33    time: Optional[str] = None
34    duration: Optional[int] = None
35    notes: Optional[str] = None
36    role: Optional[str] = None
37    status: Optional[str] = None
38 
39class BulkReschedule(BaseModel):
40    appointment_ids: List[int]
41    new_time: str
42    new_duration: Optional[int] = None
43 
44def get_current_user(authorization: str = Header(None)):
45    if not authorization:
46        raise HTTPException(status_code=401, detail="Missing auth header")
47    token = authorization.replace("Bearer ", "")
48    if token not in tokens:
49        raise HTTPException(status_code=401, detail="Invalid token")
50    return tokens[token]
51 
52@app.post("/signup")
53def signup(user: UserCreate):
54    if user.username in users:
55        raise HTTPException(status_code=400, detail="Username already exists")
56    users[user.username] = {"username": user.username, "password": user.password, "role": user.role}
57    return {"message": "User created"}
58 
59@app.post("/login")
60def login(user: UserLogin):
61    if user.username not in users or users[user.username]["password"] != user.password:
62        raise HTTPException(status_code=401, detail="Invalid credentials")
63    token = secrets.token_hex(16)
64    tokens[token] = user.username
65    return {"token": token}
66 
67@app.post("/appointments")
68def create_appointment(appt: AppointmentCreate, authorization: str = Header(None)):
69    current_user = get_current_user(authorization)
70    global appointment_id_counter
71    appointment_id_counter += 1
72    appt_data = appt.dict()
73    appt_data["id"] = appointment_id_counter
74    appt_data["created_by"] = current_user
75    appt_data["created_at"] = datetime.now().isoformat()
76    appointments[appointment_id_counter] = appt_data
77    attendance_history[appointment_id_counter] = []
78    return appt_data
79 
80@app.get("/appointments/{appointment_id}")
81def get_appointment(appointment_id: int, authorization: str = Header(None)):
82    current_user = get_current_user(authorization)
83    if appointment_id not in appointments:
84        raise HTTPException(status_code=404, detail="Appointment not found")
85    return appointments[appointment_id]
86 
87@app.put("/appointments/{appointment_id}/settings")
88def update_appointment_settings(appointment_id: int, settings: AppointmentSettingsUpdate, authorization: str = Header(None)):
89    current_user = get_current_user(authorization)
90    if appointment_id not in appointments:
91        raise HTTPException(status_code=404, detail="Appointment not found")
92    appt = appointments[appointment_id]
93    if settings.time is not None:
94        appt["time"] = settings.time
95    if settings.duration is not None:
96        appt["duration"] = settings.duration
97    if settings.notes is not None:
98        appt["notes"] = settings.notes
99    if settings.role is not None:
100        appt["role"] = settings.role
101    if settings.status is not None:
102        appt["status"] = settings.status
103        attendance_history[appointment_id].append({"status": settings.status, "timestamp": datetime.now().isoformat(), "changed_by": current_user})
104    return appt
105 
106@app.post("/appointments/bulk-reschedule")
107def bulk_reschedule(bulk: BulkReschedule, authorization: str = Header(None)):
108    current_user = get_current_user(authorization)
109    updated = []
110    for aid in bulk.appointment_ids:
111        if aid not in appointments:
112            raise HTTPException(status_code=404, detail=f"Appointment {aid} not found")
113        appt = appointments[aid]
114        appt["time"] = bulk.new_time
115        if bulk.new_duration is not None:
116            appt["duration"] = bulk.new_duration
117        appt["rescheduled_by"] = current_user
118        appt["rescheduled_at"] = datetime.now().isoformat()
119        updated.append(appt)
120    return {"updated": updated}
121 
122@app.get("/appointments/{appointment_id}/attendance")
123def get_attendance_history(appointment_id: int, authorization: str = Header(None)):
124    current_user = get_current_user(authorization)
125    if appointment_id not in appointments:
126        raise HTTPException(status_code=404, detail="Appointment not found")
127    return attendance_history.get(appointment_id, [])

1fastapi
2uvicorn