Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import random
4 
5app = FastAPI()
6 
7users = {}
8doctors = {}
9appointments = {}
10tokens = {}
11 
12user_id_counter = 1
13doctor_id_counter = 1
14appointment_id_counter = 1
15token_counter = 1
16 
17doctors_list = [
18    {"name": "Dr. Smith", "specialty": "Cardiology", "location": "NYC", "years_experience": 15, "rating": 4.5},
19    {"name": "Dr. Jones", "specialty": "Cardiology", "location": "LA", "years_experience": 10, "rating": 4.2},
20    {"name": "Dr. Lee", "specialty": "Dermatology", "location": "NYC", "years_experience": 8, "rating": 4.8},
21    {"name": "Dr. Patel", "specialty": "Orthopedics", "location": "NYC", "years_experience": 20, "rating": 4.9},
22    {"name": "Dr. Brown", "specialty": "Cardiology", "location": "NYC", "years_experience": 5, "rating": 3.8},
23]
24 
25for doc in doctors_list:
26    doctors[doctor_id_counter] = {"id": doctor_id_counter, **doc}
27    doctor_id_counter += 1
28 
29appointments_list = [
30    {"doctor_id": 1, "patient_id": 1, "date": "2024-01-15"},
31    {"doctor_id": 1, "patient_id": 2, "date": "2024-01-16"},
32    {"doctor_id": 2, "patient_id": 1, "date": "2024-01-17"},
33    {"doctor_id": 3, "patient_id": 2, "date": "2024-01-18"},
34    {"doctor_id": 1, "patient_id": 3, "date": "2024-01-19"},
35]
36 
37for apt in appointments_list:
38    appointments[appointment_id_counter] = {"id": appointment_id_counter, **apt}
39    appointment_id_counter += 1
40 
41def get_current_user(authorization: str = Header(None)):
42    if not authorization:
43        raise HTTPException(status_code=401, detail="No auth header")
44    token = authorization.replace("Bearer ", "")
45    if token not in tokens:
46        raise HTTPException(status_code=401, detail="Invalid token")
47    return tokens[token]
48 
49@app.post("/signup")
50def signup(username: str, password: str):
51    global user_id_counter
52    user_id = user_id_counter
53    users[user_id] = {"id": user_id, "username": username, "password": password}
54    user_id_counter += 1
55    return {"id": user_id, "username": username}
56 
57@app.post("/login")
58def login(username: str, password: str):
59    for uid, u in users.items():
60        if u["username"] == username and u["password"] == password:
61            global token_counter
62            token = f"tok-{token_counter}"
63            tokens[token] = uid
64            token_counter += 1
65            return {"token": token}
66    raise HTTPException(status_code=401, detail="Invalid credentials")
67 
68@app.get("/doctors/{doctor_id}")
69def get_doctor(doctor_id: int, authorization: str = Header(None)):
70    get_current_user(authorization)
71    if doctor_id not in doctors:
72        raise HTTPException(status_code=404, detail="Doctor not found")
73    return doctors[doctor_id]
74 
75@app.post("/doctors")
76def create_doctor(name: str, specialty: str, location: str, years_experience: int, rating: float, authorization: str = Header(None)):
77    get_current_user(authorization)
78    global doctor_id_counter
79    doc_id = doctor_id_counter
80    doctors[doc_id] = {"id": doc_id, "name": name, "specialty": specialty, "location": location, "years_experience": years_experience, "rating": rating}
81    doctor_id_counter += 1
82    return doctors[doc_id]
83 
84@app.get("/doctors")
85def search_doctors(specialty: Optional[str] = None, location: Optional[str] = None, authorization: str = Header(None)):
86    get_current_user(authorization)
87    result = []
88    for doc_id, doc in doctors.items():
89        if specialty and doc["specialty"].lower() != specialty.lower():
90            continue
91        if location and doc["location"].lower() != location.lower():
92            continue
93        apt_count = sum(1 for apt in appointments.values() if apt["doctor_id"] == doc_id)
94        result.append({**doc, "appointment_count": apt_count})
95    return result
96 
97@app.get("/appointments/{appointment_id}")
98def get_appointment(appointment_id: int, authorization: str = Header(None)):
99    get_current_user(authorization)
100    if appointment_id not in appointments:
101        raise HTTPException(status_code=404, detail="Appointment not found")
102    return appointments[appointment_id]
103 
104@app.post("/appointments")
105def create_appointment(doctor_id: int, patient_id: int, date: str, authorization: str = Header(None)):
106    get_current_user(authorization)
107    global appointment_id_counter
108    apt_id = appointment_id_counter
109    appointments[apt_id] = {"id": apt_id, "doctor_id": doctor_id, "patient_id": patient_id, "date": date}
110    appointment_id_counter += 1
111    return appointments[apt_id]
112 
113@app.get("/users/{user_id}")
114def get_user(user_id: int, authorization: str = Header(None)):
115    get_current_user(authorization)
116    if user_id not in users:
117        raise HTTPException(status_code=404, detail="User not found")
118    return users[user_id]
119 
120@app.post("/users")
121def create_user(username: str, password: str, authorization: str = Header(None)):
122    get_current_user(authorization)
123    global user_id_counter
124    user_id = user_id_counter
125    users[user_id] = {"id": user_id, "username": username, "password": password}
126    user_id_counter += 1
127    return users[user_id]

1fastapi
2uvicorn