Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, Dict
4import secrets
5import datetime
6 
7app = FastAPI()
8 
9# In-memory storage
10users = {}
11tokens = {}
12stylists = {}
13services = {}
14slots = {}
15appointments = {}
16next_ids = {"users": 1, "stylists": 1, "services": 1, "slots": 1, "appointments": 1}
17 
18# Seed data
19stylists[next_ids["stylists"]] = {"id": 1, "name": "Alice", "active": True}
20next_ids["stylists"] = 2
21stylists[next_ids["stylists"]] = {"id": 2, "name": "Bob", "active": True}
22next_ids["stylists"] = 3
23 
24services[next_ids["services"]] = {"id": 1, "name": "Haircut", "duration_minutes": 30}
25next_ids["services"] = 2
26services[next_ids["services"]] = {"id": 2, "name": "Color", "duration_minutes": 90}
27next_ids["services"] = 3
28services[next_ids["services"]] = {"id": 3, "name": "Blow Dry", "duration_minutes": 45}
29next_ids["services"] = 4
30 
31# Generate some sample slots for today and tomorrow
32base_date = datetime.date.today()
33for day_offset in range(2):
34    current_date = base_date + datetime.timedelta(days=day_offset)
35    for stylist_id, stylist in stylists.items():
36        for hour in range(9, 17):
37            start_time = datetime.datetime.combine(current_date, datetime.time(hour, 0))
38            end_time = start_time + datetime.timedelta(hours=1)
39            for service_id, service in services.items():
40                if service["duration_minutes"] <= 60:
41                    slot_id = next_ids["slots"]
42                    slots[slot_id] = {
43                        "id": slot_id,
44                        "stylist_id": stylist_id,
45                        "stylist_name": stylist["name"],
46                        "service_id": service_id,
47                        "service_name": service["name"],
48                        "service_duration_minutes": service["duration_minutes"],
49                        "start_time": start_time.isoformat(),
50                        "end_time": end_time.isoformat(),
51                        "available": True
52                    }
53                    next_ids["slots"] += 1
54 
55class SignupRequest(BaseModel):
56    username: str
57    password: str
58 
59class LoginRequest(BaseModel):
60    username: str
61    password: str
62 
63class CreateSlotRequest(BaseModel):
64    stylist_id: int
65    service_id: int
66    start_time: str
67 
68class BookRequest(BaseModel):
69    client_name: str
70    client_phone: str
71 
72def get_current_user(authorization: Optional[str] = Header(None)):
73    if not authorization:
74        raise HTTPException(status_code=401, detail="Missing Authorization header")
75    token = authorization.replace("Bearer ", "")
76    if token not in tokens:
77        raise HTTPException(status_code=401, detail="Invalid token")
78    return tokens[token]
79 
80@app.post("/signup")
81def signup(req: SignupRequest):
82    if req.username in users:
83        raise HTTPException(status_code=400, detail="Username already exists")
84    user_id = next_ids["users"]
85    users[req.username] = {"id": user_id, "username": req.username, "password": req.password}
86    next_ids["users"] += 1
87    return {"message": "User created", "user_id": user_id}
88 
89@app.post("/login")
90def login(req: LoginRequest):
91    if req.username not in users or users[req.username]["password"] != req.password:
92        raise HTTPException(status_code=401, detail="Invalid credentials")
93    token = secrets.token_hex(16)
94    tokens[token] = req.username
95    return {"token": token}
96 
97@app.get("/slots")
98def get_slots():
99    available = [s for s in slots.values() if s["available"]]
100    return available
101 
102@app.post("/book/{slot_id}")
103def book_slot(slot_id: int, req: BookRequest, authorization: Optional[str] = Header(None)):
104    user = get_current_user(authorization)
105    if slot_id not in slots:
106        raise HTTPException(status_code=404, detail="Slot not found")
107    slot = slots[slot_id]
108    if not slot["available"]:
109        raise HTTPException(status_code=400, detail="Slot already booked")
110    slot["available"] = False
111    appointment_id = next_ids["appointments"]
112    appointments[appointment_id] = {
113        "id": appointment_id,
114        "slot_id": slot_id,
115        "client_name": req.client_name,
116        "client_phone": req.client_phone,
117        "stylist_name": slot["stylist_name"],
118        "service_name": slot["service_name"],
119        "start_time": slot["start_time"],
120        "booked_by": user
121    }
122    next_ids["appointments"] += 1
123    return {"message": "Booked", "appointment_id": appointment_id}
124 
125@app.get("/appointments/{appointment_id}")
126def get_appointment(appointment_id: int, authorization: Optional[str] = Header(None)):
127    user = get_current_user(authorization)
128    if appointment_id not in appointments:
129        raise HTTPException(status_code=404, detail="Appointment not found")
130    return appointments[appointment_id]
131 
132@app.get("/stylists/{stylist_id}")
133def get_stylist(stylist_id: int):
134    if stylist_id not in stylists:
135        raise HTTPException(status_code=404, detail="Stylist not found")
136    return stylists[stylist_id]
137 
138@app.post("/stylists")
139def create_stylist(name: str, authorization: Optional[str] = Header(None)):
140    user = get_current_user(authorization)
141    sid = next_ids["stylists"]
142    stylists[sid] = {"id": sid, "name": name, "active": True}
143    next_ids["stylists"] += 1
144    return stylists[sid]
145 
146@app.get("/services/{service_id}")
147def get_service(service_id: int):
148    if service_id not in services:
149        raise HTTPException(status_code=404, detail="Service not found")
150    return services[service_id]
151 
152@app.post("/services")
153def create_service(name: str, duration_minutes: int, authorization: Optional[str] = Header(None)):
154    user = get_current_user(authorization)
155    sid = next_ids["services"]
156    services[sid] = {"id": sid, "name": name, "duration_minutes": duration_minutes}
157    next_ids["services"] += 1
158    return services[sid]

1fastapi
2uvicorn