Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import secrets
4from datetime import datetime
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10songs = {}
11playlists = {}
12next_user_id = 1
13next_song_id = 1
14next_playlist_id = 1
15 
16# Trending music data (hardcoded for speed)
17trending_top_songs = [
18    {"id": 1, "title": "Blinding Lights", "artist": "The Weeknd", "streams": 1500000},
19    {"id": 2, "title": "Heat Waves", "artist": "Glass Animals", "streams": 1200000},
20    {"id": 3, "title": "As It Was", "artist": "Harry Styles", "streams": 1100000},
21    {"id": 4, "title": "Stay", "artist": "Justin Bieber & The Kid LAROI", "streams": 1000000},
22    {"id": 5, "title": "Good 4 U", "artist": "Olivia Rodrigo", "streams": 950000}
23]
24 
25new_releases = [
26    {"id": 1, "title": "Flowers", "artist": "Miley Cyrus", "release_date": "2024-01-12"},
27    {"id": 2, "title": "Kill Bill", "artist": "SZA", "release_date": "2024-01-15"},
28    {"id": 3, "title": "Creepin'", "artist": "Metro Boomin", "release_date": "2024-01-20"},
29    {"id": 4, "title": "Unholy", "artist": "Sam Smith", "release_date": "2024-02-01"},
30    {"id": 5, "title": "Anti-Hero", "artist": "Taylor Swift", "release_date": "2024-02-05"}
31]
32 
33genre_popularity = {
34    "pop": {"rank": 1, "share": 35.2},
35    "hip-hop": {"rank": 2, "share": 22.8},
36    "rock": {"rank": 3, "share": 15.1},
37    "electronic": {"rank": 4, "share": 9.7},
38    "r&b": {"rank": 5, "share": 8.3}
39}
40 
41def get_current_user(authorization: Optional[str] = Header(None)):
42    if not authorization:
43        raise HTTPException(status_code=401, detail="No auth token")
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(username: str, password: str):
51    global next_user_id
52    for user in users.values():
53        if user["username"] == username:
54            raise HTTPException(status_code=400, detail="Username taken")
55    user_id = next_user_id
56    next_user_id += 1
57    users[user_id] = {"id": user_id, "username": username, "password": password}
58    return {"id": user_id, "username": username}
59 
60@app.post("/login")
61def login(username: str, password: str):
62    for user in users.values():
63        if user["username"] == username and user["password"] == password:
64            token = secrets.token_hex(16)
65            tokens[token] = user["id"]
66            return {"token": token}
67    raise HTTPException(status_code=401, detail="Invalid credentials")
68 
69@app.get("/music/trending")
70def get_trending(authorization: Optional[str] = Header(None)):
71    get_current_user(authorization)
72    return {
73        "top_songs": trending_top_songs,
74        "new_releases": new_releases,
75        "genre_popularity": genre_popularity
76    }
77 
78@app.get("/songs/{song_id}")
79def get_song(song_id: int, authorization: Optional[str] = Header(None)):
80    get_current_user(authorization)
81    if song_id not in songs:
82        raise HTTPException(status_code=404, detail="Song not found")
83    return songs[song_id]
84 
85@app.post("/songs")
86def create_song(title: str, artist: str, authorization: Optional[str] = Header(None)):
87    get_current_user(authorization)
88    global next_song_id
89    song_id = next_song_id
90    next_song_id += 1
91    songs[song_id] = {"id": song_id, "title": title, "artist": artist, "created_at": datetime.now().isoformat()}
92    return songs[song_id]
93 
94@app.get("/playlists/{playlist_id}")
95def get_playlist(playlist_id: int, authorization: Optional[str] = Header(None)):
96    get_current_user(authorization)
97    if playlist_id not in playlists:
98        raise HTTPException(status_code=404, detail="Playlist not found")
99    return playlists[playlist_id]
100 
101@app.post("/playlists")
102def create_playlist(name: str, authorization: Optional[str] = Header(None)):
103    get_current_user(authorization)
104    global next_playlist_id
105    playlist_id = next_playlist_id
106    next_playlist_id += 1
107    playlists[playlist_id] = {"id": playlist_id, "name": name, "songs": [], "created_at": datetime.now().isoformat()}
108    return playlists[playlist_id]

1fastapi
2uvicorn