Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import secrets
5import uvicorn
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11playlists = {}
12playlist_id_counter = 0
13user_id_counter = 0
14 
15class SignupRequest(BaseModel):
16    username: str
17    password: str
18 
19class LoginRequest(BaseModel):
20    username: str
21    password: str
22 
23class CreatePlaylistRequest(BaseModel):
24    name: str
25    description: Optional[str] = ""
26    visibility: Optional[str] = "public"
27    collaborative: Optional[bool] = False
28    tracks: Optional[list] = []
29 
30class UpdatePlaylistRequest(BaseModel):
31    name: Optional[str] = None
32    description: Optional[str] = None
33    visibility: Optional[str] = None
34    collaborative: Optional[bool] = None
35    tracks: Optional[list] = None
36 
37def get_current_user(authorization: str = Header(None)):
38    if not authorization:
39        raise HTTPException(status_code=401, detail="Missing auth header")
40    token = authorization.replace("Bearer ", "")
41    user_id = tokens.get(token)
42    if not user_id:
43        raise HTTPException(status_code=401, detail="Invalid token")
44    return user_id
45 
46@app.post("/signup")
47def signup(req: SignupRequest):
48    global user_id_counter
49    for u in users.values():
50        if u["username"] == req.username:
51            raise HTTPException(status_code=400, detail="Username already exists")
52    user_id_counter += 1
53    user_id = user_id_counter
54    users[user_id] = {"username": req.username, "password": req.password}
55    token = secrets.token_hex(16)
56    tokens[token] = user_id
57    return {"user_id": user_id, "token": token}
58 
59@app.post("/login")
60def login(req: LoginRequest):
61    for uid, u in users.items():
62        if u["username"] == req.username and u["password"] == req.password:
63            token = secrets.token_hex(16)
64            tokens[token] = uid
65            return {"token": token}
66    raise HTTPException(status_code=401, detail="Invalid credentials")
67 
68@app.get("/playlists/{playlist_id}")
69def get_playlist(playlist_id: int):
70    playlist = playlists.get(playlist_id)
71    if not playlist:
72        raise HTTPException(status_code=404, detail="Playlist not found")
73    return playlist
74 
75@app.post("/playlists")
76def create_playlist(req: CreatePlaylistRequest, authorization: str = Header(None)):
77    global playlist_id_counter
78    user_id = get_current_user(authorization)
79    playlist_id_counter += 1
80    pid = playlist_id_counter
81    playlists[pid] = {
82        "id": pid,
83        "name": req.name,
84        "description": req.description,
85        "visibility": req.visibility,
86        "collaborative": req.collaborative,
87        "tracks": req.tracks or [],
88        "owner_id": user_id
89    }
90    return playlists[pid]
91 
92@app.patch("/playlists/{playlist_id}")
93def update_playlist(playlist_id: int, req: UpdatePlaylistRequest, authorization: str = Header(None)):
94    user_id = get_current_user(authorization)
95    playlist = playlists.get(playlist_id)
96    if not playlist:
97        raise HTTPException(status_code=404, detail="Playlist not found")
98    if playlist["owner_id"] != user_id:
99        raise HTTPException(status_code=403, detail="Not your playlist")
100    if req.name is not None:
101        playlist["name"] = req.name
102    if req.description is not None:
103        playlist["description"] = req.description
104    if req.visibility is not None:
105        playlist["visibility"] = req.visibility
106    if req.collaborative is not None:
107        playlist["collaborative"] = req.collaborative
108    if req.tracks is not None:
109        playlist["tracks"] = req.tracks
110    return playlist

1fastapi
2uvicorn