Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, Dict, List
4import uuid
5import datetime
6 
7app = FastAPI()
8 
9# In-memory stores
10users = {}
11tokens = {}
12events = {}
13orders = {}
14waitlist = {}
15event_id_counter = 1
16order_id_counter = 1
17 
18# Seed some events
19events[1] = {"id": 1, "name": "Rock Night", "date": "2025-03-15", "capacity": 100, "sold": 0}
20events[2] = {"id": 2, "name": "Jazz Evening", "date": "2025-03-20", "capacity": 50, "sold": 0}
21events[3] = {"id": 3, "name": "Comedy Hour", "date": "2025-03-25", "capacity": 75, "sold": 0}
22event_id_counter = 4
23 
24# Promo codes
25promo_codes = {"EARLYBIRD10": 10, "FRIENDS20": 20}
26 
27class SignupRequest(BaseModel):
28    username: str
29    password: str
30 
31class LoginRequest(BaseModel):
32    username: str
33    password: str
34 
35class PurchaseRequest(BaseModel):
36    event_id: int
37    quantity: int
38    promo_code: Optional[str] = None
39 
40def get_current_user(authorization: Optional[str] = Header(None)):
41    if not authorization:
42        raise HTTPException(status_code=401, detail="Missing auth token")
43    token = authorization.replace("Bearer ", "")
44    user_id = tokens.get(token)
45    if user_id is None:
46        raise HTTPException(status_code=401, detail="Invalid token")
47    return user_id
48 
49@app.post("/signup")
50def signup(req: SignupRequest):
51    if req.username in users:
52        raise HTTPException(status_code=400, detail="User already exists")
53    user_id = len(users) + 1
54    users[req.username] = {"id": user_id, "username": req.username, "password": req.password}
55    return {"id": user_id, "username": req.username}
56 
57@app.post("/login")
58def login(req: LoginRequest):
59    user = users.get(req.username)
60    if not user or user["password"] != req.password:
61        raise HTTPException(status_code=401, detail="Invalid credentials")
62    token = str(uuid.uuid4())
63    tokens[token] = user["id"]
64    return {"token": token}
65 
66@app.get("/events")
67def list_events():
68    return [e for e in events.values()]
69 
70@app.post("/purchase")
71def purchase(req: PurchaseRequest, authorization: Optional[str] = Header(None)):
72    user_id = get_current_user(authorization)
73    event = events.get(req.event_id)
74    if not event:
75        raise HTTPException(status_code=404, detail="Event not found")
76    
77    discount = 0
78    if req.promo_code:
79        discount = promo_codes.get(req.promo_code, 0)
80    
81    if event["sold"] + req.quantity > event["capacity"]:
82        # Add to waitlist
83        if req.event_id not in waitlist:
84            waitlist[req.event_id] = []
85        waitlist[req.event_id].append({"user_id": user_id, "quantity": req.quantity})
86        return {"message": "Sold out, added to waitlist", "waitlist_position": len(waitlist[req.event_id])}
87    
88    event["sold"] += req.quantity
89    global order_id_counter
90    order_id = order_id_counter
91    order_id_counter += 1
92    orders[order_id] = {
93        "id": order_id,
94        "user_id": user_id,
95        "event_id": req.event_id,
96        "quantity": req.quantity,
97        "discount": discount
98    }
99    return {"order_id": order_id, "tickets_purchased": req.quantity, "discount_applied": discount}
100 
101@app.get("/waitlist")
102def get_waitlist():
103    result = {}
104    for event_id, entries in waitlist.items():
105        result[event_id] = len(entries)
106    return result
107 
108@app.get("/events/{event_id}")
109def get_event(event_id: int):
110    event = events.get(event_id)
111    if not event:
112        raise HTTPException(status_code=404, detail="Event not found")
113    return event
114 
115@app.post("/events")
116def create_event(name: str, date: str, capacity: int):
117    global event_id_counter
118    event_id = event_id_counter
119    event_id_counter += 1
120    events[event_id] = {
121        "id": event_id,
122        "name": name,
123        "date": date,
124        "capacity": capacity,
125        "sold": 0
126    }
127    return events[event_id]
128 
129@app.get("/orders/{order_id}")
130def get_order(order_id: int, authorization: Optional[str] = Header(None)):
131    user_id = get_current_user(authorization)
132    order = orders.get(order_id)
133    if not order:
134        raise HTTPException(status_code=404, detail="Order not found")
135    return order
136 
137@app.post("/orders")
138def create_order(event_id: int, quantity: int, authorization: Optional[str] = Header(None)):
139    user_id = get_current_user(authorization)
140    event = events.get(event_id)
141    if not event:
142        raise HTTPException(status_code=404, detail="Event not found")
143    if event["sold"] + quantity > event["capacity"]:
144        raise HTTPException(status_code=400, detail="Not enough tickets")
145    event["sold"] += quantity
146    global order_id_counter
147    order_id = order_id_counter
148    order_id_counter += 1
149    orders[order_id] = {
150        "id": order_id,
151        "user_id": user_id,
152        "event_id": event_id,
153        "quantity": quantity
154    }
155    return orders[order_id]

1fastapi
2uvicorn