Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import secrets
4import datetime
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10events = {}
11rsvps = {}
12next_user_id = 1
13next_event_id = 1
14next_rsvp_id = 1
15 
16def get_current_user(authorization: Optional[str] = Header(None)):
17    if not authorization:
18        raise HTTPException(status_code=401, detail="Missing auth header")
19    token = authorization.replace("Bearer ", "")
20    user_id = tokens.get(token)
21    if user_id is None:
22        raise HTTPException(status_code=401, detail="Invalid token")
23    return user_id
24 
25@app.post("/signup")
26def signup(username: str, password: str):
27    global next_user_id
28    for u in users.values():
29        if u["username"] == username:
30            raise HTTPException(status_code=400, detail="Username taken")
31    user_id = next_user_id
32    next_user_id += 1
33    users[user_id] = {"id": user_id, "username": username, "password": password}
34    return {"id": user_id, "username": username}
35 
36@app.post("/login")
37def login(username: str, password: str):
38    for u in users.values():
39        if u["username"] == username and u["password"] == password:
40            token = secrets.token_hex(16)
41            tokens[token] = u["id"]
42            return {"token": token}
43    raise HTTPException(status_code=401, detail="Invalid credentials")
44 
45@app.get("/events/{event_id}")
46def get_event(event_id: int, authorization: Optional[str] = Header(None)):
47    user_id = get_current_user(authorization)
48    event = events.get(event_id)
49    if not event:
50        raise HTTPException(status_code=404, detail="Event not found")
51    return event
52 
53@app.post("/events")
54def create_event(title: str, date: str, location: str, max_capacity: int, guest_list_visibility: bool, authorization: Optional[str] = Header(None)):
55    user_id = get_current_user(authorization)
56    global next_event_id
57    event_id = next_event_id
58    next_event_id += 1
59    events[event_id] = {
60        "id": event_id,
61        "title": title,
62        "date": date,
63        "location": location,
64        "max_capacity": max_capacity,
65        "guest_list_visibility": guest_list_visibility,
66        "created_by": user_id,
67        "guests": []
68    }
69    return events[event_id]
70 
71@app.put("/events/{event_id}")
72def update_event(event_id: int, title: Optional[str] = None, date: Optional[str] = None, location: Optional[str] = None, max_capacity: Optional[int] = None, guest_list_visibility: Optional[bool] = None, bulk_invite_statuses: Optional[dict] = None, authorization: Optional[str] = Header(None)):
73    user_id = get_current_user(authorization)
74    event = events.get(event_id)
75    if not event:
76        raise HTTPException(status_code=404, detail="Event not found")
77    if title is not None:
78        event["title"] = title
79    if date is not None:
80        event["date"] = date
81    if location is not None:
82        event["location"] = location
83    if max_capacity is not None:
84        event["max_capacity"] = max_capacity
85    if guest_list_visibility is not None:
86        event["guest_list_visibility"] = guest_list_visibility
87    if bulk_invite_statuses:
88        for guest_id, status in bulk_invite_statuses.items():
89            guest_id = int(guest_id)
90            for g in event["guests"]:
91                if g["user_id"] == guest_id:
92                    g["status"] = status
93    return event
94 
95@app.post("/events/{event_id}/rsvp")
96def create_rsvp(event_id: int, status: str, authorization: Optional[str] = Header(None)):
97    user_id = get_current_user(authorization)
98    event = events.get(event_id)
99    if not event:
100        raise HTTPException(status_code=404, detail="Event not found")
101    for g in event["guests"]:
102        if g["user_id"] == user_id:
103            g["status"] = status
104            return g
105    guest = {"user_id": user_id, "status": status}
106    event["guests"].append(guest)
107    return guest
108 
109@app.get("/events/{event_id}/rsvp/{rsvp_id}")
110def get_rsvp(event_id: int, rsvp_id: int, authorization: Optional[str] = Header(None)):
111    user_id = get_current_user(authorization)
112    event = events.get(event_id)
113    if not event:
114        raise HTTPException(status_code=404, detail="Event not found")
115    for g in event["guests"]:
116        if g["user_id"] == rsvp_id:
117            return g
118    raise HTTPException(status_code=404, detail="RSVP not found")

1fastapi
2uvicorn