Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from fastapi.responses import FileResponse
3import os
4import hashlib
5import time
6import secrets
7 
8app = FastAPI()
9 
10users = {}
11tokens = {}
12orders = {}
13tickets = {}
14events = {}
15ticket_id_counter = 1
16event_id_counter = 1
17order_id_counter = 1
18user_id_counter = 1
19 
20def get_current_user(authorization: str = Header(None)):
21    if not authorization:
22        raise HTTPException(status_code=401, detail="Missing auth token")
23    token = authorization.replace("Bearer ", "")
24    if token not in tokens:
25        raise HTTPException(status_code=401, detail="Invalid token")
26    return tokens[token]
27 
28@app.post("/signup")
29def signup(username: str, password: str):
30    global user_id_counter
31    if any(u["username"] == username for u in users.values()):
32        raise HTTPException(status_code=400, detail="Username exists")
33    user_id = user_id_counter
34    user_id_counter += 1
35    users[user_id] = {"id": user_id, "username": username, "password": hashlib.sha256(password.encode()).hexdigest()}
36    return {"user_id": user_id, "username": username}
37 
38@app.post("/login")
39def login(username: str, password: str):
40    user = next((u for u in users.values() if u["username"] == username and u["password"] == hashlib.sha256(password.encode()).hexdigest()), None)
41    if not user:
42        raise HTTPException(status_code=401, detail="Invalid credentials")
43    token = secrets.token_hex(16)
44    tokens[token] = user["id"]
45    return {"token": token}
46 
47@app.post("/events")
48def create_event(name: str, authorization: str = Header(None)):
49    get_current_user(authorization)
50    global event_id_counter
51    event_id = event_id_counter
52    event_id_counter += 1
53    events[event_id] = {"id": event_id, "name": name, "tickets": {}}
54    return events[event_id]
55 
56@app.get("/events/{event_id}")
57def get_event(event_id: int, authorization: str = Header(None)):
58    get_current_user(authorization)
59    if event_id not in events:
60        raise HTTPException(status_code=404, detail="Event not found")
61    return events[event_id]
62 
63@app.post("/events/{event_id}/tickets")
64def create_ticket(event_id: int, price: float, authorization: str = Header(None)):
65    get_current_user(authorization)
66    if event_id not in events:
67        raise HTTPException(status_code=404, detail="Event not found")
68    global ticket_id_counter
69    ticket_id = ticket_id_counter
70    ticket_id_counter += 1
71    ticket = {"id": ticket_id, "event_id": event_id, "price": price, "status": "available"}
72    events[event_id]["tickets"][ticket_id] = ticket
73    tickets[ticket_id] = ticket
74    return ticket
75 
76@app.get("/events/{event_id}/tickets/{ticket_id}")
77def get_ticket(event_id: int, ticket_id: int, authorization: str = Header(None)):
78    get_current_user(authorization)
79    if event_id not in events or ticket_id not in events[event_id]["tickets"]:
80        raise HTTPException(status_code=404, detail="Ticket not found")
81    return events[event_id]["tickets"][ticket_id]
82 
83@app.post("/orders")
84def create_order(ticket_id: int, authorization: str = Header(None)):
85    user_id = get_current_user(authorization)
86    if ticket_id not in tickets:
87        raise HTTPException(status_code=404, detail="Ticket not found")
88    if tickets[ticket_id]["status"] != "available":
89        raise HTTPException(status_code=400, detail="Ticket not available")
90    global order_id_counter
91    order_id = order_id_counter
92    order_id_counter += 1
93    tickets[ticket_id]["status"] = "sold"
94    orders[order_id] = {"id": order_id, "ticket_id": ticket_id, "user_id": user_id, "status": "confirmed"}
95    return orders[order_id]
96 
97@app.get("/orders/{order_id}")
98def get_order(order_id: int, authorization: str = Header(None)):
99    get_current_user(authorization)
100    if order_id not in orders:
101        raise HTTPException(status_code=404, detail="Order not found")
102    return orders[order_id]
103 
104@app.get("/tickets/{event_id}/{ticket_id}/barcode")
105def get_barcode(event_id: int, ticket_id: int):
106    if event_id not in events or ticket_id not in events[event_id]["tickets"]:
107        raise HTTPException(status_code=404, detail="Ticket not found")
108    barcode_path = f"/tickets/barcodes/{ticket_id}.png"
109    if not os.path.exists(barcode_path):
110        raise HTTPException(status_code=404, detail="Barcode not found")
111    return FileResponse(barcode_path, media_type="image/png")

1fastapi
2uvicorn
3python-multipart