Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3import secrets
4import hashlib
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10drugs = {}
11drug_id_counter = 1
12 
13DRUGS_DATA = {
14    "aspirin": {
15        "dosage": "325-650 mg every 4-6 hours",
16        "interactions": ["warfarin", "ibuprofen", "alcohol"],
17        "requires_refrigeration": False
18    },
19    "insulin": {
20        "dosage": "Per sliding scale or as directed",
21        "interactions": ["beta-blockers", "MAOIs", "alcohol"],
22        "requires_refrigeration": True
23    },
24    "amoxicillin": {
25        "dosage": "500 mg every 8 hours",
26        "interactions": ["methotrexate", "warfarin", "oral contraceptives"],
27        "requires_refrigeration": False
28    }
29}
30 
31class SignupRequest(BaseModel):
32    username: str
33    password: str
34 
35class LoginRequest(BaseModel):
36    username: str
37    password: str
38 
39class DrugCreate(BaseModel):
40    name: str
41    dosage: str
42    interactions: list[str]
43    requires_refrigeration: bool
44 
45def get_current_user(authorization: str = Header(None)):
46    if not authorization:
47        raise HTTPException(status_code=401, detail="No auth header")
48    token = authorization.replace("Bearer ", "")
49    for uid, t in tokens.items():
50        if t == token:
51            return uid
52    raise HTTPException(status_code=401, detail="Invalid token")
53 
54@app.post("/signup")
55def signup(req: SignupRequest):
56    if req.username in users:
57        raise HTTPException(status_code=400, detail="User exists")
58    users[req.username] = hashlib.sha256(req.password.encode()).hexdigest()
59    return {"message": "ok"}
60 
61@app.post("/login")
62def login(req: LoginRequest):
63    if req.username not in users:
64        raise HTTPException(status_code=400, detail="Invalid credentials")
65    if users[req.username] != hashlib.sha256(req.password.encode()).hexdigest():
66        raise HTTPException(status_code=400, detail="Invalid credentials")
67    token = secrets.token_hex(16)
68    tokens[req.username] = token
69    return {"token": token}
70 
71@app.get("/drugs/{drug_id}")
72def get_drug(drug_id: int, authorization: str = Header(None)):
73    get_current_user(authorization)
74    if drug_id not in drugs:
75        raise HTTPException(status_code=404, detail="Drug not found")
76    return drugs[drug_id]
77 
78@app.get("/drugs")
79def search_drugs(name: str = "", authorization: str = Header(None)):
80    get_current_user(authorization)
81    if not name:
82        return list(drugs.values())
83    result = []
84    for d in drugs.values():
85        if name.lower() in d["name"].lower():
86            result.append(d)
87    return result
88 
89@app.post("/drugs")
90def create_drug(drug: DrugCreate, authorization: str = Header(None)):
91    global drug_id_counter
92    get_current_user(authorization)
93    new_id = drug_id_counter
94    drug_id_counter += 1
95    drugs[new_id] = {
96        "id": new_id,
97        "name": drug.name,
98        "dosage": drug.dosage,
99        "interactions": drug.interactions,
100        "requires_refrigeration": drug.requires_refrigeration
101    }
102    return drugs[new_id]
103 
104# Seed some drugs
105for name, data in DRUGS_DATA.items():
106    new_id = drug_id_counter
107    drug_id_counter += 1
108    drugs[new_id] = {
109        "id": new_id,
110        "name": name,
111        "dosage": data["dosage"],
112        "interactions": data["interactions"],
113        "requires_refrigeration": data["requires_refrigeration"]
114    }

1fastapi
2uvicorn