Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from datetime import datetime, timedelta
4import secrets
5import threading
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11shows = {}
12tiers = {}
13reservations = {}
14reservation_locks = {}
15reservation_timers = {}
16 
17show_id_counter = 0
18tier_id_counter = 0
19reservation_id_counter = 0
20user_id_counter = 0
21 
22class SignupRequest(BaseModel):
23    username: str
24    password: str
25 
26class LoginRequest(BaseModel):
27    username: str
28    password: str
29 
30class ShowCreate(BaseModel):
31    name: str
32    date: str
33    venue: str
34 
35class TierCreate(BaseModel):
36    show_id: int
37    name: str
38    price: float
39    capacity: int
40 
41class ReserveRequest(BaseModel):
42    show_id: int
43    tier_id: int
44    quantity: int
45 
46def require_auth(authorization: str = Header(None)):
47    if not authorization:
48        raise HTTPException(status_code=401, detail="Missing auth token")
49    token = authorization.replace("Bearer ", "")
50    if token not in tokens:
51        raise HTTPException(status_code=401, detail="Invalid token")
52    return tokens[token]
53 
54@app.post("/signup")
55def signup(req: SignupRequest):
56    global user_id_counter
57    user_id_counter += 1
58    users[user_id_counter] = {"id": user_id_counter, "username": req.username, "password": req.password}
59    return {"id": user_id_counter, "username": req.username}
60 
61@app.post("/login")
62def login(req: LoginRequest):
63    for user in users.values():
64        if user["username"] == req.username and user["password"] == req.password:
65            token = secrets.token_hex(16)
66            tokens[token] = user["id"]
67            return {"token": token}
68    raise HTTPException(status_code=401, detail="Invalid credentials")
69 
70@app.get("/shows/{show_id}")
71def get_show(show_id: int):
72    if show_id not in shows:
73        raise HTTPException(status_code=404, detail="Show not found")
74    return shows[show_id]
75 
76@app.post("/shows")
77def create_show(show: ShowCreate):
78    global show_id_counter
79    show_id_counter += 1
80    shows[show_id_counter] = {"id": show_id_counter, "name": show.name, "date": show.date, "venue": show.venue}
81    return shows[show_id_counter]
82 
83@app.get("/tiers/{tier_id}")
84def get_tier(tier_id: int):
85    if tier_id not in tiers:
86        raise HTTPException(status_code=404, detail="Tier not found")
87    return tiers[tier_id]
88 
89@app.post("/tiers")
90def create_tier(tier: TierCreate):
91    global tier_id_counter
92    if tier.show_id not in shows:
93        raise HTTPException(status_code=404, detail="Show not found")
94    tier_id_counter += 1
95    tiers[tier_id_counter] = {"id": tier_id_counter, "show_id": tier.show_id, "name": tier.name, "price": tier.price, "capacity": tier.capacity, "reserved": 0}
96    return tiers[tier_id_counter]
97 
98@app.get("/reservations/{reservation_id}")
99def get_reservation(reservation_id: int, authorization: str = Header(None)):
100    user_id = require_auth(authorization)
101    if reservation_id not in reservations:
102        raise HTTPException(status_code=404, detail="Reservation not found")
103    return reservations[reservation_id]
104 
105@app.post("/reservations")
106def reserve(req: ReserveRequest, authorization: str = Header(None)):
107    global reservation_id_counter
108    user_id = require_auth(authorization)
109    if req.show_id not in shows:
110        raise HTTPException(status_code=404, detail="Show not found")
111    if req.tier_id not in tiers:
112        raise HTTPException(status_code=404, detail="Tier not found")
113    tier = tiers[req.tier_id]
114    if tier["show_id"] != req.show_id:
115        raise HTTPException(status_code=400, detail="Tier does not belong to this show")
116    
117    lock_key = f"{req.show_id}_{req.tier_id}"
118    if lock_key not in reservation_locks:
119        reservation_locks[lock_key] = threading.Lock()
120    
121    with reservation_locks[lock_key]:
122        available = tier["capacity"] - tier["reserved"]
123        if req.quantity > available:
124            raise HTTPException(status_code=400, detail="Not enough tickets available")
125        
126        reservation_id_counter += 1
127        expiry = datetime.utcnow() + timedelta(minutes=15)
128        reservations[reservation_id_counter] = {
129            "id": reservation_id_counter,
130            "user_id": user_id,
131            "show_id": req.show_id,
132            "tier_id": req.tier_id,
133            "quantity": req.quantity,
134            "expires_at": expiry.isoformat(),
135            "confirmed": False
136        }
137        tier["reserved"] += req.quantity
138        
139        timer = threading.Timer(900, release_reservation, args=[reservation_id_counter, req.tier_id, req.quantity])
140        reservation_timers[reservation_id_counter] = timer
141        timer.start()
142        
143        return reservations[reservation_id_counter]
144 
145def release_reservation(reservation_id, tier_id, quantity):
146    if reservation_id in reservations and not reservations[reservation_id]["confirmed"]:
147        lock_key = f"{reservations[reservation_id]['show_id']}_{tier_id}"
148        with reservation_locks.get(lock_key, threading.Lock()):
149            if reservation_id in reservations and not reservations[reservation_id]["confirmed"]:
150                tiers[tier_id]["reserved"] -= quantity
151                del reservations[reservation_id]

1fastapi
2uvicorn