Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import random
4import string
5import hashlib
6import datetime
7 
8app = FastAPI()
9 
10users = {}
11tokens = {}
12playlists = {}
13songs = {}
14playlist_songs = {}
15shuffled_playlists = {}
16last_shuffle_date = {}
17playlist_id_counter = 1
18song_id_counter = 1
19user_id_counter = 1
20 
21def generate_token():
22    return ''.join(random.choices(string.ascii_letters + string.digits, k=32))
23 
24def hash_password(password):
25    return hashlib.sha256(password.encode()).hexdigest()
26 
27def verify_token(authorization: Optional[str] = Header(None)):
28    if not authorization:
29        raise HTTPException(status_code=401, detail="Missing authorization header")
30    token = authorization.replace("Bearer ", "")
31    if token not in tokens:
32        raise HTTPException(status_code=401, detail="Invalid token")
33    return tokens[token]
34 
35@app.post("/signup")
36def signup(username: str, password: str):
37    global user_id_counter
38    for user in users.values():
39        if user["username"] == username:
40            raise HTTPException(status_code=400, detail="Username already exists")
41    user_id = user_id_counter
42    user_id_counter += 1
43    users[user_id] = {"id": user_id, "username": username, "password": hash_password(password)}
44    token = generate_token()
45    tokens[token] = user_id
46    return {"user_id": user_id, "token": token}
47 
48@app.post("/login")
49def login(username: str, password: str):
50    for user in users.values():
51        if user["username"] == username and user["password"] == hash_password(password):
52            token = generate_token()
53            tokens[token] = user["id"]
54            return {"token": token}
55    raise HTTPException(status_code=401, detail="Invalid credentials")
56 
57@app.post("/playlists")
58def create_playlist(name: str, genre_tags: str, mood: str, authorization: Optional[str] = Header(None)):
59    user_id = verify_token(authorization)
60    global playlist_id_counter
61    playlist_id = playlist_id_counter
62    playlist_id_counter += 1
63    playlists[playlist_id] = {
64        "id": playlist_id,
65        "user_id": user_id,
66        "name": name,
67        "genre_tags": genre_tags.split(","),
68        "mood": mood,
69        "created_at": datetime.datetime.now().isoformat()
70    }
71    playlist_songs[playlist_id] = []
72    return playlists[playlist_id]
73 
74@app.get("/playlists/{playlist_id}")
75def get_playlist(playlist_id: int, authorization: Optional[str] = Header(None)):
76    verify_token(authorization)
77    if playlist_id not in playlists:
78        raise HTTPException(status_code=404, detail="Playlist not found")
79    return playlists[playlist_id]
80 
81@app.post("/songs")
82def create_song(title: str, artist: str, genre: str, authorization: Optional[str] = Header(None)):
83    verify_token(authorization)
84    global song_id_counter
85    song_id = song_id_counter
86    song_id_counter += 1
87    songs[song_id] = {
88        "id": song_id,
89        "title": title,
90        "artist": artist,
91        "genre": genre
92    }
93    return songs[song_id]
94 
95@app.get("/songs/{song_id}")
96def get_song(song_id: int, authorization: Optional[str] = Header(None)):
97    verify_token(authorization)
98    if song_id not in songs:
99        raise HTTPException(status_code=404, detail="Song not found")
100    return songs[song_id]
101 
102@app.post("/playlists/{playlist_id}/songs")
103def add_song_to_playlist(playlist_id: int, song_id: int, authorization: Optional[str] = Header(None)):
104    user_id = verify_token(authorization)
105    if playlist_id not in playlists:
106        raise HTTPException(status_code=404, detail="Playlist not found")
107    if playlists[playlist_id]["user_id"] != user_id:
108        raise HTTPException(status_code=403, detail="Not your playlist")
109    if song_id not in songs:
110        raise HTTPException(status_code=404, detail="Song not found")
111    if song_id in playlist_songs[playlist_id]:
112        raise HTTPException(status_code=400, detail="Song already in playlist")
113    playlist_songs[playlist_id].append(song_id)
114    return {"message": "Song added"}
115 
116@app.get("/playlists/{playlist_id}/songs")
117def get_playlist_songs(playlist_id: int, authorization: Optional[str] = Header(None)):
118    verify_token(authorization)
119    if playlist_id not in playlists:
120        raise HTTPException(status_code=404, detail="Playlist not found")
121    
122    today = datetime.date.today().isoformat()
123    if playlist_id not in last_shuffle_date or last_shuffle_date[playlist_id] != today:
124        song_ids = playlist_songs[playlist_id].copy()
125        random.shuffle(song_ids)
126        shuffled_playlists[playlist_id] = song_ids
127        last_shuffle_date[playlist_id] = today
128    
129    shuffled_ids = shuffled_playlists.get(playlist_id, [])
130    result = [songs[sid] for sid in shuffled_ids if sid in songs]
131    return result

1fastapi
2uvicorn