Browse the data

1import os
2from fastapi import FastAPI, HTTPException, Header
3from fastapi.responses import Response
4from pydantic import BaseModel
5import uvicorn
6 
7app = FastAPI()
8 
9tokens = {}
10users = {}
11events = {}
12attendees = {}
13next_user_id = 1
14next_event_id = 1
15next_attendee_id = 1
16next_token_id = 1
17 
18class SignupRequest(BaseModel):
19    username: str
20    password: str
21 
22class LoginRequest(BaseModel):
23    username: str
24    password: str
25 
26class EventCreate(BaseModel):
27    name: str
28 
29class AttendeeCreate(BaseModel):
30    name: str
31    event_id: int
32 
33def get_current_user(authorization: str = Header(None)):
34    if not authorization or not authorization.startswith("Bearer "):
35        raise HTTPException(status_code=401, detail="Invalid token")
36    token = authorization[len("Bearer "):]
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 {"user_id": user_id}
48 
49@app.post("/login")
50def login(req: LoginRequest):
51    global next_token_id
52    for u in users.values():
53        if u["username"] == req.username and u["password"] == req.password:
54            token = f"token_{next_token_id}"
55            next_token_id += 1
56            tokens[token] = u["id"]
57            return {"token": token}
58    raise HTTPException(status_code=401, detail="Invalid credentials")
59 
60@app.post("/events")
61def create_event(event: EventCreate, authorization: str = Header(None)):
62    get_current_user(authorization)
63    global next_event_id
64    event_id = next_event_id
65    next_event_id += 1
66    events[event_id] = {"id": event_id, "name": event.name}
67    return {"event_id": event_id}
68 
69@app.get("/events/{event_id}")
70def get_event(event_id: int, authorization: str = Header(None)):
71    get_current_user(authorization)
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.post("/attendees")
77def create_attendee(attendee: AttendeeCreate, authorization: str = Header(None)):
78    get_current_user(authorization)
79    if attendee.event_id not in events:
80        raise HTTPException(status_code=404, detail="Event not found")
81    global next_attendee_id
82    attendee_id = next_attendee_id
83    next_attendee_id += 1
84    attendees[attendee_id] = {"id": attendee_id, "name": attendee.name, "event_id": attendee.event_id}
85    return {"attendee_id": attendee_id}
86 
87@app.get("/attendees/{attendee_id}")
88def get_attendee(attendee_id: int, authorization: str = Header(None)):
89    get_current_user(authorization)
90    if attendee_id not in attendees:
91        raise HTTPException(status_code=404, detail="Attendee not found")
92    return attendees[attendee_id]
93 
94@app.get("/events/{event_id}/qr/{attendee_id}")
95def get_qr(event_id: int, attendee_id: int, authorization: str = Header(None)):
96    get_current_user(authorization)
97    if event_id not in events:
98        raise HTTPException(status_code=404, detail="Event not found")
99    if attendee_id not in attendees or attendees[attendee_id]["event_id"] != event_id:
100        raise HTTPException(status_code=404, detail="Attendee not found for this event")
101    qr_dir = f"qr_codes/{event_id}"
102    qr_path = os.path.join(qr_dir, f"{attendee_id}.png")
103    if not os.path.exists(qr_path):
104        raise HTTPException(status_code=404, detail="QR code not found")
105    with open(qr_path, "rb") as f:
106        image_data = f.read()
107    return Response(content=image_data, media_type="image/png")

1fastapi
2uvicorn