Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, Dict
4import secrets
5 
6app = FastAPI()
7 
8users: Dict[int, dict] = {}
9patients: Dict[int, dict] = {}
10next_user_id = 1
11next_patient_id = 1
12tokens: Dict[str, int] = {}
13 
14class SignupRequest(BaseModel):
15    username: str
16    password: str
17 
18class LoginRequest(BaseModel):
19    username: str
20    password: str
21 
22class PatientCreate(BaseModel):
23    name: str
24    contact_info: str
25    insurance_details: str
26    medical_profile: str
27 
28class PatientUpdate(BaseModel):
29    name: Optional[str] = None
30    contact_info: Optional[str] = None
31    insurance_details: Optional[str] = None
32    medical_profile: Optional[str] = None
33 
34def get_user_id_from_token(authorization: str = Header(...)) -> int:
35    if not authorization.startswith("Bearer "):
36        raise HTTPException(status_code=401, detail="Invalid auth header")
37    token = authorization.split(" ")[1]
38    user_id = tokens.get(token)
39    if user_id is None:
40        raise HTTPException(status_code=401, detail="Invalid token")
41    return user_id
42 
43@app.post("/signup")
44def signup(req: SignupRequest):
45    global next_user_id
46    user_id = next_user_id
47    next_user_id += 1
48    users[user_id] = {"username": req.username, "password": req.password}
49    return {"id": user_id, "username": req.username}
50 
51@app.post("/login")
52def login(req: LoginRequest):
53    for uid, u in users.items():
54        if u["username"] == req.username and u["password"] == req.password:
55            token = secrets.token_hex(16)
56            tokens[token] = uid
57            return {"token": token}
58    raise HTTPException(status_code=401, detail="Invalid credentials")
59 
60@app.get("/patients/{patient_id}")
61def get_patient(patient_id: int, authorization: str = Header(...)):
62    get_user_id_from_token(authorization)
63    patient = patients.get(patient_id)
64    if patient is None:
65        raise HTTPException(status_code=404, detail="Patient not found")
66    return patient
67 
68@app.post("/patients")
69def create_patient(req: PatientCreate, authorization: str = Header(...)):
70    global next_patient_id
71    get_user_id_from_token(authorization)
72    patient_id = next_patient_id
73    next_patient_id += 1
74    patients[patient_id] = {
75        "id": patient_id,
76        "name": req.name,
77        "contact_info": req.contact_info,
78        "insurance_details": req.insurance_details,
79        "medical_profile": req.medical_profile
80    }
81    return patients[patient_id]
82 
83@app.patch("/patients/{patient_id}")
84def update_patient(patient_id: int, req: PatientUpdate, authorization: str = Header(...)):
85    get_user_id_from_token(authorization)
86    patient = patients.get(patient_id)
87    if patient is None:
88        raise HTTPException(status_code=404, detail="Patient not found")
89    if req.name is not None:
90        patient["name"] = req.name
91    if req.contact_info is not None:
92        patient["contact_info"] = req.contact_info
93    if req.insurance_details is not None:
94        patient["insurance_details"] = req.insurance_details
95    if req.medical_profile is not None:
96        patient["medical_profile"] = req.medical_profile
97    return patient

1fastapi
2uvicorn