Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, List
4import secrets
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10playlists = {}
11playlist_id_counter = 1
12 
13# Simple auth helper
14def get_current_user(authorization: str = Header(None)):
15    if not authorization:
16        raise HTTPException(status_code=401, detail="Missing auth header")
17    token = authorization.replace("Bearer ", "")
18    user_id = tokens.get(token)
19    if not user_id:
20        raise HTTPException(status_code=401, detail="Invalid token")
21    return user_id
22 
23class SignupRequest(BaseModel):
24    username: str
25    password: str
26 
27class LoginRequest(BaseModel):
28    username: str
29    password: str
30 
31class PlaylistCreate(BaseModel):
32    name: str
33    description: str = ""
34 
35class PlaylistUpdate(BaseModel):
36    name: Optional[str] = None
37    description: Optional[str] = None
38    track_order: Optional[List[int]] = None
39    collaborator_roles: Optional[dict] = None
40    tier: Optional[str] = None
41 
42@app.post("/signup")
43def signup(req: SignupRequest):
44    if req.username in users:
45        raise HTTPException(status_code=400, detail="User exists")
46    users[req.username] = {"password": req.password}
47    return {"ok": True}
48 
49@app.post("/login")
50def login(req: LoginRequest):
51    user = users.get(req.username)
52    if not user or user["password"] != req.password:
53        raise HTTPException(status_code=401, detail="Bad creds")
54    token = secrets.token_hex(16)
55    tokens[token] = req.username
56    return {"token": token}
57 
58@app.get("/playlists/{playlist_id}")
59def get_playlist(playlist_id: int, authorization: str = Header(None)):
60    user_id = get_current_user(authorization)
61    playlist = playlists.get(playlist_id)
62    if not playlist:
63        raise HTTPException(status_code=404, detail="Not found")
64    return playlist
65 
66@app.post("/playlists")
67def create_playlist(req: PlaylistCreate, authorization: str = Header(None)):
68    global playlist_id_counter
69    user_id = get_current_user(authorization)
70    playlist = {
71        "id": playlist_id_counter,
72        "name": req.name,
73        "description": req.description,
74        "owner": user_id,
75        "tracks": [],
76        "collaborator_roles": {},
77        "tier": "free"
78    }
79    playlists[playlist_id_counter] = playlist
80    playlist_id_counter += 1
81    return playlist
82 
83@app.put("/playlists/{playlist_id}")
84def update_playlist(playlist_id: int, req: PlaylistUpdate, authorization: str = Header(None)):
85    user_id = get_current_user(authorization)
86    playlist = playlists.get(playlist_id)
87    if not playlist:
88        raise HTTPException(status_code=404, detail="Not found")
89    if playlist["owner"] != user_id:
90        raise HTTPException(status_code=403, detail="Not your playlist")
91    if req.name is not None:
92        playlist["name"] = req.name
93    if req.description is not None:
94        playlist["description"] = req.description
95    if req.track_order is not None:
96        playlist["tracks"] = req.track_order
97    if req.collaborator_roles is not None:
98        playlist["collaborator_roles"] = req.collaborator_roles
99    if req.tier is not None:
100        playlist["tier"] = req.tier
101    return playlist

1fastapi
2uvicorn