Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from datetime import datetime, timedelta
3import secrets
4import hashlib
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10prescriptions = {}
11prescription_id_counter = 1
12 
13def hash_password(password):
14    return hashlib.sha256(password.encode()).hexdigest()
15 
16def generate_token():
17    return secrets.token_hex(32)
18 
19def get_current_user(authorization: str = Header(...)):
20    if not authorization.startswith("Bearer "):
21        raise HTTPException(status_code=401, detail="Invalid auth header")
22    token = authorization.replace("Bearer ", "")
23    if token not in tokens:
24        raise HTTPException(status_code=401, detail="Invalid token")
25    return tokens[token]
26 
27@app.post("/signup")
28def signup(username: str, password: str):
29    if username in users:
30        raise HTTPException(status_code=400, detail="User exists")
31    users[username] = {"username": username, "password": hash_password(password)}
32    return {"message": "User created"}
33 
34@app.post("/login")
35def login(username: str, password: str):
36    if username not in users or users[username]["password"] != hash_password(password):
37        raise HTTPException(status_code=401, detail="Invalid credentials")
38    token = generate_token()
39    tokens[token] = username
40    return {"token": token}
41 
42@app.get("/prescriptions/{prescription_id}")
43def get_prescription(prescription_id: int, authorization: str = Header(...)):
44    get_current_user(authorization)
45    if prescription_id not in prescriptions:
46        raise HTTPException(status_code=404, detail="Prescription not found")
47    return prescriptions[prescription_id]
48 
49@app.post("/prescriptions")
50def create_prescription(medication: str, dosage: str, refills: int, days_supply: int, authorization: str = Header(...)):
51    get_current_user(authorization)
52    global prescription_id_counter
53    pid = prescription_id_counter
54    prescription_id_counter += 1
55    prescriptions[pid] = {
56        "id": pid,
57        "medication": medication,
58        "dosage": dosage,
59        "refills": refills,
60        "days_supply": days_supply,
61        "created_at": datetime.now().isoformat()
62    }
63    return prescriptions[pid]
64 
65@app.get("/prescriptions")
66def list_prescriptions(authorization: str = Header(...)):
67    get_current_user(authorization)
68    result = []
69    for pid, p in prescriptions.items():
70        days_remaining = p["days_supply"] - (datetime.now() - datetime.fromisoformat(p["created_at"])).days
71        result.append({
72            "id": pid,
73            "medication": p["medication"],
74            "dosage": p["dosage"],
75            "refills": p["refills"],
76            "days_remaining": max(0, days_remaining)
77        })
78    return result
79 
80@app.patch("/prescriptions")
81def bulk_update_prescriptions(updates: list[dict], authorization: str = Header(...)):
82    get_current_user(authorization)
83    for update in updates:
84        pid = update.get("id")
85        if pid not in prescriptions:
86            continue
87        if "refills" in update:
88            prescriptions[pid]["refills"] = update["refills"]
89        if "days_supply" in update:
90            prescriptions[pid]["days_supply"] = update["days_supply"]
91    return {"message": "Updated"}

1fastapi
2uvicorn