Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import secrets
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10patients = {}
11user_id_counter = 1
12patient_id_counter = 1
13 
14class SignupRequest(BaseModel):
15    email: str
16    password: str
17    name: str
18 
19class LoginRequest(BaseModel):
20    email: str
21    password: str
22 
23class PatientUpdate(BaseModel):
24    name: Optional[str] = None
25    date_of_birth: Optional[str] = None
26    insurance_provider: Optional[str] = None
27    insurance_policy_number: Optional[str] = None
28    emergency_contact_name: Optional[str] = None
29    emergency_contact_phone: Optional[str] = None
30    preferred_pharmacy: Optional[str] = None
31 
32def get_current_user(authorization: str = Header(None)):
33    if not authorization:
34        raise HTTPException(status_code=401, detail="Missing auth header")
35    token = authorization.replace("Bearer ", "")
36    if token not in tokens:
37        raise HTTPException(status_code=401, detail="Invalid token")
38    return tokens[token]
39 
40@app.post("/signup")
41def signup(req: SignupRequest):
42    global user_id_counter
43    for u in users.values():
44        if u["email"] == req.email:
45            raise HTTPException(status_code=400, detail="Email already exists")
46    user_id = user_id_counter
47    user_id_counter += 1
48    users[user_id] = {
49        "id": user_id,
50        "email": req.email,
51        "password": req.password,
52        "name": req.name
53    }
54    return {"id": user_id, "email": req.email, "name": req.name}
55 
56@app.post("/login")
57def login(req: LoginRequest):
58    for u in users.values():
59        if u["email"] == req.email and u["password"] == req.password:
60            token = secrets.token_hex(32)
61            tokens[token] = u["id"]
62            return {"token": token, "user_id": u["id"]}
63    raise HTTPException(status_code=401, detail="Invalid credentials")
64 
65@app.get("/patients/{patient_id}")
66def get_patient(patient_id: int, authorization: str = Header(None)):
67    user_id = get_current_user(authorization)
68    if patient_id not in patients:
69        raise HTTPException(status_code=404, detail="Patient not found")
70    return patients[patient_id]
71 
72@app.post("/patients")
73def create_patient(authorization: str = Header(None)):
74    global patient_id_counter
75    user_id = get_current_user(authorization)
76    patient_id = patient_id_counter
77    patient_id_counter += 1
78    patients[patient_id] = {
79        "id": patient_id,
80        "user_id": user_id,
81        "name": "",
82        "date_of_birth": "",
83        "insurance_provider": "",
84        "insurance_policy_number": "",
85        "emergency_contact_name": "",
86        "emergency_contact_phone": "",
87        "preferred_pharmacy": ""
88    }
89    return patients[patient_id]
90 
91@app.patch("/patients/{patient_id}")
92def update_patient(patient_id: int, update: PatientUpdate, authorization: str = Header(None)):
93    user_id = get_current_user(authorization)
94    if patient_id not in patients:
95        raise HTTPException(status_code=404, detail="Patient not found")
96    patient = patients[patient_id]
97    if update.name is not None:
98        patient["name"] = update.name
99    if update.date_of_birth is not None:
100        patient["date_of_birth"] = update.date_of_birth
101    if update.insurance_provider is not None:
102        patient["insurance_provider"] = update.insurance_provider
103    if update.insurance_policy_number is not None:
104        patient["insurance_policy_number"] = update.insurance_policy_number
105    if update.emergency_contact_name is not None:
106        patient["emergency_contact_name"] = update.emergency_contact_name
107    if update.emergency_contact_phone is not None:
108        patient["emergency_contact_phone"] = update.emergency_contact_phone
109    if update.preferred_pharmacy is not None:
110        patient["preferred_pharmacy"] = update.preferred_pharmacy
111    return patient

1fastapi
2uvicorn