Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import hashlib
5import secrets
6 
7app = FastAPI()
8 
9# In-memory stores
10users = {}
11tokens = {}
12drugs = {}
13interactions = {}
14prescriptions = {}
15prescription_drugs = {}
16 
17# ID counters
18user_id_counter = 1
19drug_id_counter = 1
20interaction_id_counter = 1
21prescription_id_counter = 1
22prescription_drug_id_counter = 1
23 
24def hash_password(password: str) -> str:
25    return hashlib.sha256(password.encode()).hexdigest()
26 
27def generate_token() -> str:
28    return secrets.token_hex(32)
29 
30def get_current_user(authorization: Optional[str] = Header(None)):
31    if not authorization:
32        raise HTTPException(status_code=401, detail="Missing authorization header")
33    token = authorization.replace("Bearer ", "")
34    if token not in tokens:
35        raise HTTPException(status_code=401, detail="Invalid token")
36    return tokens[token]
37 
38class SignupRequest(BaseModel):
39    username: str
40    password: str
41 
42class LoginRequest(BaseModel):
43    username: str
44    password: str
45 
46class DrugCreate(BaseModel):
47    name: str
48    description: str = ""
49 
50class InteractionCreate(BaseModel):
51    drug1_id: int
52    drug2_id: int
53    severity: str
54    description: str = ""
55 
56class PrescriptionCreate(BaseModel):
57    patient_name: str
58    drug_ids: list[int]
59 
60# Auth endpoints
61@app.post("/signup")
62def signup(req: SignupRequest):
63    global user_id_counter
64    for u in users.values():
65        if u["username"] == req.username:
66            raise HTTPException(status_code=400, detail="Username already exists")
67    user_id = user_id_counter
68    user_id_counter += 1
69    users[user_id] = {
70        "id": user_id,
71        "username": req.username,
72        "password_hash": hash_password(req.password)
73    }
74    return {"id": user_id, "username": req.username}
75 
76@app.post("/login")
77def login(req: LoginRequest):
78    for u in users.values():
79        if u["username"] == req.username and u["password_hash"] == hash_password(req.password):
80            token = generate_token()
81            tokens[token] = u["id"]
82            return {"token": token}
83    raise HTTPException(status_code=401, detail="Invalid credentials")
84 
85# Drug endpoints
86@app.get("/drug/{drug_id}")
87def get_drug(drug_id: int, authorization: Optional[str] = Header(None)):
88    get_current_user(authorization)
89    if drug_id not in drugs:
90        raise HTTPException(status_code=404, detail="Drug not found")
91    return drugs[drug_id]
92 
93@app.post("/drug")
94def create_drug(drug: DrugCreate, authorization: Optional[str] = Header(None)):
95    get_current_user(authorization)
96    global drug_id_counter
97    drug_id = drug_id_counter
98    drug_id_counter += 1
99    drugs[drug_id] = {
100        "id": drug_id,
101        "name": drug.name,
102        "description": drug.description
103    }
104    return drugs[drug_id]
105 
106# Interaction endpoints
107@app.get("/interaction/{interaction_id}")
108def get_interaction(interaction_id: int, authorization: Optional[str] = Header(None)):
109    get_current_user(authorization)
110    if interaction_id not in interactions:
111        raise HTTPException(status_code=404, detail="Interaction not found")
112    return interactions[interaction_id]
113 
114@app.post("/interaction")
115def create_interaction(interaction: InteractionCreate, authorization: Optional[str] = Header(None)):
116    get_current_user(authorization)
117    global interaction_id_counter
118    if interaction.drug1_id not in drugs or interaction.drug2_id not in drugs:
119        raise HTTPException(status_code=400, detail="One or both drugs not found")
120    interaction_id = interaction_id_counter
121    interaction_id_counter += 1
122    interactions[interaction_id] = {
123        "id": interaction_id,
124        "drug1_id": interaction.drug1_id,
125        "drug2_id": interaction.drug2_id,
126        "severity": interaction.severity,
127        "description": interaction.description
128    }
129    return interactions[interaction_id]
130 
131# Prescription endpoints
132@app.get("/prescription/{prescription_id}")
133def get_prescription(prescription_id: int, authorization: Optional[str] = Header(None)):
134    get_current_user(authorization)
135    if prescription_id not in prescriptions:
136        raise HTTPException(status_code=404, detail="Prescription not found")
137    return prescriptions[prescription_id]
138 
139@app.post("/prescription")
140def create_prescription(prescription: PrescriptionCreate, authorization: Optional[str] = Header(None)):
141    get_current_user(authorization)
142    global prescription_id_counter, prescription_drug_id_counter
143 
144    # Validate all drug IDs exist
145    for drug_id in prescription.drug_ids:
146        if drug_id not in drugs:
147            raise HTTPException(status_code=400, detail=f"Drug {drug_id} not found")
148 
149    prescription_id = prescription_id_counter
150    prescription_id_counter += 1
151 
152    prescriptions[prescription_id] = {
153        "id": prescription_id,
154        "patient_name": prescription.patient_name,
155        "drug_ids": prescription.drug_ids
156    }
157 
158    # Check for interactions between all drug pairs
159    conflicts = []
160    for i in range(len(prescription.drug_ids)):
161        for j in range(i+1, len(prescription.drug_ids)):
162            d1 = prescription.drug_ids[i]
163            d2 = prescription.drug_ids[j]
164            for inter in interactions.values():
165                if (inter["drug1_id"] == d1 and inter["drug2_id"] == d2) or \
166                   (inter["drug1_id"] == d2 and inter["drug2_id"] == d1):
167                    conflicts.append({
168                        "interaction_id": inter["id"],
169                        "drug1_id": d1,
170                        "drug2_id": d2,
171                        "severity": inter["severity"],
172                        "description": inter["description"]
173                    })
174 
175    result = prescriptions[prescription_id].copy()
176    if conflicts:
177        result["conflicts"] = conflicts
178    else:
179        result["conflicts"] = []
180 
181    return result

1fastapi
2uvicorn