Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, List
4import secrets
5import time
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11projects = {}
12backings = {}
13next_user_id = 1
14next_project_id = 1
15next_backing_id = 1
16 
17class SignupRequest(BaseModel):
18    username: str
19    password: str
20 
21class LoginRequest(BaseModel):
22    username: str
23    password: str
24 
25class ProjectCreate(BaseModel):
26    title: str
27    description: str
28    goal: float
29    rewards: List[dict] = []
30 
31class BackingCreate(BaseModel):
32    amount: float
33 
34class PromoteRequest(BaseModel):
35    backer_id: int
36 
37def get_current_user(token: str = Header(None)):
38    if not token or token not in tokens:
39        raise HTTPException(status_code=401, detail="Invalid or missing token")
40    return tokens[token]
41 
42@app.post("/signup")
43def signup(req: SignupRequest):
44    global next_user_id
45    if any(u["username"] == req.username for u in users.values()):
46        raise HTTPException(status_code=400, detail="Username already exists")
47    user_id = next_user_id
48    next_user_id += 1
49    users[user_id] = {"id": user_id, "username": req.username, "password": req.password, "is_patron": False}
50    token = secrets.token_hex(16)
51    tokens[token] = user_id
52    return {"user_id": user_id, "token": token}
53 
54@app.post("/login")
55def login(req: LoginRequest):
56    for uid, u in users.items():
57        if u["username"] == req.username and u["password"] == req.password:
58            token = secrets.token_hex(16)
59            tokens[token] = uid
60            return {"user_id": uid, "token": token}
61    raise HTTPException(status_code=401, detail="Invalid credentials")
62 
63@app.post("/project")
64def create_project(project: ProjectCreate, token: str = Header(None)):
65    user_id = get_current_user(token)
66    global next_project_id
67    pid = next_project_id
68    next_project_id += 1
69    projects[pid] = {
70        "id": pid,
71        "owner_id": user_id,
72        "title": project.title,
73        "description": project.description,
74        "goal": project.goal,
75        "raised": 0.0,
76        "rewards": project.rewards,
77        "backers": [],
78        "patrons": []
79    }
80    return {"project_id": pid}
81 
82@app.get("/project/{project_id}")
83def get_project(project_id: int):
84    if project_id not in projects:
85        raise HTTPException(status_code=404, detail="Project not found")
86    return projects[project_id]
87 
88@app.post("/project/{project_id}/back")
89def back_project(project_id: int, backing: BackingCreate, token: str = Header(None)):
90    user_id = get_current_user(token)
91    if project_id not in projects:
92        raise HTTPException(status_code=404, detail="Project not found")
93    project = projects[project_id]
94    global next_backing_id
95    bid = next_backing_id
96    next_backing_id += 1
97    backings[bid] = {
98        "id": bid,
99        "project_id": project_id,
100        "backer_id": user_id,
101        "amount": backing.amount,
102        "timestamp": time.time()
103    }
104    project["raised"] += backing.amount
105    project["backers"].append(user_id)
106    return {"backing_id": bid, "raised": project["raised"], "goal": project["goal"]}
107 
108@app.post("/project/{project_id}/promote")
109def promote_backer(project_id: int, req: PromoteRequest, token: str = Header(None)):
110    user_id = get_current_user(token)
111    if project_id not in projects:
112        raise HTTPException(status_code=404, detail="Project not found")
113    project = projects[project_id]
114    if project["owner_id"] != user_id:
115        raise HTTPException(status_code=403, detail="Only the project owner can promote backers")
116    if req.backer_id not in project["backers"]:
117        raise HTTPException(status_code=400, detail="Backer not found in this project")
118    if req.backer_id in project["patrons"]:
119        raise HTTPException(status_code=400, detail="Already a patron")
120    project["patrons"].append(req.backer_id)
121    users[req.backer_id]["is_patron"] = True
122    return {"message": "Backer promoted to patron", "patron_id": req.backer_id}
123 
124@app.get("/user/{user_id}")
125def get_user(user_id: int):
126    if user_id not in users:
127        raise HTTPException(status_code=404, detail="User not found")
128    u = users[user_id].copy()
129    u.pop("password", None)
130    return u
131 
132@app.get("/backing/{backing_id}")
133def get_backing(backing_id: int):
134    if backing_id not in backings:
135        raise HTTPException(status_code=404, detail="Backing not found")
136    return backings[backing_id]

1fastapi
2uvicorn
3pydantic