Browse the data

1import requests
2import json
3import hashlib
4import secrets
5from fastapi import FastAPI, HTTPException, Header
6from pydantic import BaseModel, HttpUrl
7from typing import Optional
8 
9app = FastAPI()
10 
11users = {}
12tokens = {}
13playlists = {}
14playlist_id_counter = 1
15 
16class SignupRequest(BaseModel):
17    username: str
18    password: str
19 
20class LoginRequest(BaseModel):
21    username: str
22    password: str
23 
24class PlaylistCreate(BaseModel):
25    url: HttpUrl
26    name: Optional[str] = None
27 
28@app.post("/signup")
29def signup(req: SignupRequest):
30    if req.username in users:
31        raise HTTPException(status_code=400, detail="User exists")
32    users[req.username] = {"password": req.password}
33    return {"message": "User created"}
34 
35@app.post("/login")
36def login(req: LoginRequest):
37    user = users.get(req.username)
38    if not user or user["password"] != req.password:
39        raise HTTPException(status_code=401, detail="Invalid credentials")
40    token = secrets.token_hex(16)
41    tokens[token] = req.username
42    return {"token": token}
43 
44def get_current_user(authorization: Optional[str] = Header(None)):
45    if not authorization:
46        raise HTTPException(status_code=401, detail="Missing auth header")
47    token = authorization.replace("Bearer ", "")
48    user = tokens.get(token)
49    if not user:
50        raise HTTPException(status_code=401, detail="Invalid token")
51    return user
52 
53@app.post("/playlist/analyze")
54def analyze_playlist(req: PlaylistCreate, authorization: Optional[str] = Header(None)):
55    user = get_current_user(authorization)
56    global playlist_id_counter
57    try:
58        resp = requests.get(str(req.url))
59        data = resp.json()
60    except:
61        raise HTTPException(status_code=400, detail="Failed to fetch playlist")
62    
63    tracks = data.get("tracks", data.get("items", []))
64    if not tracks:
65        raise HTTPException(status_code=400, detail="No tracks found")
66    
67    tempos = []
68    for track in tracks:
69        t = track.get("track", track)
70        # Try to get tempo from Spotify's audio features if available
71        # For simplicity, we'll just return a mock average
72        # In real scenario you'd fetch audio features per track
73        if "tempo" in t:
74            tempos.append(t["tempo"])
75    
76    if not tempos:
77        # Mock tempos for demo
78        import random
79        tempos = [random.randint(60, 180) for _ in range(len(tracks))]
80    
81    avg_tempo = sum(tempos) / len(tempos)
82    
83    playlist_id = playlist_id_counter
84    playlist_id_counter += 1
85    playlists[playlist_id] = {
86        "url": str(req.url),
87        "name": req.name or data.get("name", "Unnamed"),
88        "average_tempo": round(avg_tempo, 2),
89        "track_count": len(tracks)
90    }
91    
92    return playlists[playlist_id]
93 
94@app.get("/playlist/{playlist_id}")
95def get_playlist(playlist_id: int, authorization: Optional[str] = Header(None)):
96    user = get_current_user(authorization)
97    p = playlists.get(playlist_id)
98    if not p:
99        raise HTTPException(status_code=404, detail="Playlist not found")
100    return p
101 
102@app.post("/playlist")
103def create_playlist(req: PlaylistCreate, authorization: Optional[str] = Header(None)):
104    user = get_current_user(authorization)
105    global playlist_id_counter
106    playlist_id = playlist_id_counter
107    playlist_id_counter += 1
108    playlists[playlist_id] = {
109        "url": str(req.url),
110        "name": req.name or "Unnamed",
111        "average_tempo": 0,
112        "track_count": 0
113    }
114    return playlists[playlist_id]

1fastapi
2uvicorn
3requests
4pydantic