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: Dict[int, dict] = {}
9events: Dict[int, dict] = {}
10orders: Dict[int, dict] = {}
11tokens: Dict[str, int] = {}
12next_user_id = 1
13next_event_id = 1
14next_order_id = 1
15 
16class SignupRequest(BaseModel):
17    username: str
18    password: str
19 
20class LoginRequest(BaseModel):
21    username: str
22    password: str
23 
24class CreateEventRequest(BaseModel):
25    title: str
26    description: str
27    date: str
28    location: str
29    price: float
30    total_tickets: int
31 
32class EditEventRequest(BaseModel):
33    title: Optional[str] = None
34    description: Optional[str] = None
35    date: Optional[str] = None
36    location: Optional[str] = None
37    price: Optional[float] = None
38    total_tickets: Optional[int] = None
39 
40class BuyTicketRequest(BaseModel):
41    event_id: int
42    quantity: int = 1
43 
44class RefundRequest(BaseModel):
45    order_id: int
46 
47class PromoteCoHostRequest(BaseModel):
48    user_id: int
49 
50def get_user_from_token(authorization: str = Header(None)):
51    if not authorization:
52        raise HTTPException(status_code=401, detail="Missing auth header")
53    token = authorization.replace("Bearer ", "")
54    user_id = tokens.get(token)
55    if user_id is None:
56        raise HTTPException(status_code=401, detail="Invalid token")
57    return users[user_id]
58 
59@app.post("/signup")
60def signup(req: SignupRequest):
61    global next_user_id
62    uid = next_user_id
63    next_user_id += 1
64    users[uid] = {
65        "id": uid,
66        "username": req.username,
67        "password": req.password,
68        "co_host_events": []
69    }
70    return {"user_id": uid}
71 
72@app.post("/login")
73def login(req: LoginRequest):
74    for uid, u in users.items():
75        if u["username"] == req.username and u["password"] == req.password:
76            token = secrets.token_hex(16)
77            tokens[token] = uid
78            return {"token": token}
79    raise HTTPException(status_code=401, detail="Invalid credentials")
80 
81@app.get("/users/{user_id}")
82def get_user(user_id: int):
83    user = users.get(user_id)
84    if not user:
85        raise HTTPException(status_code=404, detail="User not found")
86    return user
87 
88@app.get("/events/{event_id}")
89def get_event(event_id: int):
90    event = events.get(event_id)
91    if not event:
92        raise HTTPException(status_code=404, detail="Event not found")
93    return event
94 
95@app.get("/orders/{order_id}")
96def get_order(order_id: int):
97    order = orders.get(order_id)
98    if not order:
99        raise HTTPException(status_code=404, detail="Order not found")
100    return order
101 
102@app.post("/events")
103def create_event(req: CreateEventRequest, authorization: str = Header(None)):
104    user = get_user_from_token(authorization)
105    global next_event_id
106    eid = next_event_id
107    next_event_id += 1
108    events[eid] = {
109        "id": eid,
110        "title": req.title,
111        "description": req.description,
112        "date": req.date,
113        "location": req.location,
114        "price": req.price,
115        "total_tickets": req.total_tickets,
116        "available_tickets": req.total_tickets,
117        "host_id": user["id"],
118        "co_hosts": []
119    }
120    return {"event_id": eid}
121 
122@app.post("/events/{event_id}/edit")
123def edit_event(event_id: int, req: EditEventRequest, authorization: str = Header(None)):
124    user = get_user_from_token(authorization)
125    event = events.get(event_id)
126    if not event:
127        raise HTTPException(status_code=404, detail="Event not found")
128    if user["id"] != event["host_id"] and user["id"] not in event["co_hosts"]:
129        raise HTTPException(status_code=403, detail="Not authorized")
130    if req.title is not None:
131        event["title"] = req.title
132    if req.description is not None:
133        event["description"] = req.description
134    if req.date is not None:
135        event["date"] = req.date
136    if req.location is not None:
137        event["location"] = req.location
138    if req.price is not None:
139        event["price"] = req.price
140    if req.total_tickets is not None:
141        diff = req.total_tickets - event["total_tickets"]
142        event["total_tickets"] = req.total_tickets
143        event["available_tickets"] += diff
144        if event["available_tickets"] < 0:
145            event["available_tickets"] = 0
146    return {"message": "Event updated"}
147 
148@app.post("/orders")
149def buy_ticket(req: BuyTicketRequest, authorization: str = Header(None)):
150    user = get_user_from_token(authorization)
151    event = events.get(req.event_id)
152    if not event:
153        raise HTTPException(status_code=404, detail="Event not found")
154    if event["available_tickets"] < req.quantity:
155        raise HTTPException(status_code=400, detail="Not enough tickets")
156    global next_order_id
157    oid = next_order_id
158    next_order_id += 1
159    event["available_tickets"] -= req.quantity
160    orders[oid] = {
161        "id": oid,
162        "event_id": req.event_id,
163        "buyer_id": user["id"],
164        "quantity": req.quantity,
165        "total_price": event["price"] * req.quantity
166    }
167    return {"order_id": oid}
168 
169@app.post("/orders/{order_id}/refund")
170def refund_ticket(order_id: int, authorization: str = Header(None)):
171    user = get_user_from_token(authorization)
172    order = orders.get(order_id)
173    if not order:
174        raise HTTPException(status_code=404, detail="Order not found")
175    event = events.get(order["event_id"])
176    if not event:
177        raise HTTPException(status_code=404, detail="Event not found")
178    if user["id"] != event["host_id"] and user["id"] not in event["co_hosts"]:
179        raise HTTPException(status_code=403, detail="Not authorized")
180    event["available_tickets"] += order["quantity"]
181    del orders[order_id]
182    return {"message": "Refund processed"}
183 
184@app.post("/events/{event_id}/promote")
185def promote_cohost(event_id: int, req: PromoteCoHostRequest, authorization: str = Header(None)):
186    user = get_user_from_token(authorization)
187    event = events.get(event_id)
188    if not event:
189        raise HTTPException(status_code=404, detail="Event not found")
190    if user["id"] != event["host_id"]:
191        raise HTTPException(status_code=403, detail="Only host can promote")
192    target_user = users.get(req.user_id)
193    if not target_user:
194        raise HTTPException(status_code=404, detail="User not found")
195    if req.user_id not in event["co_hosts"]:
196        event["co_hosts"].append(req.user_id)
197        target_user["co_host_events"].append(event_id)
198    return {"message": "Co-host added"}

1fastapi
2uvicorn