Browse the data

1import secrets
2from fastapi import FastAPI, HTTPException, Header
3from pydantic import BaseModel
4from typing import Optional
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10campaigns = {}
11donations = {}
12next_user_id = 1
13next_campaign_id = 1
14next_donation_id = 1
15 
16class SignupRequest(BaseModel):
17    username: str
18    password: str
19 
20class LoginRequest(BaseModel):
21    username: str
22    password: str
23 
24class CampaignCreate(BaseModel):
25    title: str
26    goal: float
27 
28class DonationCreate(BaseModel):
29    campaign_id: int
30    amount: float
31 
32def get_current_user(authorization: Optional[str] = Header(None)):
33    if not authorization:
34        raise HTTPException(status_code=401, detail="No auth header")
35    token = authorization.replace("Bearer ", "")
36    if token not in tokens:
37        raise HTTPException(status_code=401, detail="Invalid token")
38    return tokens[token]
39 
40@app.post("/signup")
41def signup(req: SignupRequest):
42    global next_user_id
43    for u in users.values():
44        if u["username"] == req.username:
45            raise HTTPException(status_code=400, detail="Username taken")
46    user_id = next_user_id
47    next_user_id += 1
48    users[user_id] = {"id": user_id, "username": req.username, "password": req.password}
49    return {"id": user_id, "username": req.username}
50 
51@app.post("/login")
52def login(req: LoginRequest):
53    for u in users.values():
54        if u["username"] == req.username and u["password"] == req.password:
55            token = secrets.token_hex(16)
56            tokens[token] = u["id"]
57            return {"token": token}
58    raise HTTPException(status_code=401, detail="Invalid credentials")
59 
60@app.get("/users/{user_id}")
61def get_user(user_id: int, authorization: Optional[str] = Header(None)):
62    get_current_user(authorization)
63    if user_id not in users:
64        raise HTTPException(status_code=404, detail="User not found")
65    return users[user_id]
66 
67@app.post("/campaigns")
68def create_campaign(campaign: CampaignCreate, authorization: Optional[str] = Header(None)):
69    get_current_user(authorization)
70    global next_campaign_id
71    cid = next_campaign_id
72    next_campaign_id += 1
73    campaigns[cid] = {
74        "id": cid,
75        "title": campaign.title,
76        "goal": campaign.goal,
77        "raised": 0.0
78    }
79    return campaigns[cid]
80 
81@app.get("/campaigns/{campaign_id}")
82def get_campaign(campaign_id: int, authorization: Optional[str] = Header(None)):
83    get_current_user(authorization)
84    if campaign_id not in campaigns:
85        raise HTTPException(status_code=404, detail="Campaign not found")
86    return campaigns[campaign_id]
87 
88@app.post("/donations")
89def create_donation(donation: DonationCreate, authorization: Optional[str] = Header(None)):
90    user_id = get_current_user(authorization)
91    if donation.campaign_id not in campaigns:
92        raise HTTPException(status_code=404, detail="Campaign not found")
93    global next_donation_id
94    did = next_donation_id
95    next_donation_id += 1
96    donations[did] = {
97        "id": did,
98        "user_id": user_id,
99        "campaign_id": donation.campaign_id,
100        "amount": donation.amount
101    }
102    campaigns[donation.campaign_id]["raised"] += donation.amount
103    return donations[did]
104 
105@app.get("/donations/{donation_id}")
106def get_donation(donation_id: int, authorization: Optional[str] = Header(None)):
107    get_current_user(authorization)
108    if donation_id not in donations:
109        raise HTTPException(status_code=404, detail="Donation not found")
110    return donations[donation_id]
111 
112@app.get("/campaigns/{campaign_id}/progress")
113def get_progress(campaign_id: int, authorization: Optional[str] = Header(None)):
114    get_current_user(authorization)
115    if campaign_id not in campaigns:
116        raise HTTPException(status_code=404, detail="Campaign not found")
117    c = campaigns[campaign_id]
118    progress = min(c["raised"] / c["goal"] * 100, 100) if c["goal"] > 0 else 0
119    return {"campaign_id": campaign_id, "raised": c["raised"], "goal": c["goal"], "progress": round(progress, 2)}

1fastapi
2uvicorn