Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import secrets
4 
5app = FastAPI()
6 
7users = {}
8tokens = {}
9drives = {}
10donations = {}
11next_user_id = 1
12next_drive_id = 1
13next_donation_id = 1
14campaign_admin_id = None
15 
16def get_current_user(authorization: Optional[str] = Header(None)):
17    if not authorization:
18        raise HTTPException(status_code=401, detail="Missing auth token")
19    token = authorization.replace("Bearer ", "")
20    user_id = tokens.get(token)
21    if not user_id:
22        raise HTTPException(status_code=401, detail="Invalid token")
23    return user_id
24 
25@app.post("/signup")
26def signup(username: str, password: str):
27    global next_user_id, campaign_admin_id
28    user_id = next_user_id
29    next_user_id += 1
30    users[user_id] = {"id": user_id, "username": username, "password": password, "role": "user"}
31    if campaign_admin_id is None:
32        campaign_admin_id = user_id
33        users[user_id]["role"] = "admin"
34    return {"id": user_id, "username": username, "role": users[user_id]["role"]}
35 
36@app.post("/login")
37def login(username: str, password: str):
38    for uid, u in users.items():
39        if u["username"] == username and u["password"] == password:
40            token = secrets.token_hex(16)
41            tokens[token] = uid
42            return {"token": token, "user_id": uid}
43    raise HTTPException(status_code=401, detail="Invalid credentials")
44 
45@app.post("/promote/{user_id}")
46def promote_user(user_id: int, authorization: Optional[str] = Header(None)):
47    current_user = get_current_user(authorization)
48    if current_user != campaign_admin_id:
49        raise HTTPException(status_code=403, detail="Only campaign admin can promote")
50    if user_id not in users:
51        raise HTTPException(status_code=404, detail="User not found")
52    users[user_id]["role"] = "fundraiser"
53    return {"user_id": user_id, "role": "fundraiser"}
54 
55@app.get("/users/{user_id}")
56def get_user(user_id: int, authorization: Optional[str] = Header(None)):
57    get_current_user(authorization)
58    if user_id not in users:
59        raise HTTPException(status_code=404, detail="User not found")
60    return users[user_id]
61 
62@app.post("/drives")
63def create_drive(name: str, goal: float, authorization: Optional[str] = Header(None)):
64    current_user = get_current_user(authorization)
65    if users[current_user]["role"] not in ("admin", "fundraiser"):
66        raise HTTPException(status_code=403, detail="Only fundraisers and admin can create drives")
67    global next_drive_id
68    drive_id = next_drive_id
69    next_drive_id += 1
70    drives[drive_id] = {"id": drive_id, "name": name, "goal": goal, "raised": 0.0, "created_by": current_user, "donations": []}
71    return drives[drive_id]
72 
73@app.get("/drives/{drive_id}")
74def get_drive(drive_id: int, authorization: Optional[str] = Header(None)):
75    get_current_user(authorization)
76    if drive_id not in drives:
77        raise HTTPException(status_code=404, detail="Drive not found")
78    return drives[drive_id]
79 
80@app.post("/donations")
81def create_donation(drive_id: int, donor_name: str, amount: float, authorization: Optional[str] = Header(None)):
82    current_user = get_current_user(authorization)
83    if users[current_user]["role"] not in ("admin", "fundraiser"):
84        raise HTTPException(status_code=403, detail="Only fundraisers and admin can record donations")
85    if drive_id not in drives:
86        raise HTTPException(status_code=404, detail="Drive not found")
87    global next_donation_id
88    donation_id = next_donation_id
89    next_donation_id += 1
90    donation = {"id": donation_id, "drive_id": drive_id, "donor_name": donor_name, "amount": amount, "recorded_by": current_user}
91    donations[donation_id] = donation
92    drives[drive_id]["raised"] += amount
93    drives[drive_id]["donations"].append(donation_id)
94    return donation
95 
96@app.get("/donations/{donation_id}")
97def get_donation(donation_id: int, authorization: Optional[str] = Header(None)):
98    get_current_user(authorization)
99    if donation_id not in donations:
100        raise HTTPException(status_code=404, detail="Donation not found")
101    return donations[donation_id]
102 
103@app.get("/donors")
104def get_donor_history(authorization: Optional[str] = Header(None)):
105    current_user = get_current_user(authorization)
106    if users[current_user]["role"] not in ("admin", "fundraiser"):
107        raise HTTPException(status_code=403, detail="Only fundraisers and admin can view donor history")
108    history = {}
109    for d in donations.values():
110        name = d["donor_name"]
111        if name not in history:
112            history[name] = {"donor_name": name, "total_donated": 0.0, "donations": []}
113        history[name]["total_donated"] += d["amount"]
114        history[name]["donations"].append(d)
115    return list(history.values())

1fastapi
2uvicorn