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 = 0
12 
13class UserCreate(BaseModel):
14    username: str
15    password: str
16 
17class UserLogin(BaseModel):
18    username: str
19    password: str
20 
21class Track(BaseModel):
22    id: int
23    title: str
24    artist: str
25    duration: int
26 
27class PlaylistCreate(BaseModel):
28    name: str
29    cover_art: Optional[str] = None
30    tracks: List[Track] = []
31    collaborative: bool = False
32    offline_download: bool = False
33 
34class PlaylistUpdate(BaseModel):
35    name: Optional[str] = None
36    cover_art: Optional[str] = None
37    tracks: Optional[List[Track]] = None
38    collaborative: Optional[bool] = None
39    offline_download: Optional[bool] = None
40 
41def get_current_user(authorization: str = Header(None)):
42    if not authorization:
43        raise HTTPException(status_code=401, detail="Missing auth header")
44    token = authorization.replace("Bearer ", "")
45    if token not in tokens:
46        raise HTTPException(status_code=401, detail="Invalid token")
47    return tokens[token]
48 
49@app.post("/signup")
50def signup(user: UserCreate):
51    if user.username in users:
52        raise HTTPException(status_code=400, detail="User exists")
53    users[user.username] = {"username": user.username, "password": user.password}
54    return {"msg": "User created"}
55 
56@app.post("/login")
57def login(user: UserLogin):
58    if user.username not in users or users[user.username]["password"] != user.password:
59        raise HTTPException(status_code=401, detail="Invalid credentials")
60    token = secrets.token_hex(16)
61    tokens[token] = user.username
62    return {"token": token}
63 
64@app.get("/playlists/{playlist_id}")
65def get_playlist(playlist_id: int, authorization: str = Header(None)):
66    current_user = get_current_user(authorization)
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(playlist: PlaylistCreate, authorization: str = Header(None)):
73    current_user = get_current_user(authorization)
74    global playlist_id_counter
75    playlist_id_counter += 1
76    playlists[playlist_id_counter] = {
77        "id": playlist_id_counter,
78        "name": playlist.name,
79        "cover_art": playlist.cover_art,
80        "tracks": [t.dict() for t in playlist.tracks],
81        "collaborative": playlist.collaborative,
82        "offline_download": playlist.offline_download,
83        "owner": current_user
84    }
85    return playlists[playlist_id_counter]
86 
87@app.put("/playlists/{playlist_id}")
88def update_playlist(playlist_id: int, update: PlaylistUpdate, authorization: str = Header(None)):
89    current_user = get_current_user(authorization)
90    if playlist_id not in playlists:
91        raise HTTPException(status_code=404, detail="Playlist not found")
92    playlist = playlists[playlist_id]
93    if not playlist["collaborative"] and playlist["owner"] != current_user:
94        raise HTTPException(status_code=403, detail="Not allowed")
95    if update.name is not None:
96        playlist["name"] = update.name
97    if update.cover_art is not None:
98        playlist["cover_art"] = update.cover_art
99    if update.tracks is not None:
100        playlist["tracks"] = [t.dict() for t in update.tracks]
101    if update.collaborative is not None:
102        playlist["collaborative"] = update.collaborative
103    if update.offline_download is not None:
104        playlist["offline_download"] = update.offline_download
105    return playlist

1fastapi
2uvicorn