Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import secrets
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10game_nights = {}
11rsvps = {}
12 
13user_counter = 0
14night_counter = 0
15rsvp_counter = 0
16 
17 
18class SignupRequest(BaseModel):
19    username: str
20    password: str
21 
22 
23class LoginRequest(BaseModel):
24    username: str
25    password: str
26 
27 
28def get_user_from_token(authorization: Optional[str]):
29    if not authorization:
30        raise HTTPException(status_code=401, detail="Missing token")
31    token = authorization.replace("Bearer ", "").strip()
32    user_id = tokens.get(token)
33    if user_id is None:
34        raise HTTPException(status_code=401, detail="Invalid token")
35    return user_id
36 
37 
38@app.post("/signup")
39def signup(req: dict):
40    global user_counter
41    username = req.get("username")
42    password = req.get("password")
43    if not username or not password:
44        raise HTTPException(status_code=400, detail="username and password required")
45    for u in users.values():
46        if u["username"] == username:
47            raise HTTPException(status_code=400, detail="username taken")
48    user_counter += 1
49    user = {
50        "id": user_counter,
51        "username": username,
52        "password": password,
53        "role": "user",
54        "is_admin": False,
55    }
56    for k, v in req.items():
57        if k not in ("username", "password"):
58            user[k] = v
59    users[user_counter] = user
60    return {"id": user_counter, "username": username, "role": user["role"], "is_admin": user["is_admin"]}
61 
62 
63@app.post("/login")
64def login(req: LoginRequest):
65    for user in users.values():
66        if user["username"] == req.username and user["password"] == req.password:
67            token = secrets.token_hex(16)
68            tokens[token] = user["id"]
69            return {"token": token}
70    raise HTTPException(status_code=401, detail="invalid credentials")
71 
72 
73@app.post("/game_nights")
74def create_game_night(req: dict, authorization: Optional[str] = Header(None)):
75    global night_counter
76    user_id = get_user_from_token(authorization)
77    night_counter += 1
78    night = {
79        "id": night_counter,
80        "user_id": user_id,
81        "title": req.get("title"),
82        "game": req.get("game"),
83        "location": req.get("location"),
84        "date": req.get("date"),
85        "max_players": req.get("max_players"),
86    }
87    for k, v in req.items():
88        if k not in night:
89            night[k] = v
90    game_nights[night_counter] = night
91    return night
92 
93 
94@app.get("/game_nights/{night_id}")
95def get_game_night(night_id: int):
96    night = game_nights.get(night_id)
97    if night is None:
98        raise HTTPException(status_code=404, detail="not found")
99    return night
100 
101 
102@app.post("/rsvps")
103def create_rsvp(req: dict, authorization: Optional[str] = Header(None)):
104    global rsvp_counter
105    user_id = get_user_from_token(authorization)
106    night_id = req.get("game_night_id")
107    if night_id not in game_nights:
108        raise HTTPException(status_code=404, detail="game night not found")
109    rsvp_counter += 1
110    rsvp = {
111        "id": rsvp_counter,
112        "user_id": user_id,
113        "game_night_id": night_id,
114        "status": req.get("status", "yes"),
115        "guest_count": req.get("guest_count", 0),
116    }
117    for k, v in req.items():
118        if k not in rsvp:
119            rsvp[k] = v
120    rsvps[rsvp_counter] = rsvp
121    return rsvp
122 
123 
124@app.get("/rsvps/{rsvp_id}")
125def get_rsvp(rsvp_id: int):
126    rsvp = rsvps.get(rsvp_id)
127    if rsvp is None:
128        raise HTTPException(status_code=404, detail="not found")
129    return rsvp
130 
131 
132@app.get("/users/{user_id}")
133def get_user(user_id: int):
134    user = users.get(user_id)
135    if user is None:
136        raise HTTPException(status_code=404, detail="not found")
137    return user

1fastapi
2uvicorn
3pydantic