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 = {}
10events = {}
11signups = {}
12 
13user_counter = 0
14event_counter = 0
15signup_counter = 0
16 
17 
18class SignupRequest(BaseModel):
19    username: str
20    password: str
21 
22    class Config:
23        extra = "allow"
24 
25 
26class LoginRequest(BaseModel):
27    username: str
28    password: str
29 
30 
31class EventCreate(BaseModel):
32    title: str
33    description: Optional[str] = None
34    location: Optional[str] = None
35    date: Optional[str] = None
36 
37    class Config:
38        extra = "allow"
39 
40 
41class VolunteerSignup(BaseModel):
42    event_id: int
43    note: Optional[str] = None
44 
45    class Config:
46        extra = "allow"
47 
48 
49def get_user_from_token(authorization: Optional[str]):
50    if not authorization:
51        raise HTTPException(status_code=401, detail="Missing Authorization header")
52    token = authorization.replace("Bearer ", "").strip()
53    user_id = tokens.get(token)
54    if user_id is None:
55        raise HTTPException(status_code=401, detail="Invalid token")
56    return users[user_id]
57 
58 
59@app.post("/signup")
60def signup(req: SignupRequest):
61    global user_counter
62    data = req.dict()
63    for u in users.values():
64        if u["username"] == data["username"]:
65            raise HTTPException(status_code=400, detail="Username taken")
66    user_counter += 1
67    record = {"id": user_counter}
68    record.update(data)
69    users[user_counter] = record
70    return record
71 
72 
73@app.post("/login")
74def login(req: LoginRequest):
75    for u in users.values():
76        if u["username"] == req.username and u["password"] == req.password:
77            token = secrets.token_hex(16)
78            tokens[token] = u["id"]
79            return {"token": token, "user_id": u["id"]}
80    raise HTTPException(status_code=401, detail="Bad credentials")
81 
82 
83@app.post("/events")
84def create_event(req: EventCreate, authorization: Optional[str] = Header(None)):
85    global event_counter
86    user = get_user_from_token(authorization)
87    data = req.dict()
88    event_counter += 1
89    record = {"id": event_counter, "user_id": user["id"]}
90    record.update(data)
91    events[event_counter] = record
92    return record
93 
94 
95@app.get("/events/{event_id}")
96def get_event(event_id: int):
97    event = events.get(event_id)
98    if event is None:
99        raise HTTPException(status_code=404, detail="Event not found")
100    return event
101 
102 
103@app.post("/signups")
104def create_signup(req: VolunteerSignup, authorization: Optional[str] = Header(None)):
105    global signup_counter
106    user = get_user_from_token(authorization)
107    data = req.dict()
108    if data["event_id"] not in events:
109        raise HTTPException(status_code=404, detail="Event not found")
110    signup_counter += 1
111    record = {"id": signup_counter, "user_id": user["id"]}
112    record.update(data)
113    signups[signup_counter] = record
114    return record
115 
116 
117@app.get("/signups/{signup_id}")
118def get_signup(signup_id: int):
119    record = signups.get(signup_id)
120    if record is None:
121        raise HTTPException(status_code=404, detail="Signup not found")
122    return record
123 
124 
125@app.get("/users/{user_id}")
126def get_user(user_id: int):
127    user = users.get(user_id)
128    if user is None:
129        raise HTTPException(status_code=404, detail="User not found")
130    return user

1fastapi
2uvicorn
3pydantic