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] = {}
10user_id_counter = 1
11patient_id_counter = 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: Optional[str] = None
25    insurance_details: Optional[str] = None
26    medical_profile: Optional[str] = None
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(...)):
35    token = authorization.replace("Bearer ", "").strip()
36    user_id = tokens.get(token)
37    if user_id is None:
38        raise HTTPException(status_code=401, detail="Invalid token")
39    return user_id
40 
41@app.post("/signup")
42def signup(req: SignupRequest):
43    global user_id_counter
44    for u in users.values():
45        if u["username"] == req.username:
46            raise HTTPException(status_code=400, detail="Username already exists")
47    user_id = user_id_counter
48    user_id_counter += 1
49    users[user_id] = {"id": user_id, "username": req.username, "password": req.password}
50    token = secrets.token_hex(16)
51    tokens[token] = user_id
52    return {"user_id": user_id, "token": token}
53 
54@app.post("/login")
55def login(req: LoginRequest):
56    for u in users.values():
57        if u["username"] == req.username and u["password"] == req.password:
58            token = secrets.token_hex(16)
59            tokens[token] = u["id"]
60            return {"token": token}
61    raise HTTPException(status_code=401, detail="Invalid credentials")
62 
63@app.post("/patients")
64def create_patient(req: PatientCreate, authorization: str = Header(...)):
65    global patient_id_counter
66    get_user_id_from_token(authorization)
67    patient_id = patient_id_counter
68    patient_id_counter += 1
69    patients[patient_id] = {
70        "id": patient_id,
71        "name": req.name,
72        "contact_info": req.contact_info,
73        "insurance_details": req.insurance_details,
74        "medical_profile": req.medical_profile
75    }
76    return patients[patient_id]
77 
78@app.get("/patients/{patient_id}")
79def get_patient(patient_id: int, authorization: str = Header(...)):
80    get_user_id_from_token(authorization)
81    patient = patients.get(patient_id)
82    if patient is None:
83        raise HTTPException(status_code=404, detail="Patient not found")
84    return patient
85 
86@app.patch("/patients/{patient_id}")
87def update_patient(patient_id: int, req: PatientUpdate, authorization: str = Header(...)):
88    get_user_id_from_token(authorization)
89    patient = patients.get(patient_id)
90    if patient is None:
91        raise HTTPException(status_code=404, detail="Patient not found")
92    if req.name is not None:
93        patient["name"] = req.name
94    if req.contact_info is not None:
95        patient["contact_info"] = req.contact_info
96    if req.insurance_details is not None:
97        patient["insurance_details"] = req.insurance_details
98    if req.medical_profile is not None:
99        patient["medical_profile"] = req.medical_profile
100    return patient

1fastapi
2uvicorn