Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import hashlib
5import secrets
6 
7app = FastAPI()
8 
9users = {}
10patients = {}
11appointments = {}
12medications = {}
13visit_summaries = {}
14tokens = {}
15id_counter = {"users": 1, "patients": 1, "appointments": 1, "medications": 1, "visit_summaries": 1}
16 
17def generate_token():
18    return secrets.token_hex(32)
19 
20def hash_password(password: str) -> str:
21    return hashlib.sha256(password.encode()).hexdigest()
22 
23def get_current_user(authorization: Optional[str] = Header(None)):
24    if not authorization:
25        raise HTTPException(status_code=401, detail="Missing auth header")
26    token = authorization.replace("Bearer ", "")
27    if token not in tokens:
28        raise HTTPException(status_code=401, detail="Invalid token")
29    return tokens[token]
30 
31class SignupRequest(BaseModel):
32    username: str
33    password: str
34 
35class LoginRequest(BaseModel):
36    username: str
37    password: str
38 
39@app.post("/signup")
40def signup(req: SignupRequest):
41    if req.username in users:
42        raise HTTPException(status_code=400, detail="Username already exists")
43    user_id = id_counter["users"]
44    id_counter["users"] += 1
45    users[req.username] = {"id": user_id, "username": req.username, "password_hash": hash_password(req.password)}
46    return {"id": user_id, "username": req.username}
47 
48@app.post("/login")
49def login(req: LoginRequest):
50    if req.username not in users:
51        raise HTTPException(status_code=401, detail="Invalid credentials")
52    user = users[req.username]
53    if user["password_hash"] != hash_password(req.password):
54        raise HTTPException(status_code=401, detail="Invalid credentials")
55    token = generate_token()
56    tokens[token] = user["id"]
57    return {"token": token}
58 
59@app.get("/patients/{patient_id}")
60def get_patient(patient_id: int, authorization: Optional[str] = Header(None)):
61    current_user = get_current_user(authorization)
62    if patient_id not in patients:
63        raise HTTPException(status_code=404, detail="Patient not found")
64    patient = patients[patient_id]
65    patient["upcoming_appointments"] = [a for a in appointments.values() if a["patient_id"] == patient_id]
66    patient["medication_list"] = [m for m in medications.values() if m["patient_id"] == patient_id]
67    patient["last_visit_summary"] = [v for v in visit_summaries.values() if v["patient_id"] == patient_id]
68    if patient["last_visit_summary"]:
69        patient["last_visit_summary"] = max(patient["last_visit_summary"], key=lambda x: x["date"])
70    else:
71        patient["last_visit_summary"] = None
72    return patient
73 
74@app.get("/patients")
75def get_all_patients(authorization: Optional[str] = Header(None)):
76    current_user = get_current_user(authorization)
77    return list(patients.values())
78 
79@app.post("/patients")
80def create_patient(name: str, authorization: Optional[str] = Header(None)):
81    current_user = get_current_user(authorization)
82    patient_id = id_counter["patients"]
83    id_counter["patients"] += 1
84    patients[patient_id] = {"id": patient_id, "name": name}
85    return patients[patient_id]
86 
87@app.get("/appointments/{appointment_id}")
88def get_appointment(appointment_id: int, authorization: Optional[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    return appointments[appointment_id]
93 
94@app.post("/appointments")
95def create_appointment(patient_id: int, date: str, description: str = "", authorization: Optional[str] = Header(None)):
96    current_user = get_current_user(authorization)
97    if patient_id not in patients:
98        raise HTTPException(status_code=404, detail="Patient not found")
99    appointment_id = id_counter["appointments"]
100    id_counter["appointments"] += 1
101    appointments[appointment_id] = {"id": appointment_id, "patient_id": patient_id, "date": date, "description": description}
102    return appointments[appointment_id]
103 
104@app.get("/medications/{medication_id}")
105def get_medication(medication_id: int, authorization: Optional[str] = Header(None)):
106    current_user = get_current_user(authorization)
107    if medication_id not in medications:
108        raise HTTPException(status_code=404, detail="Medication not found")
109    return medications[medication_id]
110 
111@app.post("/medications")
112def create_medication(patient_id: int, name: str, dosage: str, authorization: Optional[str] = Header(None)):
113    current_user = get_current_user(authorization)
114    if patient_id not in patients:
115        raise HTTPException(status_code=404, detail="Patient not found")
116    medication_id = id_counter["medications"]
117    id_counter["medications"] += 1
118    medications[medication_id] = {"id": medication_id, "patient_id": patient_id, "name": name, "dosage": dosage}
119    return medications[medication_id]
120 
121@app.get("/visit_summaries/{visit_summary_id}")
122def get_visit_summary(visit_summary_id: int, authorization: Optional[str] = Header(None)):
123    current_user = get_current_user(authorization)
124    if visit_summary_id not in visit_summaries:
125        raise HTTPException(status_code=404, detail="Visit summary not found")
126    return visit_summaries[visit_summary_id]
127 
128@app.post("/visit_summaries")
129def create_visit_summary(patient_id: int, date: str, summary: str, authorization: Optional[str] = Header(None)):
130    current_user = get_current_user(authorization)
131    if patient_id not in patients:
132        raise HTTPException(status_code=404, detail="Patient not found")
133    visit_summary_id = id_counter["visit_summaries"]
134    id_counter["visit_summaries"] += 1
135    visit_summaries[visit_summary_id] = {"id": visit_summary_id, "patient_id": patient_id, "date": date, "summary": summary}
136    return visit_summaries[visit_summary_id]

1fastapi
2uvicorn