Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, List
4import random
5import string
6from datetime import datetime, timedelta
7 
8app = FastAPI()
9 
10users = {}
11tokens = {}
12classes = {}
13waitlists = {}
14bookings = {}
15class_id_counter = 1
16user_id_counter = 1
17 
18def generate_token():
19    return ''.join(random.choices(string.ascii_letters + string.digits, k=32))
20 
21def get_current_user(authorization: str = Header(None)):
22    if not authorization:
23        raise HTTPException(status_code=401, detail="No auth header")
24    token = authorization.replace("Bearer ", "")
25    if token not in tokens:
26        raise HTTPException(status_code=401, detail="Invalid token")
27    return tokens[token]
28 
29class UserCreate(BaseModel):
30    username: str
31    password: str
32 
33class UserLogin(BaseModel):
34    username: str
35    password: str
36 
37class ClassCreate(BaseModel):
38    name: str
39    time: str
40    instructor: str
41    capacity: int
42    difficulty: str
43    recurring: bool = False
44    recurring_end_date: Optional[str] = None
45 
46class ClassUpdate(BaseModel):
47    time: Optional[str] = None
48    instructor: Optional[str] = None
49    capacity: Optional[int] = None
50    difficulty: Optional[str] = None
51 
52class BookingCreate(BaseModel):
53    class_id: int
54 
55@app.post("/signup")
56def signup(user: UserCreate):
57    global user_id_counter
58    if any(u["username"] == user.username for u in users.values()):
59        raise HTTPException(status_code=400, detail="Username exists")
60    uid = user_id_counter
61    users[uid] = {"id": uid, "username": user.username, "password": user.password}
62    user_id_counter += 1
63    token = generate_token()
64    tokens[token] = uid
65    return {"user_id": uid, "token": token}
66 
67@app.post("/login")
68def login(user: UserLogin):
69    for uid, u in users.items():
70        if u["username"] == user.username and u["password"] == user.password:
71            token = generate_token()
72            tokens[token] = uid
73            return {"user_id": uid, "token": token}
74    raise HTTPException(status_code=401, detail="Invalid credentials")
75 
76@app.get("/users/{user_id}")
77def get_user(user_id: int, authorization: str = Header(None)):
78    get_current_user(authorization)
79    if user_id not in users:
80        raise HTTPException(status_code=404, detail="User not found")
81    return users[user_id]
82 
83@app.post("/classes")
84def create_class(cls: ClassCreate, authorization: str = Header(None)):
85    global class_id_counter
86    get_current_user(authorization)
87    cid = class_id_counter
88    classes[cid] = {
89        "id": cid,
90        "name": cls.name,
91        "time": cls.time,
92        "instructor": cls.instructor,
93        "capacity": cls.capacity,
94        "difficulty": cls.difficulty,
95        "recurring": cls.recurring,
96        "recurring_end_date": cls.recurring_end_date,
97        "booked_count": 0
98    }
99    waitlists[cid] = []
100    class_id_counter += 1
101    return classes[cid]
102 
103@app.get("/classes/{class_id}")
104def get_class(class_id: int, authorization: str = Header(None)):
105    get_current_user(authorization)
106    if class_id not in classes:
107        raise HTTPException(status_code=404, detail="Class not found")
108    return classes[class_id]
109 
110@app.put("/classes/{class_id}")
111def update_class(class_id: int, update: ClassUpdate, authorization: str = Header(None)):
112    get_current_user(authorization)
113    if class_id not in classes:
114        raise HTTPException(status_code=404, detail="Class not found")
115    cls = classes[class_id]
116    if update.time is not None:
117        cls["time"] = update.time
118    if update.instructor is not None:
119        cls["instructor"] = update.instructor
120    if update.capacity is not None:
121        cls["capacity"] = update.capacity
122    if update.difficulty is not None:
123        cls["difficulty"] = update.difficulty
124    return cls
125 
126@app.post("/bookings")
127def create_booking(booking: BookingCreate, authorization: str = Header(None)):
128    user_id = get_current_user(authorization)
129    cid = booking.class_id
130    if cid not in classes:
131        raise HTTPException(status_code=404, detail="Class not found")
132    cls = classes[cid]
133    if cls["booked_count"] < cls["capacity"]:
134        bid = len(bookings) + 1
135        bookings[bid] = {"id": bid, "user_id": user_id, "class_id": cid}
136        cls["booked_count"] += 1
137        return bookings[bid]
138    else:
139        if user_id in waitlists[cid]:
140            raise HTTPException(status_code=400, detail="Already on waitlist")
141        waitlists[cid].append(user_id)
142        return {"message": "Added to waitlist", "waitlist_position": len(waitlists[cid])}
143 
144@app.get("/bookings/{booking_id}")
145def get_booking(booking_id: int, authorization: str = Header(None)):
146    get_current_user(authorization)
147    if booking_id not in bookings:
148        raise HTTPException(status_code=404, detail="Booking not found")
149    return bookings[booking_id]
150 
151@app.get("/classes")
152def list_classes(authorization: str = Header(None)):
153    get_current_user(authorization)
154    return list(classes.values())

1fastapi
2uvicorn