Browse the data

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

1fastapi
2uvicorn