Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, Dict, List
4import secrets
5 
6app = FastAPI()
7 
8users = {}
9projects = {}
10pledges = {}
11stretch_goals = {}
12backer_promotions = {}
13auth_tokens = {}
14next_user_id = 1
15next_project_id = 1
16next_pledge_id = 1
17next_stretch_goal_id = 1
18next_promotion_id = 1
19 
20class SignupRequest(BaseModel):
21    username: str
22    password: str
23 
24class LoginRequest(BaseModel):
25    username: str
26    password: str
27 
28class ProjectCreate(BaseModel):
29    title: str
30    description: str
31    funding_goal: float
32 
33class PledgeCreate(BaseModel):
34    project_id: int
35    amount: float
36 
37class StretchGoalCreate(BaseModel):
38    project_id: int
39    target_amount: float
40    reward_description: str
41 
42class PromoteRequest(BaseModel):
43    backer_user_id: int
44 
45def get_current_user(authorization: Optional[str] = Header(None)):
46    if not authorization:
47        raise HTTPException(status_code=401, detail="Missing auth token")
48    token = authorization.replace("Bearer ", "")
49    if token not in auth_tokens:
50        raise HTTPException(status_code=401, detail="Invalid auth token")
51    return auth_tokens[token]
52 
53@app.post("/signup")
54def signup(req: SignupRequest):
55    global next_user_id
56    user_id = next_user_id
57    next_user_id += 1
58    users[user_id] = {"id": user_id, "username": req.username, "password": req.password}
59    return {"user_id": user_id, "message": "User created"}
60 
61@app.post("/login")
62def login(req: LoginRequest):
63    for user in users.values():
64        if user["username"] == req.username and user["password"] == req.password:
65            token = secrets.token_hex(32)
66            auth_tokens[token] = user["id"]
67            return {"token": token}
68    raise HTTPException(status_code=401, detail="Invalid credentials")
69 
70@app.post("/project")
71def create_project(req: ProjectCreate, authorization: Optional[str] = Header(None)):
72    user_id = get_current_user(authorization)
73    global next_project_id
74    project_id = next_project_id
75    next_project_id += 1
76    projects[project_id] = {
77        "id": project_id,
78        "title": req.title,
79        "description": req.description,
80        "funding_goal": req.funding_goal,
81        "creator_user_id": user_id,
82        "total_pledged": 0.0
83    }
84    return {"project_id": project_id}
85 
86@app.get("/project/{project_id}")
87def get_project(project_id: int):
88    project = projects.get(project_id)
89    if not project:
90        raise HTTPException(status_code=404, detail="Project not found")
91    total_pledged = sum(p["amount"] for p in pledges.values() if p["project_id"] == project_id)
92    project["total_pledged"] = total_pledged
93    progress = min(100, (total_pledged / project["funding_goal"]) * 100) if project["funding_goal"] > 0 else 0
94    return {
95        **project,
96        "funding_progress_percent": round(progress, 2)
97    }
98 
99@app.post("/pledge")
100def create_pledge(req: PledgeCreate, authorization: Optional[str] = Header(None)):
101    user_id = get_current_user(authorization)
102    if req.project_id not in projects:
103        raise HTTPException(status_code=404, detail="Project not found")
104    global next_pledge_id
105    pledge_id = next_pledge_id
106    next_pledge_id += 1
107    pledges[pledge_id] = {
108        "id": pledge_id,
109        "project_id": req.project_id,
110        "backer_user_id": user_id,
111        "amount": req.amount
112    }
113    project = projects[req.project_id]
114    project["total_pledged"] += req.amount
115    return {"pledge_id": pledge_id, "message": "Pledge created"}
116 
117@app.get("/pledge/{pledge_id}")
118def get_pledge(pledge_id: int):
119    pledge = pledges.get(pledge_id)
120    if not pledge:
121        raise HTTPException(status_code=404, detail="Pledge not found")
122    return pledge
123 
124@app.post("/stretch_goal")
125def create_stretch_goal(req: StretchGoalCreate, authorization: Optional[str] = Header(None)):
126    user_id = get_current_user(authorization)
127    if req.project_id not in projects:
128        raise HTTPException(status_code=404, detail="Project not found")
129    project = projects[req.project_id]
130    if project["creator_user_id"] != user_id:
131        raise HTTPException(status_code=403, detail="Only project creator can add stretch goals")
132    global next_stretch_goal_id
133    sg_id = next_stretch_goal_id
134    next_stretch_goal_id += 1
135    stretch_goals[sg_id] = {
136        "id": sg_id,
137        "project_id": req.project_id,
138        "target_amount": req.target_amount,
139        "reward_description": req.reward_description,
140        "reached": False
141    }
142    return {"stretch_goal_id": sg_id}
143 
144@app.get("/stretch_goal/{sg_id}")
145def get_stretch_goal(sg_id: int):
146    sg = stretch_goals.get(sg_id)
147    if not sg:
148        raise HTTPException(status_code=404, detail="Stretch goal not found")
149    project = projects.get(sg["project_id"])
150    if project and project["total_pledged"] >= sg["target_amount"]:
151        sg["reached"] = True
152    return sg
153 
154@app.post("/project/{project_id}/promote")
155def promote_backer(project_id: int, req: PromoteRequest, authorization: Optional[str] = Header(None)):
156    user_id = get_current_user(authorization)
157    project = projects.get(project_id)
158    if not project:
159        raise HTTPException(status_code=404, detail="Project not found")
160    if project["creator_user_id"] != user_id:
161        raise HTTPException(status_code=403, detail="Only project creator can promote backers")
162    backer = users.get(req.backer_user_id)
163    if not backer:
164        raise HTTPException(status_code=404, detail="Backer not found")
165    has_pledged = any(p["backer_user_id"] == req.backer_user_id and p["project_id"] == project_id for p in pledges.values())
166    if not has_pledged:
167        raise HTTPException(status_code=400, detail="User has not pledged to this project")
168    global next_promotion_id
169    promo_id = next_promotion_id
170    next_promotion_id += 1
171    backer_promotions[promo_id] = {
172        "id": promo_id,
173        "project_id": project_id,
174        "backer_user_id": req.backer_user_id,
175        "role": "co-producer"
176    }
177    return {"promotion_id": promo_id, "message": f"User {req.backer_user_id} promoted to co-producer"}
178 
179@app.get("/project/{project_id}/promotions")
180def get_promotions(project_id: int):
181    return [p for p in backer_promotions.values() if p["project_id"] == project_id]

1fastapi
2uvicorn