Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import secrets
5 
6app = FastAPI()
7 
8users = {}
9events = {}
10rsvps = {}
11tokens = {}
12next_user_id = 1
13next_event_id = 1
14next_rsvp_id = 1
15 
16class SignupRequest(BaseModel):
17    username: str
18    password: str
19 
20class LoginRequest(BaseModel):
21    username: str
22    password: str
23 
24class EventCreate(BaseModel):
25    title: str
26    description: str
27    capacity: int
28 
29class RSVPCreate(BaseModel):
30    event_id: int
31 
32def get_current_user(authorization: Optional[str] = Header(None)):
33    if not authorization:
34        raise HTTPException(status_code=401, detail="Missing auth header")
35    token = authorization.replace("Bearer ", "")
36    if token not in tokens:
37        raise HTTPException(status_code=401, detail="Invalid token")
38    return tokens[token]
39 
40@app.post("/signup")
41def signup(req: SignupRequest):
42    global next_user_id
43    for u in users.values():
44        if u["username"] == req.username:
45            raise HTTPException(status_code=400, detail="Username taken")
46    user_id = next_user_id
47    next_user_id += 1
48    users[user_id] = {"id": user_id, "username": req.username, "password": req.password}
49    return {"id": user_id, "username": req.username}
50 
51@app.post("/login")
52def login(req: LoginRequest):
53    user = None
54    for u in users.values():
55        if u["username"] == req.username and u["password"] == req.password:
56            user = u
57            break
58    if not user:
59        raise HTTPException(status_code=401, detail="Invalid credentials")
60    token = secrets.token_hex(16)
61    tokens[token] = user["id"]
62    return {"token": token}
63 
64@app.get("/users/{user_id}")
65def get_user(user_id: int):
66    if user_id not in users:
67        raise HTTPException(status_code=404, detail="User not found")
68    return users[user_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.get("/rsvps/{rsvp_id}")
77def get_rsvp(rsvp_id: int):
78    if rsvp_id not in rsvps:
79        raise HTTPException(status_code=404, detail="RSVP not found")
80    return rsvps[rsvp_id]
81 
82@app.post("/events")
83def create_event(event: EventCreate, authorization: Optional[str] = Header(None)):
84    user_id = get_current_user(authorization)
85    global next_event_id
86    event_id = next_event_id
87    next_event_id += 1
88    events[event_id] = {
89        "id": event_id,
90        "title": event.title,
91        "description": event.description,
92        "capacity": event.capacity,
93        "organizer_id": user_id,
94        "rsvp_count": 0
95    }
96    return events[event_id]
97 
98@app.post("/rsvps")
99def create_rsvp(rsvp: RSVPCreate, authorization: Optional[str] = Header(None)):
100    user_id = get_current_user(authorization)
101    if rsvp.event_id not in events:
102        raise HTTPException(status_code=404, detail="Event not found")
103    event = events[rsvp.event_id]
104    if event["rsvp_count"] >= event["capacity"]:
105        raise HTTPException(status_code=400, detail="Event is full")
106    # Check if user already RSVPed
107    for r in rsvps.values():
108        if r["user_id"] == user_id and r["event_id"] == rsvp.event_id:
109            raise HTTPException(status_code=400, detail="Already RSVPed")
110    global next_rsvp_id
111    rsvp_id = next_rsvp_id
112    next_rsvp_id += 1
113    rsvps[rsvp_id] = {
114        "id": rsvp_id,
115        "user_id": user_id,
116        "event_id": rsvp.event_id
117    }
118    events[rsvp.event_id]["rsvp_count"] += 1
119    return rsvps[rsvp_id]

1fastapi
2uvicorn