Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional, List
3import secrets
4import datetime
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10events = {}
11orders = {}
12event_id_counter = 1
13order_id_counter = 1
14user_id_counter = 1
15 
16def get_current_user(authorization: Optional[str] = Header(None)):
17    if not authorization:
18        raise HTTPException(status_code=401, detail="Missing auth token")
19    token = authorization.replace("Bearer ", "")
20    if token not in tokens:
21        raise HTTPException(status_code=401, detail="Invalid token")
22    return tokens[token]
23 
24@app.post("/signup")
25def signup(username: str, password: str):
26    global user_id_counter
27    if any(u["username"] == username for u in users.values()):
28        raise HTTPException(status_code=400, detail="Username taken")
29    user_id = user_id_counter
30    user_id_counter += 1
31    users[user_id] = {"id": user_id, "username": username, "password": password}
32    return {"id": user_id, "username": username}
33 
34@app.post("/login")
35def login(username: str, password: str):
36    for u in users.values():
37        if u["username"] == username and u["password"] == password:
38            token = secrets.token_hex(16)
39            tokens[token] = u["id"]
40            return {"token": token}
41    raise HTTPException(status_code=401, detail="Invalid credentials")
42 
43@app.post("/events")
44def create_event(name: str, date: str, price: float, total_tickets: int, authorization: Optional[str] = Header(None)):
45    get_current_user(authorization)
46    global event_id_counter
47    event_id = event_id_counter
48    event_id_counter += 1
49    events[event_id] = {
50        "id": event_id,
51        "name": name,
52        "date": date,
53        "price": price,
54        "total_tickets": total_tickets,
55        "sold_tickets": 0
56    }
57    return events[event_id]
58 
59@app.get("/events/{event_id}")
60def get_event(event_id: int):
61    if event_id not in events:
62        raise HTTPException(status_code=404, detail="Event not found")
63    return events[event_id]
64 
65@app.get("/events/upcoming")
66def get_upcoming_events():
67    today = datetime.date.today().isoformat()
68    upcoming = []
69    for e in events.values():
70        if e["date"] >= today:
71            available = e["total_tickets"] - e["sold_tickets"]
72            upcoming.append({
73                "id": e["id"],
74                "name": e["name"],
75                "date": e["date"],
76                "available_tickets": available,
77                "price": e["price"]
78            })
79    return upcoming
80 
81@app.post("/orders")
82def create_order(event_id: int, quantity: int, authorization: Optional[str] = Header(None)):
83    user_id = get_current_user(authorization)
84    if event_id not in events:
85        raise HTTPException(status_code=404, detail="Event not found")
86    event = events[event_id]
87    if event["sold_tickets"] + quantity > event["total_tickets"]:
88        raise HTTPException(status_code=400, detail="Not enough tickets")
89    event["sold_tickets"] += quantity
90    global order_id_counter
91    order_id = order_id_counter
92    order_id_counter += 1
93    total_price = event["price"] * quantity
94    orders[order_id] = {
95        "id": order_id,
96        "user_id": user_id,
97        "event_id": event_id,
98        "quantity": quantity,
99        "total_price": total_price
100    }
101    return orders[order_id]
102 
103@app.get("/orders/{order_id}")
104def get_order(order_id: int, authorization: Optional[str] = Header(None)):
105    get_current_user(authorization)
106    if order_id not in orders:
107        raise HTTPException(status_code=404, detail="Order not found")
108    return orders[order_id]

1fastapi
2uvicorn