Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import secrets
4import time
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10classes = {}
11bookings = {}
12next_user_id = 1
13next_class_id = 1
14next_booking_id = 1
15 
16def require_admin(authorization: Optional[str] = Header(None)):
17    if not authorization:
18        raise HTTPException(401, "No auth header")
19    token = authorization.replace("Bearer ", "")
20    user_id = tokens.get(token)
21    if not user_id:
22        raise HTTPException(401, "Invalid token")
23    user = users.get(user_id)
24    if not user or user.get("role") != "admin":
25        raise HTTPException(403, "Admins only")
26    return user_id
27 
28def require_auth(authorization: Optional[str] = Header(None)):
29    if not authorization:
30        raise HTTPException(401, "No auth header")
31    token = authorization.replace("Bearer ", "")
32    user_id = tokens.get(token)
33    if not user_id:
34        raise HTTPException(401, "Invalid token")
35    return user_id
36 
37@app.post("/signup")
38def signup(username: str, password: str, role: str = "member"):
39    global next_user_id
40    uid = next_user_id
41    next_user_id += 1
42    users[uid] = {"id": uid, "username": username, "password": password, "role": role}
43    return {"id": uid, "username": username, "role": role}
44 
45@app.post("/login")
46def login(username: str, password: str):
47    for uid, u in users.items():
48        if u["username"] == username and u["password"] == password:
49            token = secrets.token_hex(16)
50            tokens[token] = uid
51            return {"token": token, "user_id": uid}
52    raise HTTPException(401, "Bad credentials")
53 
54@app.post("/classes")
55def create_class(
56    name: str,
57    time: str,
58    capacity: int,
59    instructor_id: Optional[int] = None,
60    difficulty: Optional[str] = None,
61    authorization: Optional[str] = Header(None)
62):
63    require_admin(authorization)
64    global next_class_id
65    cid = next_class_id
66    next_class_id += 1
67    classes[cid] = {
68        "id": cid,
69        "name": name,
70        "time": time,
71        "capacity": capacity,
72        "instructor_id": instructor_id,
73        "difficulty": difficulty
74    }
75    return classes[cid]
76 
77@app.get("/classes/{class_id}")
78def get_class(class_id: int):
79    c = classes.get(class_id)
80    if not c:
81        raise HTTPException(404, "Class not found")
82    return c
83 
84@app.patch("/classes/{class_id}")
85def update_class(
86    class_id: int,
87    name: Optional[str] = None,
88    time: Optional[str] = None,
89    capacity: Optional[int] = None,
90    instructor_id: Optional[int] = None,
91    difficulty: Optional[str] = None,
92    authorization: Optional[str] = Header(None)
93):
94    require_admin(authorization)
95    c = classes.get(class_id)
96    if not c:
97        raise HTTPException(404, "Class not found")
98    if name is not None:
99        c["name"] = name
100    if time is not None:
101        c["time"] = time
102    if capacity is not None:
103        c["capacity"] = capacity
104    if instructor_id is not None:
105        c["instructor_id"] = instructor_id
106    if difficulty is not None:
107        c["difficulty"] = difficulty
108    return c
109 
110@app.post("/bookings")
111def create_booking(class_id: int, authorization: Optional[str] = Header(None)):
112    user_id = require_auth(authorization)
113    c = classes.get(class_id)
114    if not c:
115        raise HTTPException(404, "Class not found")
116    count = sum(1 for b in bookings.values() if b["class_id"] == class_id)
117    if count >= c["capacity"]:
118        raise HTTPException(400, "Class full")
119    global next_booking_id
120    bid = next_booking_id
121    next_booking_id += 1
122    bookings[bid] = {"id": bid, "user_id": user_id, "class_id": class_id}
123    return bookings[bid]
124 
125@app.get("/bookings/{booking_id}")
126def get_booking(booking_id: int):
127    b = bookings.get(booking_id)
128    if not b:
129        raise HTTPException(404, "Booking not found")
130    return b
131 
132@app.post("/users")
133def create_user(username: str, password: str, role: str = "member"):
134    global next_user_id
135    uid = next_user_id
136    next_user_id += 1
137    users[uid] = {"id": uid, "username": username, "password": password, "role": role}
138    return users[uid]
139 
140@app.get("/users/{user_id}")
141def get_user(user_id: int):
142    u = users.get(user_id)
143    if not u:
144        raise HTTPException(404, "User not found")
145    return {"id": u["id"], "username": u["username"], "role": u["role"]}

1fastapi
2uvicorn