Browse the data

1import hashlib
2import secrets
3from fastapi import FastAPI, HTTPException, Header
4from pydantic import BaseModel
5 
6app = FastAPI()
7 
8users = {}
9sessions = {}
10tokens = {}
11 
12songs = {}
13releases = {}
14genres = {}
15 
16song_id_counter = 1
17release_id_counter = 1
18genre_id_counter = 1
19user_id_counter = 1
20 
21TRENDING_SONGS = [
22    {"title": "Blinding Lights", "artist": "The Weeknd", "genre": "pop"},
23    {"title": "Shape of You", "artist": "Ed Sheeran", "genre": "pop"},
24    {"title": "Bohemian Rhapsody", "artist": "Queen", "genre": "rock"},
25    {"title": "Stairway to Heaven", "artist": "Led Zeppelin", "genre": "rock"},
26    {"title": "Lose Yourself", "artist": "Eminem", "genre": "hip-hop"},
27    {"title": "Sicko Mode", "artist": "Travis Scott", "genre": "hip-hop"},
28    {"title": "Take Five", "artist": "Dave Brubeck", "genre": "jazz"},
29    {"title": "So What", "artist": "Miles Davis", "genre": "jazz"},
30    {"title": "Billie Jean", "artist": "Michael Jackson", "genre": "pop"},
31    {"title": "Smells Like Teen Spirit", "artist": "Nirvana", "genre": "rock"},
32]
33 
34NEW_RELEASES = [
35    {"title": "Flowers", "artist": "Miley Cyrus", "genre": "pop"},
36    {"title": "Kill Bill", "artist": "SZA", "genre": "r&b"},
37    {"title": "Creepin'", "artist": "Metro Boomin", "genre": "hip-hop"},
38    {"title": "Unholy", "artist": "Sam Smith", "genre": "pop"},
39    {"title": "Anti-Hero", "artist": "Taylor Swift", "genre": "pop"},
40]
41 
42GENRE_POPULARITY = {
43    "pop": 95,
44    "hip-hop": 88,
45    "rock": 72,
46    "r&b": 65,
47    "jazz": 40,
48    "electronic": 55,
49    "country": 50,
50    "latin": 60,
51}
52 
53class SignupRequest(BaseModel):
54    username: str
55    password: str
56 
57class LoginRequest(BaseModel):
58    username: str
59    password: str
60 
61class SongCreate(BaseModel):
62    title: str
63    artist: str
64    genre: str
65 
66class ReleaseCreate(BaseModel):
67    title: str
68    artist: str
69    genre: str
70 
71class GenreCreate(BaseModel):
72    name: str
73    popularity: int
74 
75def get_current_user(authorization: str = Header(None)):
76    if not authorization:
77        raise HTTPException(status_code=401, detail="Missing authorization header")
78    token = authorization.replace("Bearer ", "")
79    if token not in tokens:
80        raise HTTPException(status_code=401, detail="Invalid token")
81    return tokens[token]
82 
83@app.post("/signup")
84def signup(req: SignupRequest):
85    global user_id_counter
86    if any(u["username"] == req.username for u in users.values()):
87        raise HTTPException(status_code=400, detail="Username already exists")
88    user_id = user_id_counter
89    user_id_counter += 1
90    users[user_id] = {"id": user_id, "username": req.username, "password": hashlib.sha256(req.password.encode()).hexdigest()}
91    return {"id": user_id, "username": req.username}
92 
93@app.post("/login")
94def login(req: LoginRequest):
95    for uid, u in users.items():
96        if u["username"] == req.username and u["password"] == hashlib.sha256(req.password.encode()).hexdigest():
97            token = secrets.token_hex(32)
98            tokens[token] = uid
99            return {"token": token}
100    raise HTTPException(status_code=401, detail="Invalid credentials")
101 
102@app.get("/music/trending")
103def get_trending(authorization: str = Header(None)):
104    get_current_user(authorization)
105    return {
106        "top_songs": TRENDING_SONGS,
107        "new_releases": NEW_RELEASES,
108        "genre_popularity": GENRE_POPULARITY,
109    }
110 
111@app.get("/songs/{song_id}")
112def get_song(song_id: int, authorization: str = Header(None)):
113    get_current_user(authorization)
114    if song_id not in songs:
115        raise HTTPException(status_code=404, detail="Song not found")
116    return songs[song_id]
117 
118@app.post("/songs")
119def create_song(song: SongCreate, authorization: str = Header(None)):
120    global song_id_counter
121    get_current_user(authorization)
122    sid = song_id_counter
123    song_id_counter += 1
124    songs[sid] = {"id": sid, "title": song.title, "artist": song.artist, "genre": song.genre}
125    return songs[sid]
126 
127@app.get("/releases/{release_id}")
128def get_release(release_id: int, authorization: str = Header(None)):
129    get_current_user(authorization)
130    if release_id not in releases:
131        raise HTTPException(status_code=404, detail="Release not found")
132    return releases[release_id]
133 
134@app.post("/releases")
135def create_release(release: ReleaseCreate, authorization: str = Header(None)):
136    global release_id_counter
137    get_current_user(authorization)
138    rid = release_id_counter
139    release_id_counter += 1
140    releases[rid] = {"id": rid, "title": release.title, "artist": release.artist, "genre": release.genre}
141    return releases[rid]
142 
143@app.get("/genres/{genre_id}")
144def get_genre(genre_id: int, authorization: str = Header(None)):
145    get_current_user(authorization)
146    if genre_id not in genres:
147        raise HTTPException(status_code=404, detail="Genre not found")
148    return genres[genre_id]
149 
150@app.post("/genres")
151def create_genre(genre: GenreCreate, authorization: str = Header(None)):
152    global genre_id_counter
153    get_current_user(authorization)
154    gid = genre_id_counter
155    genre_id_counter += 1
156    genres[gid] = {"id": gid, "name": genre.name, "popularity": genre.popularity}
157    return genres[gid]

1fastapi
2uvicorn
3pydantic