Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from datetime import datetime
3import secrets
4import uuid
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10appointments = {}
11staff = {}
12appointment_id_counter = 1
13staff_id_counter = 1
14 
15def get_current_user(authorization: str = Header(None)):
16    if not authorization or not authorization.startswith("Bearer "):
17        raise HTTPException(status_code=401, detail="Invalid auth")
18    token = authorization.split(" ")[1]
19    if token not in tokens:
20        raise HTTPException(status_code=401, detail="Invalid token")
21    return tokens[token]
22 
23@app.post("/signup")
24def signup(username: str, password: str):
25    if username in users:
26        raise HTTPException(status_code=400, detail="User exists")
27    users[username] = {"password": password}
28    return {"ok": True}
29 
30@app.post("/login")
31def login(username: str, password: str):
32    if username not in users or users[username]["password"] != password:
33        raise HTTPException(status_code=401, detail="Bad credentials")
34    token = secrets.token_hex(16)
35    tokens[token] = username
36    return {"token": token}
37 
38@app.get("/appointments/{appointment_id}")
39def get_appointment(appointment_id: int, authorization: str = Header(None)):
40    get_current_user(authorization)
41    if appointment_id not in appointments:
42        raise HTTPException(status_code=404, detail="Not found")
43    return appointments[appointment_id]
44 
45@app.post("/appointments")
46def create_appointment(date: str, time: str, duration: int, client_name: str, staff_role: str = None, tier: int = 0, authorization: str = Header(None)):
47    get_current_user(authorization)
48    global appointment_id_counter
49    appt = {
50        "id": appointment_id_counter,
51        "date": date,
52        "time": time,
53        "duration": duration,
54        "client_name": client_name,
55        "staff_role": staff_role,
56        "tier": tier
57    }
58    appointments[appointment_id_counter] = appt
59    appointment_id_counter += 1
60    return appt
61 
62@app.patch("/appointments/{appointment_id}")
63def update_appointment(appointment_id: int, date: str = None, time: str = None, duration: int = None, client_name: str = None, authorization: str = Header(None)):
64    get_current_user(authorization)
65    if appointment_id not in appointments:
66        raise HTTPException(status_code=404, detail="Not found")
67    appt = appointments[appointment_id]
68    if date is not None:
69        appt["date"] = date
70    if time is not None:
71        appt["time"] = time
72    if duration is not None:
73        appt["duration"] = duration
74    if client_name is not None:
75        appt["client_name"] = client_name
76    return appt
77 
78@app.post("/staff")
79def create_staff(name: str, role: str, tier: int = 0, authorization: str = Header(None)):
80    get_current_user(authorization)
81    global staff_id_counter
82    s = {"id": staff_id_counter, "name": name, "role": role, "tier": tier}
83    staff[staff_id_counter] = s
84    staff_id_counter += 1
85    return s
86 
87@app.get("/staff/{staff_id}")
88def get_staff(staff_id: int, authorization: str = Header(None)):
89    get_current_user(authorization)
90    if staff_id not in staff:
91        raise HTTPException(status_code=404, detail="Not found")
92    return staff[staff_id]

1fastapi
2uvicorn