Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import secrets
5import datetime
6 
7app = FastAPI()
8 
9# In-memory stores
10users = {}
11user_id_counter = 1
12tokens = {}
13events = {}
14event_id_counter = 1
15orders = {}
16order_id_counter = 1
17 
18class SignupRequest(BaseModel):
19    username: str
20    password: str
21 
22class LoginRequest(BaseModel):
23    username: str
24    password: str
25 
26class EventCreate(BaseModel):
27    name: str
28    date: str
29    venue: str
30    total_tickets: int
31    ticket_price: float
32 
33class OrderCreate(BaseModel):
34    event_id: int
35    quantity: int
36    token: Optional[str] = None
37 
38def get_current_user(authorization: str = Header(None)):
39    if not authorization:
40        raise HTTPException(status_code=401, detail="Missing authorization header")
41    token = authorization.replace("Bearer ", "")
42    if token not in tokens:
43        raise HTTPException(status_code=401, detail="Invalid token")
44    return tokens[token]
45 
46@app.post("/signup")
47def signup(req: SignupRequest):
48    global user_id_counter
49    for u in users.values():
50        if u["username"] == req.username:
51            raise HTTPException(status_code=400, detail="User already exists")
52    user_id = user_id_counter
53    user_id_counter += 1
54    users[user_id] = {"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    for u in users.values():
60        if u["username"] == req.username and u["password"] == req.password:
61            token = secrets.token_hex(16)
62            tokens[token] = u["id"]
63            return {"token": token}
64    raise HTTPException(status_code=401, detail="Invalid credentials")
65 
66@app.post("/events")
67def create_event(event: EventCreate, authorization: str = Header(None)):
68    get_current_user(authorization)
69    global event_id_counter
70    event_id = event_id_counter
71    event_id_counter += 1
72    events[event_id] = {
73        "id": event_id,
74        "name": event.name,
75        "date": event.date,
76        "venue": event.venue,
77        "total_tickets": event.total_tickets,
78        "ticket_price": event.ticket_price,
79        "available_tickets": event.total_tickets
80    }
81    return events[event_id]
82 
83@app.get("/events/{event_id}")
84def get_event(event_id: int):
85    if event_id not in events:
86        raise HTTPException(status_code=404, detail="Event not found")
87    return events[event_id]
88 
89@app.get("/events/upcoming")
90def get_upcoming_events():
91    today = datetime.date.today().isoformat()
92    upcoming = []
93    for e in events.values():
94        if e["date"] >= today:
95            upcoming.append({
96                "id": e["id"],
97                "name": e["name"],
98                "date": e["date"],
99                "venue": e["venue"],
100                "available_tickets": e["available_tickets"],
101                "price_range": {
102                    "min": e["ticket_price"],
103                    "max": e["ticket_price"]
104                }
105            })
106    return upcoming
107 
108@app.post("/orders")
109def create_order(order: OrderCreate, authorization: str = Header(None)):
110    user_id = get_current_user(authorization)
111    if order.event_id not in events:
112        raise HTTPException(status_code=404, detail="Event not found")
113    event = events[order.event_id]
114    if event["available_tickets"] < order.quantity:
115        raise HTTPException(status_code=400, detail="Not enough tickets available")
116    global order_id_counter
117    order_id = order_id_counter
118    order_id_counter += 1
119    total_price = order.quantity * event["ticket_price"]
120    orders[order_id] = {
121        "id": order_id,
122        "user_id": user_id,
123        "event_id": order.event_id,
124        "quantity": order.quantity,
125        "total_price": total_price
126    }
127    event["available_tickets"] -= order.quantity
128    return orders[order_id]
129 
130@app.get("/orders/{order_id}")
131def get_order(order_id: int, authorization: str = Header(None)):
132    get_current_user(authorization)
133    if order_id not in orders:
134        raise HTTPException(status_code=404, detail="Order not found")
135    return orders[order_id]
136 
137@app.get("/users/{user_id}")
138def get_user(user_id: int):
139    if user_id not in users:
140        raise HTTPException(status_code=404, detail="User not found")
141    return {"id": users[user_id]["id"], "username": users[user_id]["username"]}

1fastapi
2uvicorn