Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, Dict, List
4import secrets
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10events = {}
11orders = {}
12next_user_id = 1
13next_event_id = 1
14next_order_id = 1
15next_seat_id = 1
16 
17class SignupRequest(BaseModel):
18    username: str
19    password: str
20 
21class LoginRequest(BaseModel):
22    username: str
23    password: str
24 
25class EventCreate(BaseModel):
26    name: str
27    date: str
28    total_seats: int
29 
30class OrderCreate(BaseModel):
31    event_id: int
32    seat_count: int
33 
34def get_current_user(authorization: Optional[str] = Header(None)):
35    if not authorization:
36        raise HTTPException(status_code=401, detail="Missing auth token")
37    token = authorization.replace("Bearer ", "")
38    if token not in tokens:
39        raise HTTPException(status_code=401, detail="Invalid token")
40    return tokens[token]
41 
42@app.post("/signup")
43def signup(req: SignupRequest):
44    global next_user_id
45    for u in users.values():
46        if u["username"] == req.username:
47            raise HTTPException(status_code=400, detail="Username already exists")
48    user_id = next_user_id
49    next_user_id += 1
50    users[user_id] = {"id": user_id, "username": req.username, "password": req.password}
51    return {"user_id": user_id}
52 
53@app.post("/login")
54def login(req: LoginRequest):
55    for u in users.values():
56        if u["username"] == req.username and u["password"] == req.password:
57            token = secrets.token_hex(16)
58            tokens[token] = u["id"]
59            return {"token": token}
60    raise HTTPException(status_code=401, detail="Invalid credentials")
61 
62@app.post("/events")
63def create_event(event: EventCreate, authorization: Optional[str] = Header(None)):
64    get_current_user(authorization)
65    global next_event_id
66    event_id = next_event_id
67    next_event_id += 1
68    events[event_id] = {
69        "id": event_id,
70        "name": event.name,
71        "date": event.date,
72        "total_seats": event.total_seats,
73        "available_seats": event.total_seats
74    }
75    return {"event_id": event_id}
76 
77@app.get("/events/{event_id}")
78def get_event(event_id: int, authorization: Optional[str] = Header(None)):
79    get_current_user(authorization)
80    if event_id not in events:
81        raise HTTPException(status_code=404, detail="Event not found")
82    return events[event_id]
83 
84@app.post("/orders")
85def create_order(order: OrderCreate, authorization: Optional[str] = Header(None)):
86    user_id = get_current_user(authorization)
87    global next_order_id, next_seat_id
88    if order.event_id not in events:
89        raise HTTPException(status_code=404, detail="Event not found")
90    event = events[order.event_id]
91    if event["available_seats"] < order.seat_count:
92        raise HTTPException(status_code=400, detail="Not enough seats available")
93    
94    order_id = next_order_id
95    next_order_id += 1
96    seat_ids = []
97    for _ in range(order.seat_count):
98        seat_id = next_seat_id
99        next_seat_id += 1
100        seat_ids.append(seat_id)
101    
102    event["available_seats"] -= order.seat_count
103    orders[order_id] = {
104        "id": order_id,
105        "user_id": user_id,
106        "event_id": order.event_id,
107        "seat_count": order.seat_count,
108        "seat_ids": seat_ids
109    }
110    return {"order_id": order_id, "seat_ids": seat_ids}
111 
112@app.get("/orders/{order_id}")
113def get_order(order_id: int, authorization: Optional[str] = Header(None)):
114    get_current_user(authorization)
115    if order_id not in orders:
116        raise HTTPException(status_code=404, detail="Order not found")
117    return orders[order_id]

1fastapi
2uvicorn