Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, Dict
4import time
5import secrets
6 
7app = FastAPI()
8 
9users = {}
10registrants = {}
11checkins = {}
12capacity_limit = 100
13next_user_id = 1
14next_registrant_id = 1
15next_checkin_id = 1
16tokens = {}
17 
18class SignupRequest(BaseModel):
19    username: str
20    password: str
21 
22class LoginRequest(BaseModel):
23    username: str
24    password: str
25 
26class RegistrantCreate(BaseModel):
27    name: str
28    ticket_type: str
29 
30class CheckInRequest(BaseModel):
31    registrant_id: int
32 
33def get_current_user(authorization: str = Header(...)):
34    if not authorization.startswith("Bearer "):
35        raise HTTPException(status_code=401, detail="Invalid auth header")
36    token = authorization.split(" ")[1]
37    if token not in tokens:
38        raise HTTPException(status_code=401, detail="Invalid token")
39    return tokens[token]
40 
41@app.post("/signup")
42def signup(req: SignupRequest):
43    global next_user_id
44    user_id = next_user_id
45    next_user_id += 1
46    users[user_id] = {"id": user_id, "username": req.username, "password": req.password}
47    return {"id": user_id, "username": req.username}
48 
49@app.post("/login")
50def login(req: LoginRequest):
51    for u in users.values():
52        if u["username"] == req.username and u["password"] == req.password:
53            token = secrets.token_hex(32)
54            tokens[token] = u["id"]
55            return {"token": token}
56    raise HTTPException(status_code=401, detail="Invalid credentials")
57 
58@app.get("/users/{user_id}")
59def get_user(user_id: int, authorization: str = Header(...)):
60    get_current_user(authorization)
61    if user_id not in users:
62        raise HTTPException(status_code=404, detail="User not found")
63    return users[user_id]
64 
65@app.post("/registrants")
66def create_registrant(req: RegistrantCreate, authorization: str = Header(...)):
67    get_current_user(authorization)
68    global next_registrant_id
69    reg_id = next_registrant_id
70    next_registrant_id += 1
71    registrants[reg_id] = {"id": reg_id, "name": req.name, "ticket_type": req.ticket_type, "checked_in": False}
72    return registrants[reg_id]
73 
74@app.get("/registrants/{registrant_id}")
75def get_registrant(registrant_id: int, authorization: str = Header(...)):
76    get_current_user(authorization)
77    if registrant_id not in registrants:
78        raise HTTPException(status_code=404, detail="Registrant not found")
79    return registrants[registrant_id]
80 
81@app.get("/registrants")
82def list_registrants(authorization: str = Header(...)):
83    get_current_user(authorization)
84    return list(registrants.values())
85 
86@app.post("/check-in")
87def check_in(req: CheckInRequest, authorization: str = Header(...)):
88    get_current_user(authorization)
89    if req.registrant_id not in registrants:
90        raise HTTPException(status_code=404, detail="Registrant not found")
91    if registrants[req.registrant_id]["checked_in"]:
92        raise HTTPException(status_code=400, detail="Already checked in")
93    registrants[req.registrant_id]["checked_in"] = True
94    global next_checkin_id
95    checkin_id = next_checkin_id
96    next_checkin_id += 1
97    checkins[checkin_id] = {"registrant_id": req.registrant_id, "time": time.time()}
98    return {"status": "ok", "checkin_id": checkin_id}
99 
100@app.get("/capacity")
101def get_capacity(authorization: str = Header(...)):
102    get_current_user(authorization)
103    headcount = sum(1 for r in registrants.values() if r["checked_in"])
104    return {"headcount": headcount, "limit": capacity_limit}

1fastapi
2uvicorn