Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import secrets
5import time
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11slots = {}
12slot_id_counter = 1
13 
14def get_current_user(authorization: str = Header(None)):
15    if not authorization:
16        raise HTTPException(status_code=401, detail="Missing auth header")
17    token = authorization.replace("Bearer ", "")
18    user_id = tokens.get(token)
19    if user_id is None:
20        raise HTTPException(status_code=401, detail="Invalid token")
21    return user_id
22 
23class SignupRequest(BaseModel):
24    username: str
25    password: str
26 
27class LoginRequest(BaseModel):
28    username: str
29    password: str
30 
31class SlotCreate(BaseModel):
32    time: str
33    duration: int
34    capacity: int
35    is_priority: bool = False
36    staff_role: str = "general"
37 
38class SlotUpdate(BaseModel):
39    time: str = None
40    duration: int = None
41    capacity: int = None
42    is_priority: bool = None
43    staff_role: str = None
44 
45class Slot(BaseModel):
46    id: int
47    time: str
48    duration: int
49    capacity: int
50    is_priority: bool
51    staff_role: str
52 
53@app.post("/signup")
54def signup(req: SignupRequest):
55    if req.username in users:
56        raise HTTPException(status_code=400, detail="User already exists")
57    users[req.username] = {"password": req.password}
58    return {"message": "User created"}
59 
60@app.post("/login")
61def login(req: LoginRequest):
62    user = users.get(req.username)
63    if not user or user["password"] != req.password:
64        raise HTTPException(status_code=401, detail="Invalid credentials")
65    token = secrets.token_hex(16)
66    tokens[token] = req.username
67    return {"token": token}
68 
69@app.post("/slots")
70def create_slot(slot: SlotCreate, authorization: str = Header(None)):
71    get_current_user(authorization)
72    global slot_id_counter
73    new_slot = {
74        "id": slot_id_counter,
75        "time": slot.time,
76        "duration": slot.duration,
77        "capacity": slot.capacity,
78        "is_priority": slot.is_priority,
79        "staff_role": slot.staff_role
80    }
81    slots[slot_id_counter] = new_slot
82    slot_id_counter += 1
83    return new_slot
84 
85@app.get("/slots/{slot_id}")
86def get_slot(slot_id: int, authorization: str = Header(None)):
87    get_current_user(authorization)
88    slot = slots.get(slot_id)
89    if not slot:
90        raise HTTPException(status_code=404, detail="Slot not found")
91    return slot
92 
93@app.patch("/slots/{slot_id}")
94def update_slot(slot_id: int, update: SlotUpdate, authorization: str = Header(None)):
95    get_current_user(authorization)
96    slot = slots.get(slot_id)
97    if not slot:
98        raise HTTPException(status_code=404, detail="Slot not found")
99    if update.time is not None:
100        slot["time"] = update.time
101    if update.duration is not None:
102        slot["duration"] = update.duration
103    if update.capacity is not None:
104        slot["capacity"] = update.capacity
105    if update.is_priority is not None:
106        slot["is_priority"] = update.is_priority
107    if update.staff_role is not None:
108        slot["staff_role"] = update.staff_role
109    return slot

1fastapi
2uvicorn