Browse the data

1from fastapi import FastAPI, Header, HTTPException
2from pydantic import BaseModel
3from typing import Optional
4import secrets
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10events = {}
11registrations = {}
12 
13user_counter = 0
14event_counter = 0
15registration_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_current_user(authorization: Optional[str] = Header(None)):
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 users[user_id]
36 
37 
38@app.post("/signup")
39def signup(req: dict):
40    global user_counter
41    user_counter += 1
42    user = {
43        "id": user_counter,
44        "username": req.get("username"),
45        "password": req.get("password"),
46        "role": req.get("role", "artist"),
47        "is_admin": req.get("is_admin", False),
48    }
49    for k, v in req.items():
50        if k not in user:
51            user[k] = v
52    users[user_counter] = user
53    return user
54 
55 
56@app.post("/login")
57def login(req: LoginRequest):
58    for user in users.values():
59        if user["username"] == req.username and user["password"] == req.password:
60            token = secrets.token_hex(16)
61            tokens[token] = user["id"]
62            return {"token": token}
63    raise HTTPException(status_code=401, detail="Invalid credentials")
64 
65 
66@app.post("/events")
67def create_event(req: dict, authorization: Optional[str] = Header(None)):
68    global event_counter
69    user = get_current_user(authorization)
70    event_counter += 1
71    event = {
72        "id": event_counter,
73        "user_id": user["id"],
74    }
75    for k, v in req.items():
76        event[k] = v
77    event["id"] = event_counter
78    event["user_id"] = user["id"]
79    events[event_counter] = event
80    return event
81 
82 
83@app.get("/events/{event_id}")
84def get_event(event_id: int):
85    event = events.get(event_id)
86    if event is None:
87        raise HTTPException(status_code=404, detail="Event not found")
88    return event
89 
90 
91@app.post("/registrations")
92def create_registration(req: dict, authorization: Optional[str] = Header(None)):
93    global registration_counter
94    user = get_current_user(authorization)
95    registration_counter += 1
96    registration = {
97        "id": registration_counter,
98        "user_id": user["id"],
99    }
100    for k, v in req.items():
101        registration[k] = v
102    registration["id"] = registration_counter
103    registration["user_id"] = user["id"]
104    registrations[registration_counter] = registration
105    return registration
106 
107 
108@app.get("/registrations/{registration_id}")
109def get_registration(registration_id: int):
110    registration = registrations.get(registration_id)
111    if registration is None:
112        raise HTTPException(status_code=404, detail="Registration not found")
113    return registration
114 
115 
116@app.get("/users/{user_id}")
117def get_user(user_id: int):
118    user = users.get(user_id)
119    if user is None:
120        raise HTTPException(status_code=404, detail="User not found")
121    return user

1fastapi
2uvicorn
3pydantic