Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import uuid
4import random
5from datetime import datetime
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11patients = {}
12vitals = {}
13user_id_counter = 1
14patient_id_counter = 1
15vital_id_counter = 1
16 
17def get_current_user(authorization: Optional[str] = Header(None)):
18    if not authorization:
19        raise HTTPException(status_code=401, detail="Missing Authorization header")
20    token = authorization.replace("Bearer ", "")
21    if token not in tokens:
22        raise HTTPException(status_code=401, detail="Invalid token")
23    return tokens[token]
24 
25@app.post("/signup")
26def signup(username: str, password: str):
27    global user_id_counter
28    if any(u["username"] == username for u in users.values()):
29        raise HTTPException(status_code=400, detail="Username already exists")
30    user_id = user_id_counter
31    users[user_id] = {"id": user_id, "username": username, "password": password}
32    user_id_counter += 1
33    return {"id": user_id, "username": username}
34 
35@app.post("/login")
36def login(username: str, password: str):
37    for u in users.values():
38        if u["username"] == username and u["password"] == password:
39            token = str(uuid.uuid4())
40            tokens[token] = u["id"]
41            return {"token": token}
42    raise HTTPException(status_code=401, detail="Invalid credentials")
43 
44@app.post("/patients")
45def create_patient(name: str, age: int, authorization: Optional[str] = Header(None)):
46    get_current_user(authorization)
47    global patient_id_counter
48    patient_id = patient_id_counter
49    patients[patient_id] = {"id": patient_id, "name": name, "age": age}
50    patient_id_counter += 1
51    return patients[patient_id]
52 
53@app.get("/patients/{patient_id}")
54def get_patient(patient_id: int, authorization: Optional[str] = Header(None)):
55    get_current_user(authorization)
56    if patient_id not in patients:
57        raise HTTPException(status_code=404, detail="Patient not found")
58    return patients[patient_id]
59 
60@app.post("/patients/{patient_id}/vitals")
61def create_vital(patient_id: int, heart_rate: int, blood_pressure_systolic: int, blood_pressure_diastolic: int, temperature: float, authorization: Optional[str] = Header(None)):
62    get_current_user(authorization)
63    if patient_id not in patients:
64        raise HTTPException(status_code=404, detail="Patient not found")
65    global vital_id_counter
66    vital_id = vital_id_counter
67    vitals[vital_id] = {
68        "id": vital_id,
69        "patient_id": patient_id,
70        "heart_rate": heart_rate,
71        "blood_pressure": f"{blood_pressure_systolic}/{blood_pressure_diastolic}",
72        "temperature": temperature,
73        "timestamp": datetime.utcnow().isoformat()
74    }
75    vital_id_counter += 1
76    return vitals[vital_id]
77 
78@app.get("/patients/{patient_id}/vitals")
79def get_patient_vitals(patient_id: int, authorization: Optional[str] = Header(None)):
80    get_current_user(authorization)
81    if patient_id not in patients:
82        raise HTTPException(status_code=404, detail="Patient not found")
83    patient_vitals = [v for v in vitals.values() if v["patient_id"] == patient_id]
84    return patient_vitals

1fastapi
2uvicorn