Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import secrets
4 
5app = FastAPI()
6 
7users = {}
8tokens = {}
9slots = {}
10books = {}
11slot_id_counter = 1
12book_id_counter = 1
13user_id_counter = 1
14 
15def get_current_user(authorization: Optional[str] = Header(None)):
16    if not authorization:
17        raise HTTPException(status_code=401, detail="No auth header")
18    token = authorization.replace("Bearer ", "")
19    for uid, t in tokens.items():
20        if t == token:
21            return uid
22    raise HTTPException(status_code=401, detail="Invalid token")
23 
24@app.post("/signup")
25def signup(username: str, password: str):
26    global user_id_counter
27    if any(u["username"] == username for u in users.values()):
28        raise HTTPException(status_code=400, detail="User exists")
29    uid = user_id_counter
30    user_id_counter += 1
31    users[uid] = {"id": uid, "username": username, "password": password}
32    return {"id": uid, "username": username}
33 
34@app.post("/login")
35def login(username: str, password: str):
36    for uid, u in users.items():
37        if u["username"] == username and u["password"] == password:
38            token = secrets.token_hex(16)
39            tokens[uid] = token
40            return {"token": token}
41    raise HTTPException(status_code=401, detail="Invalid credentials")
42 
43@app.post("/slots")
44def create_slot(expert_name: str, available_time: str, authorization: Optional[str] = Header(None)):
45    get_current_user(authorization)
46    global slot_id_counter
47    sid = slot_id_counter
48    slot_id_counter += 1
49    slots[sid] = {"id": sid, "expert_name": expert_name, "available_time": available_time, "booked": False}
50    return {"id": sid, "expert_name": expert_name, "available_time": available_time}
51 
52@app.post("/book")
53def book_slot(slot_id: int, client_name: str, authorization: Optional[str] = Header(None)):
54    get_current_user(authorization)
55    if slot_id not in slots:
56        raise HTTPException(status_code=404, detail="Slot not found")
57    if slots[slot_id]["booked"]:
58        raise HTTPException(status_code=400, detail="Slot already booked")
59    slots[slot_id]["booked"] = True
60    global book_id_counter
61    bid = book_id_counter
62    book_id_counter += 1
63    books[bid] = {"id": bid, "slot_id": slot_id, "client_name": client_name}
64    return {"id": bid, "slot_id": slot_id, "client_name": client_name}
65 
66@app.get("/slots/available")
67def get_available_slots(authorization: Optional[str] = Header(None)):
68    get_current_user(authorization)
69    return [s for s in slots.values() if not s["booked"]]
70 
71@app.get("/{resource}/{resource_id}")
72def get_resource(resource: str, resource_id: int, authorization: Optional[str] = Header(None)):
73    get_current_user(authorization)
74    if resource == "slots":
75        if resource_id not in slots:
76            raise HTTPException(status_code=404, detail="Slot not found")
77        return slots[resource_id]
78    elif resource == "books":
79        if resource_id not in books:
80            raise HTTPException(status_code=404, detail="Booking not found")
81        return books[resource_id]
82    elif resource == "users":
83        if resource_id not in users:
84            raise HTTPException(status_code=404, detail="User not found")
85        return {"id": users[resource_id]["id"], "username": users[resource_id]["username"]}
86    else:
87        raise HTTPException(status_code=404, detail="Resource not found")

1fastapi
2uvicorn