Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import secrets
4 
5app = FastAPI()
6 
7users = {}
8tokens = {}
9events = {}
10orders = {}
11tickets = {}
12 
13user_id_counter = 1
14event_id_counter = 1
15order_id_counter = 1
16ticket_id_counter = 1
17 
18def get_current_user(authorization: Optional[str] = Header(None)):
19    if not authorization:
20        raise HTTPException(status_code=401, detail="No auth header")
21    token = authorization.replace("Bearer ", "")
22    if token not in tokens:
23        raise HTTPException(status_code=401, detail="Invalid token")
24    return tokens[token]
25 
26@app.post("/signup")
27def signup(username: str, password: str, role: str = "user"):
28    global user_id_counter
29    user_id = user_id_counter
30    user_id_counter += 1
31    users[user_id] = {"id": user_id, "username": username, "password": password, "role": role}
32    return {"user_id": user_id, "username": username, "role": role}
33 
34@app.post("/login")
35def login(username: str, password: str):
36    for uid, u in users.items():
37        if u["username"] == username and u["password"] == password:
38            token = secrets.token_hex(16)
39            tokens[token] = uid
40            return {"token": token, "user_id": uid}
41    raise HTTPException(status_code=401, detail="Invalid credentials")
42 
43@app.post("/promote/{user_id}")
44def promote(user_id: int, authorization: Optional[str] = Header(None)):
45    current_user_id = get_current_user(authorization)
46    if users[current_user_id]["role"] != "admin":
47        raise HTTPException(status_code=403, detail="Only admin can promote")
48    if user_id not in users:
49        raise HTTPException(status_code=404, detail="User not found")
50    users[user_id]["role"] = "box_office_manager"
51    return {"user_id": user_id, "role": "box_office_manager"}
52 
53@app.post("/events")
54def create_event(name: str, date: str, total_tickets: int, price: float, authorization: Optional[str] = Header(None)):
55    current_user_id = get_current_user(authorization)
56    global event_id_counter
57    event_id = event_id_counter
58    event_id_counter += 1
59    events[event_id] = {
60        "id": event_id,
61        "name": name,
62        "date": date,
63        "total_tickets": total_tickets,
64        "available_tickets": total_tickets,
65        "price": price,
66        "created_by": current_user_id
67    }
68    return events[event_id]
69 
70@app.get("/events/{event_id}")
71def get_event(event_id: int):
72    if event_id not in events:
73        raise HTTPException(status_code=404, detail="Event not found")
74    return events[event_id]
75 
76@app.post("/orders")
77def create_order(event_id: int, quantity: int, authorization: Optional[str] = Header(None)):
78    current_user_id = get_current_user(authorization)
79    if event_id not in events:
80        raise HTTPException(status_code=404, detail="Event not found")
81    event = events[event_id]
82    if event["available_tickets"] < quantity:
83        raise HTTPException(status_code=400, detail="Not enough tickets")
84    total_price = event["price"] * quantity
85    global order_id_counter
86    order_id = order_id_counter
87    order_id_counter += 1
88    orders[order_id] = {
89        "id": order_id,
90        "user_id": current_user_id,
91        "event_id": event_id,
92        "quantity": quantity,
93        "total_price": total_price,
94        "status": "confirmed"
95    }
96    event["available_tickets"] -= quantity
97    for _ in range(quantity):
98        global ticket_id_counter
99        ticket_id = ticket_id_counter
100        ticket_id_counter += 1
101        tickets[ticket_id] = {
102            "id": ticket_id,
103            "order_id": order_id,
104            "event_id": event_id,
105            "owner_id": current_user_id
106        }
107    return orders[order_id]
108 
109@app.get("/orders/{order_id}")
110def get_order(order_id: int):
111    if order_id not in orders:
112        raise HTTPException(status_code=404, detail="Order not found")
113    return orders[order_id]
114 
115@app.post("/tickets/refund/{ticket_id}")
116def refund_ticket(ticket_id: int, authorization: Optional[str] = Header(None)):
117    current_user_id = get_current_user(authorization)
118    if users[current_user_id]["role"] not in ("admin", "box_office_manager"):
119        raise HTTPException(status_code=403, detail="Not authorized to refund")
120    if ticket_id not in tickets:
121        raise HTTPException(status_code=404, detail="Ticket not found")
122    ticket = tickets[ticket_id]
123    order_id = ticket["order_id"]
124    if order_id not in orders:
125        raise HTTPException(status_code=404, detail="Order not found")
126    order = orders[order_id]
127    order["quantity"] -= 1
128    event_id = ticket["event_id"]
129    if event_id in events:
130        events[event_id]["available_tickets"] += 1
131    del tickets[ticket_id]
132    return {"refunded": ticket_id}
133 
134@app.post("/events/{event_id}/pricing")
135def set_pricing(event_id: int, price: float, authorization: Optional[str] = Header(None)):
136    current_user_id = get_current_user(authorization)
137    if users[current_user_id]["role"] not in ("admin", "box_office_manager"):
138        raise HTTPException(status_code=403, detail="Not authorized to set pricing")
139    if event_id not in events:
140        raise HTTPException(status_code=404, detail="Event not found")
141    events[event_id]["price"] = price
142    return events[event_id]

1fastapi
2uvicorn