Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, List
4import uuid
5 
6app = FastAPI()
7 
8# In-memory stores
9users = {}
10events = {}
11tokens = {}  # token -> user_id
12next_user_id = 1
13next_event_id = 1
14 
15class SignupRequest(BaseModel):
16    username: str
17    password: str
18 
19class LoginRequest(BaseModel):
20    username: str
21    password: str
22 
23class TicketTier(BaseModel):
24    name: str
25    price: float
26    quantity: int
27    is_vip: Optional[bool] = False
28 
29class Organizer(BaseModel):
30    name: str
31    role: str
32 
33class CreateEventRequest(BaseModel):
34    name: str
35    venue: str
36    ticket_tiers: List[TicketTier]
37    organizer: Organizer
38 
39class UpdateEventRequest(BaseModel):
40    name: Optional[str] = None
41    venue: Optional[str] = None
42    ticket_tiers: Optional[List[TicketTier]] = None
43    organizer: Optional[Organizer] = None
44 
45def get_current_user(authorization: str = Header(...)):
46    if not authorization.startswith("Bearer "):
47        raise HTTPException(status_code=401, detail="Invalid auth header")
48    token = authorization[7:]
49    user_id = tokens.get(token)
50    if user_id is None:
51        raise HTTPException(status_code=401, detail="Invalid token")
52    return user_id
53 
54@app.post("/signup")
55def signup(req: SignupRequest):
56    global next_user_id
57    # Check if username exists
58    for u in users.values():
59        if u["username"] == req.username:
60            raise HTTPException(status_code=400, detail="Username already exists")
61    user_id = next_user_id
62    next_user_id += 1
63    users[user_id] = {"id": user_id, "username": req.username, "password": req.password}
64    return {"id": user_id, "username": req.username}
65 
66@app.post("/login")
67def login(req: LoginRequest):
68    for u in users.values():
69        if u["username"] == req.username and u["password"] == req.password:
70            token = str(uuid.uuid4())
71            tokens[token] = u["id"]
72            return {"token": token}
73    raise HTTPException(status_code=401, detail="Invalid credentials")
74 
75@app.post("/events")
76def create_event(req: CreateEventRequest, authorization: str = Header(...)):
77    global next_event_id
78    user_id = get_current_user(authorization)
79    event_id = next_event_id
80    next_event_id += 1
81    events[event_id] = {
82        "id": event_id,
83        "name": req.name,
84        "venue": req.venue,
85        "ticket_tiers": [t.dict() for t in req.ticket_tiers],
86        "organizer": req.organizer.dict(),
87        "created_by": user_id
88    }
89    return events[event_id]
90 
91@app.get("/events/{event_id}")
92def get_event(event_id: int):
93    event = events.get(event_id)
94    if not event:
95        raise HTTPException(status_code=404, detail="Event not found")
96    return event
97 
98@app.put("/events/{event_id}")
99def update_event(event_id: int, req: UpdateEventRequest, authorization: str = Header(...)):
100    user_id = get_current_user(authorization)
101    event = events.get(event_id)
102    if not event:
103        raise HTTPException(status_code=404, detail="Event not found")
104    # Update fields if provided
105    if req.name is not None:
106        event["name"] = req.name
107    if req.venue is not None:
108        event["venue"] = req.venue
109    if req.ticket_tiers is not None:
110        event["ticket_tiers"] = [t.dict() for t in req.ticket_tiers]
111    if req.organizer is not None:
112        event["organizer"] = req.organizer.dict()
113    return event

1fastapi
2uvicorn