Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, List
4import random
5import string
6from datetime import datetime, timedelta
7 
8app = FastAPI()
9 
10users = {}
11tokens = {}
12artists = {}
13shows = {}
14artist_shows = {}
15show_id_counter = 1
16artist_id_counter = 1
17 
18def generate_token():
19    return ''.join(random.choices(string.ascii_letters + string.digits, k=32))
20 
21def get_current_user(authorization: str = Header(None)):
22    if not authorization:
23        raise HTTPException(status_code=401, detail="Missing auth header")
24    token = authorization.replace("Bearer ", "")
25    if token not in tokens:
26        raise HTTPException(status_code=401, detail="Invalid token")
27    return tokens[token]
28 
29class UserCreate(BaseModel):
30    username: str
31    password: str
32 
33class UserLogin(BaseModel):
34    username: str
35    password: str
36 
37class ArtistCreate(BaseModel):
38    name: str
39    genre: Optional[str] = None
40 
41class ShowCreate(BaseModel):
42    artist_id: int
43    venue: str
44    date: str
45    attendance: int
46    setlist: List[str]
47 
48class PastShowsResponse(BaseModel):
49    shows: list
50 
51class PredictResponse(BaseModel):
52    estimated_next_tour_dates: List[str]
53 
54@app.post("/signup")
55def signup(user: UserCreate):
56    if user.username in users:
57        raise HTTPException(status_code=400, detail="User already exists")
58    user_id = len(users) + 1
59    users[user.username] = {"id": user_id, "username": user.username, "password": user.password}
60    return {"id": user_id, "username": user.username}
61 
62@app.post("/login")
63def login(user: UserLogin):
64    if user.username not in users or users[user.username]["password"] != user.password:
65        raise HTTPException(status_code=401, detail="Invalid credentials")
66    token = generate_token()
67    tokens[token] = user.username
68    return {"token": token}
69 
70@app.post("/artists")
71def create_artist(artist: ArtistCreate, authorization: str = Header(None)):
72    current_user = get_current_user(authorization)
73    global artist_id_counter
74    artist_id = artist_id_counter
75    artist_id_counter += 1
76    artists[artist_id] = {"id": artist_id, "name": artist.name, "genre": artist.genre, "shows": []}
77    return artists[artist_id]
78 
79@app.get("/artists/{artist_id}")
80def get_artist(artist_id: int, authorization: str = Header(None)):
81    current_user = get_current_user(authorization)
82    if artist_id not in artists:
83        raise HTTPException(status_code=404, detail="Artist not found")
84    return artists[artist_id]
85 
86@app.post("/shows")
87def create_show(show: ShowCreate, authorization: str = Header(None)):
88    current_user = get_current_user(authorization)
89    global show_id_counter
90    if show.artist_id not in artists:
91        raise HTTPException(status_code=404, detail="Artist not found")
92    show_id = show_id_counter
93    show_id_counter += 1
94    show_data = {
95        "id": show_id,
96        "artist_id": show.artist_id,
97        "venue": show.venue,
98        "date": show.date,
99        "attendance": show.attendance,
100        "setlist": show.setlist
101    }
102    shows[show_id] = show_data
103    artists[show.artist_id]["shows"].append(show_id)
104    if show.artist_id not in artist_shows:
105        artist_shows[show.artist_id] = []
106    artist_shows[show.artist_id].append(show_data)
107    return show_data
108 
109@app.get("/shows/{show_id}")
110def get_show(show_id: int, authorization: str = Header(None)):
111    current_user = get_current_user(authorization)
112    if show_id not in shows:
113        raise HTTPException(status_code=404, detail="Show not found")
114    return shows[show_id]
115 
116@app.get("/artists/{artist_id}/past-shows")
117def get_past_shows(artist_id: int, authorization: str = Header(None)):
118    current_user = get_current_user(authorization)
119    if artist_id not in artists:
120        raise HTTPException(status_code=404, detail="Artist not found")
121    artist_shows_list = artist_shows.get(artist_id, [])
122    sorted_shows = sorted(artist_shows_list, key=lambda x: x["date"], reverse=True)
123    result = []
124    for s in sorted_shows:
125        result.append({
126            "setlist": s["setlist"],
127            "venue": s["venue"],
128            "attendance": s["attendance"],
129            "date": s["date"]
130        })
131    return {"shows": result}
132 
133@app.post("/shows/predict")
134def predict_tour_dates(authorization: str = Header(None)):
135    current_user = get_current_user(authorization)
136    if not artist_shows:
137        return {"estimated_next_tour_dates": []}
138    
139    all_dates = []
140    for artist_id, show_list in artist_shows.items():
141        for s in show_list:
142            try:
143                all_dates.append(datetime.strptime(s["date"], "%Y-%m-%d"))
144            except:
145                pass
146    
147    if not all_dates:
148        return {"estimated_next_tour_dates": []}
149    
150    all_dates.sort()
151    gaps = []
152    for i in range(1, len(all_dates)):
153        gap = (all_dates[i] - all_dates[i-1]).days
154        gaps.append(gap)
155    
156    avg_gap = sum(gaps) / len(gaps) if gaps else 90
157    last_date = all_dates[-1]
158    
159    next_dates = []
160    for i in range(3):
161        next_date = last_date + timedelta(days=int(avg_gap * (i + 1)))
162        next_dates.append(next_date.strftime("%Y-%m-%d"))
163    
164    return {"estimated_next_tour_dates": next_dates}

1fastapi
2uvicorn
3pydantic