Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import random
4import string
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10playlists = {}
11tracks = {}
12next_user_id = 1
13next_playlist_id = 1
14next_track_id = 1
15 
16def generate_token():
17    return ''.join(random.choices(string.ascii_letters + string.digits, k=32))
18 
19def get_user_from_token(authorization: str = Header(None)):
20    if not authorization:
21        raise HTTPException(status_code=401, detail="Missing auth token")
22    token = authorization.replace("Bearer ", "")
23    if token not in tokens:
24        raise HTTPException(status_code=401, detail="Invalid token")
25    return tokens[token]
26 
27@app.post("/signup")
28def signup(username: str, password: str):
29    global next_user_id
30    if username in users:
31        raise HTTPException(status_code=400, detail="User exists")
32    user_id = next_user_id
33    next_user_id += 1
34    users[username] = {"id": user_id, "username": username, "password": password}
35    return {"id": user_id, "username": username}
36 
37@app.post("/login")
38def login(username: str, password: str):
39    if username not in users or users[username]["password"] != password:
40        raise HTTPException(status_code=401, detail="Invalid credentials")
41    token = generate_token()
42    tokens[token] = users[username]["id"]
43    return {"token": token}
44 
45@app.get("/playlists/{playlist_id}")
46def get_playlist(playlist_id: int, authorization: str = Header(None)):
47    user_id = get_user_from_token(authorization)
48    if playlist_id not in playlists:
49        raise HTTPException(status_code=404, detail="Playlist not found")
50    return playlists[playlist_id]
51 
52@app.post("/playlists")
53def create_playlist(name: str, collaborative: bool = False, authorization: str = Header(None)):
54    global next_playlist_id
55    user_id = get_user_from_token(authorization)
56    playlist_id = next_playlist_id
57    next_playlist_id += 1
58    playlists[playlist_id] = {
59        "id": playlist_id,
60        "name": name,
61        "collaborative": collaborative,
62        "owner_id": user_id,
63        "tracks": []
64    }
65    return playlists[playlist_id]
66 
67@app.get("/tracks/{track_id}")
68def get_track(track_id: int, authorization: str = Header(None)):
69    user_id = get_user_from_token(authorization)
70    if track_id not in tracks:
71        raise HTTPException(status_code=404, detail="Track not found")
72    return tracks[track_id]
73 
74@app.post("/tracks")
75def create_track(title: str, artist: str, bpm: float, key: str, mood: str, duration_sec: int, authorization: str = Header(None)):
76    global next_track_id
77    user_id = get_user_from_token(authorization)
78    track_id = next_track_id
79    next_track_id += 1
80    tracks[track_id] = {
81        "id": track_id,
82        "title": title,
83        "artist": artist,
84        "bpm": bpm,
85        "key": key,
86        "mood": mood,
87        "duration_sec": duration_sec,
88        "added_by": user_id
89    }
90    return tracks[track_id]
91 
92@app.get("/playlists")
93def get_playlists(mood: Optional[str] = None, duration_max: Optional[int] = None, authorization: str = Header(None)):
94    user_id = get_user_from_token(authorization)
95    result = []
96    for pid, playlist in playlists.items():
97        filtered_tracks = []
98        for tid in playlist["tracks"]:
99            track = tracks[tid]
100            if mood:
101                moods = mood.split(",")
102                if track["mood"] not in moods:
103                    continue
104            if duration_max and track["duration_sec"] > duration_max:
105                continue
106            filtered_tracks.append(track)
107        result.append({
108            "id": playlist["id"],
109            "name": playlist["name"],
110            "collaborative": playlist["collaborative"],
111            "tracks": filtered_tracks
112        })
113    return result

1fastapi
2uvicorn