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 = {}
10playlists = {}
11playlist_id_counter = 1
12user_id_counter = 1
13tokens = {}
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    curator_role: Optional[str] = "owner"
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    curator_role: Optional[str] = None
36 
37def get_current_user(authorization: str = Header(...)):
38    if not authorization.startswith("Bearer "):
39        raise HTTPException(status_code=401, detail="Invalid auth header")
40    token = authorization[7:]
41    if token not in tokens:
42        raise HTTPException(status_code=401, detail="Invalid token")
43    return tokens[token]
44 
45@app.post("/signup")
46def signup(req: SignupRequest):
47    global user_id_counter
48    for u in users.values():
49        if u["username"] == req.username:
50            raise HTTPException(status_code=400, detail="Username already exists")
51    user_id = user_id_counter
52    user_id_counter += 1
53    users[user_id] = {"id": user_id, "username": req.username, "password": req.password}
54    return {"id": user_id, "username": req.username}
55 
56@app.post("/login")
57def login(req: LoginRequest):
58    for u in users.values():
59        if u["username"] == req.username and u["password"] == req.password:
60            token = secrets.token_hex(32)
61            tokens[token] = u["id"]
62            return {"token": token}
63    raise HTTPException(status_code=401, detail="Invalid credentials")
64 
65@app.get("/playlists/{playlist_id}")
66def get_playlist(playlist_id: int):
67    if playlist_id not in playlists:
68        raise HTTPException(status_code=404, detail="Playlist not found")
69    return playlists[playlist_id]
70 
71@app.post("/playlists")
72def create_playlist(req: CreatePlaylistRequest, user_id: int = None):
73    global playlist_id_counter
74    if user_id is None:
75        raise HTTPException(status_code=401, detail="Authentication required")
76    playlist_id = playlist_id_counter
77    playlist_id_counter += 1
78    playlists[playlist_id] = {
79        "id": playlist_id,
80        "name": req.name,
81        "description": req.description,
82        "visibility": req.visibility,
83        "collaborative": req.collaborative,
84        "curator_role": req.curator_role,
85        "owner_id": user_id
86    }
87    return playlists[playlist_id]
88 
89@app.patch("/playlists/{playlist_id}")
90def update_playlist(playlist_id: int, req: UpdatePlaylistRequest, user_id: int = None):
91    if playlist_id not in playlists:
92        raise HTTPException(status_code=404, detail="Playlist not found")
93    if user_id is None:
94        raise HTTPException(status_code=401, detail="Authentication required")
95    playlist = playlists[playlist_id]
96    if playlist["owner_id"] != user_id:
97        raise HTTPException(status_code=403, detail="Not authorized to update this playlist")
98    if req.name is not None:
99        playlist["name"] = req.name
100    if req.description is not None:
101        playlist["description"] = req.description
102    if req.visibility is not None:
103        playlist["visibility"] = req.visibility
104    if req.collaborative is not None:
105        playlist["collaborative"] = req.collaborative
106    if req.curator_role is not None:
107        playlist["curator_role"] = req.curator_role
108    return playlist

1fastapi
2uvicorn