Browse the data

1import os
2import secrets
3from fastapi import FastAPI, HTTPException, Header
4from pydantic import BaseModel
5from typing import Optional
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11setlists = {}
12setlist_id_counter = 1
13user_id_counter = 1
14 
15class SignupRequest(BaseModel):
16    username: str
17    password: str
18 
19class LoginRequest(BaseModel):
20    username: str
21    password: str
22 
23class SetlistCreate(BaseModel):
24    artist: str
25    date: str
26    setlist: str
27 
28def get_current_user(authorization: Optional[str] = Header(None)):
29    if not authorization or not authorization.startswith("Bearer "):
30        raise HTTPException(status_code=401, detail="Invalid auth header")
31    token = authorization.split(" ")[1]
32    if token not in tokens:
33        raise HTTPException(status_code=401, detail="Invalid token")
34    return tokens[token]
35 
36@app.post("/signup")
37def signup(req: SignupRequest):
38    global user_id_counter
39    if any(u["username"] == req.username for u in users.values()):
40        raise HTTPException(status_code=400, detail="Username already exists")
41    user_id = user_id_counter
42    user_id_counter += 1
43    users[user_id] = {"id": user_id, "username": req.username, "password": req.password}
44    return {"id": user_id, "username": req.username}
45 
46@app.post("/login")
47def login(req: LoginRequest):
48    for u in users.values():
49        if u["username"] == req.username and u["password"] == req.password:
50            token = secrets.token_hex(16)
51            tokens[token] = u["id"]
52            return {"token": token}
53    raise HTTPException(status_code=401, detail="Invalid credentials")
54 
55@app.get("/setlists/{setlist_id}")
56def get_setlist(setlist_id: int, authorization: Optional[str] = Header(None)):
57    current_user = get_current_user(authorization)
58    if setlist_id not in setlists:
59        raise HTTPException(status_code=404, detail="Setlist not found")
60    return setlists[setlist_id]
61 
62@app.post("/setlists")
63def create_setlist(req: SetlistCreate, authorization: Optional[str] = Header(None)):
64    global setlist_id_counter
65    current_user = get_current_user(authorization)
66    setlist_id = setlist_id_counter
67    setlist_id_counter += 1
68    setlists[setlist_id] = {
69        "id": setlist_id,
70        "artist": req.artist,
71        "date": req.date,
72        "setlist": req.setlist,
73        "user_id": current_user
74    }
75    return {"id": setlist_id}
76 
77@app.get("/concerts/{artist}/{date}/setlist")
78def get_concert_setlist(artist: str, date: str, authorization: Optional[str] = Header(None)):
79    current_user = get_current_user(authorization)
80    for sl in setlists.values():
81        if sl["artist"] == artist and sl["date"] == date:
82            pdf_path = f"/setlists/{artist}/{date}.pdf"
83            if os.path.exists(pdf_path):
84                return {"pdf_url": pdf_path, "setlist": sl["setlist"]}
85            else:
86                raise HTTPException(status_code=404, detail="PDF not found")
87    raise HTTPException(status_code=404, detail="Setlist not found")

1fastapi
2uvicorn