Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import secrets
5 
6app = FastAPI()
7 
8users = {}
9performers = {}
10ratings = {}
11user_tokens = {}
12next_user_id = 1
13next_performer_id = 1
14next_rating_id = 1
15 
16class SignupRequest(BaseModel):
17    username: str
18    password: str
19 
20class LoginRequest(BaseModel):
21    username: str
22    password: str
23 
24class PerformerCreate(BaseModel):
25    name: str
26    location: str
27    instrument: str
28    genre: str
29 
30class RatingCreate(BaseModel):
31    performer_id: int
32    score: int
33 
34def get_current_user(authorization: Optional[str] = Header(None)):
35    if not authorization:
36        raise HTTPException(status_code=401, detail="No auth header")
37    token = authorization.replace("Bearer ", "")
38    for uid, t in user_tokens.items():
39        if t == token:
40            return uid
41    raise HTTPException(status_code=401, detail="Invalid token")
42 
43@app.post("/signup")
44def signup(req: SignupRequest):
45    global next_user_id
46    for u in users.values():
47        if u["username"] == req.username:
48            raise HTTPException(status_code=400, detail="User exists")
49    uid = next_user_id
50    next_user_id += 1
51    users[uid] = {"username": req.username, "password": req.password}
52    token = secrets.token_hex(16)
53    user_tokens[uid] = token
54    return {"user_id": uid, "token": token}
55 
56@app.post("/login")
57def login(req: LoginRequest):
58    for uid, u in users.items():
59        if u["username"] == req.username and u["password"] == req.password:
60            token = secrets.token_hex(16)
61            user_tokens[uid] = token
62            return {"user_id": uid, "token": token}
63    raise HTTPException(status_code=401, detail="Invalid credentials")
64 
65@app.get("/performers/{performer_id}")
66def get_performer(performer_id: int):
67    if performer_id not in performers:
68        raise HTTPException(status_code=404, detail="Performer not found")
69    return performers[performer_id]
70 
71@app.post("/performers")
72def create_performer(req: PerformerCreate, authorization: Optional[str] = Header(None)):
73    get_current_user(authorization)
74    global next_performer_id
75    pid = next_performer_id
76    next_performer_id += 1
77    performers[pid] = {
78        "id": pid,
79        "name": req.name,
80        "location": req.location,
81        "instrument": req.instrument,
82        "genre": req.genre
83    }
84    return performers[pid]
85 
86@app.post("/ratings")
87def create_rating(req: RatingCreate, authorization: Optional[str] = Header(None)):
88    user_id = get_current_user(authorization)
89    if req.performer_id not in performers:
90        raise HTTPException(status_code=404, detail="Performer not found")
91    if req.score < 1 or req.score > 5:
92        raise HTTPException(status_code=400, detail="Score must be 1-5")
93    global next_rating_id
94    rid = next_rating_id
95    next_rating_id += 1
96    ratings[rid] = {
97        "id": rid,
98        "performer_id": req.performer_id,
99        "user_id": user_id,
100        "score": req.score
101    }
102    return ratings[rid]
103 
104@app.get("/performers/{performer_id}/ratings")
105def get_performer_ratings(performer_id: int):
106    if performer_id not in performers:
107        raise HTTPException(status_code=404, detail="Performer not found")
108    perfs = [r for r in ratings.values() if r["performer_id"] == performer_id]
109    return perfs

1fastapi
2uvicorn