Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import hashlib
5import secrets
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11registrations = {}
12reg_id_counter = 1
13 
14class SignupRequest(BaseModel):
15    username: str
16    password: str
17 
18class LoginRequest(BaseModel):
19    username: str
20    password: str
21 
22class RegistrationCreate(BaseModel):
23    attendee_name: str
24    email: str
25    session_selected: str
26    ticket_type: str
27    speaker_role: bool = False
28 
29class RegistrationUpdate(BaseModel):
30    attendee_name: Optional[str] = None
31    email: Optional[str] = None
32    session_selected: Optional[str] = None
33    ticket_type: Optional[str] = None
34    speaker_role: Optional[bool] = None
35 
36def get_current_user(authorization: str = Header(None)):
37    if not authorization:
38        raise HTTPException(status_code=401, detail="Missing Authorization header")
39    token = authorization.replace("Bearer ", "")
40    if token not in tokens:
41        raise HTTPException(status_code=401, detail="Invalid token")
42    return tokens[token]
43 
44@app.post("/signup")
45def signup(req: SignupRequest):
46    if req.username in users:
47        raise HTTPException(status_code=400, detail="Username already exists")
48    password_hash = hashlib.sha256(req.password.encode()).hexdigest()
49    users[req.username] = password_hash
50    return {"message": "User created"}
51 
52@app.post("/login")
53def login(req: LoginRequest):
54    if req.username not in users:
55        raise HTTPException(status_code=401, detail="Invalid credentials")
56    password_hash = hashlib.sha256(req.password.encode()).hexdigest()
57    if users[req.username] != password_hash:
58        raise HTTPException(status_code=401, detail="Invalid credentials")
59    token = secrets.token_hex(32)
60    tokens[token] = req.username
61    return {"token": token}
62 
63@app.post("/registrations")
64def create_registration(req: RegistrationCreate, authorization: str = Header(None)):
65    get_current_user(authorization)
66    global reg_id_counter
67    reg_id = reg_id_counter
68    reg_id_counter += 1
69    registrations[reg_id] = {
70        "id": reg_id,
71        "attendee_name": req.attendee_name,
72        "email": req.email,
73        "session_selected": req.session_selected,
74        "ticket_type": req.ticket_type,
75        "speaker_role": req.speaker_role
76    }
77    return registrations[reg_id]
78 
79@app.get("/registrations/{reg_id}")
80def get_registration(reg_id: int, authorization: str = Header(None)):
81    get_current_user(authorization)
82    if reg_id not in registrations:
83        raise HTTPException(status_code=404, detail="Registration not found")
84    return registrations[reg_id]
85 
86@app.patch("/registrations/{reg_id}")
87def update_registration(reg_id: int, req: RegistrationUpdate, authorization: str = Header(None)):
88    get_current_user(authorization)
89    if reg_id not in registrations:
90        raise HTTPException(status_code=404, detail="Registration not found")
91    reg = registrations[reg_id]
92    if req.attendee_name is not None:
93        reg["attendee_name"] = req.attendee_name
94    if req.email is not None:
95        reg["email"] = req.email
96    if req.session_selected is not None:
97        reg["session_selected"] = req.session_selected
98    if req.ticket_type is not None:
99        reg["ticket_type"] = req.ticket_type
100    if req.speaker_role is not None:
101        reg["speaker_role"] = req.speaker_role
102    return reg

1fastapi
2uvicorn