Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from datetime import datetime, timedelta
3from typing import Optional
4import secrets
5import uvicorn
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11appointments = {}
12available_slots = {}
13peak_hours_config = {}
14resource_counters = {"users": 0, "appointments": 0, "available_slots": 0}
15 
16# Seed some data
17def seed():
18    # Create admin user
19    users[1] = {"id": 1, "email": "admin@test.com", "password": "admin123"}
20    resource_counters["users"] = 1
21    
22    # Create some available slots for today
23    now = datetime.now()
24    for i in range(8):
25        slot_id = i + 1
26        start = now.replace(hour=9 + i, minute=0, second=0, microsecond=0)
27        end = start + timedelta(hours=1)
28        available_slots[slot_id] = {
29            "id": slot_id,
30            "start": start.isoformat(),
31            "end": end.isoformat(),
32            "available": True
33        }
34        resource_counters["available_slots"] = slot_id
35    
36    # Create a booked appointment
37    appointments[1] = {
38        "id": 1,
39        "user_id": 1,
40        "start": now.replace(hour=10, minute=0, second=0, microsecond=0).isoformat(),
41        "end": now.replace(hour=11, minute=0, second=0, microsecond=0).isoformat(),
42        "status": "booked"
43    }
44    resource_counters["appointments"] = 1
45    
46    # Set peak hours
47    peak_hours_config["peak_hours"] = [
48        {"start": "09:00", "end": "11:00"},
49        {"start": "14:00", "end": "16:00"}
50    ]
51 
52seed()
53 
54def get_user_from_token(authorization: str = Header(...)):
55    if not authorization.startswith("Bearer "):
56        raise HTTPException(status_code=401, detail="Invalid auth header")
57    token = authorization.split(" ")[1]
58    for uid, t in tokens.items():
59        if t == token:
60            return users[uid]
61    raise HTTPException(status_code=401, detail="Invalid token")
62 
63@app.post("/signup")
64def signup(email: str, password: str):
65    resource_counters["users"] += 1
66    uid = resource_counters["users"]
67    users[uid] = {"id": uid, "email": email, "password": password}
68    return {"id": uid, "email": email}
69 
70@app.post("/login")
71def login(email: str, password: str):
72    for uid, user in users.items():
73        if user["email"] == email and user["password"] == password:
74            token = secrets.token_hex(16)
75            tokens[uid] = token
76            return {"token": token, "user_id": uid}
77    raise HTTPException(status_code=401, detail="Invalid credentials")
78 
79@app.get("/users/{user_id}")
80def get_user(user_id: int, authorization: str = Header(...)):
81    get_user_from_token(authorization)
82    if user_id not in users:
83        raise HTTPException(status_code=404, detail="User not found")
84    return users[user_id]
85 
86@app.post("/users")
87def create_user(email: str, password: str, authorization: str = Header(...)):
88    get_user_from_token(authorization)
89    resource_counters["users"] += 1
90    uid = resource_counters["users"]
91    users[uid] = {"id": uid, "email": email, "password": password}
92    return {"id": uid, "email": email}
93 
94@app.get("/appointments/{appointment_id}")
95def get_appointment(appointment_id: int, authorization: str = Header(...)):
96    get_user_from_token(authorization)
97    if appointment_id not in appointments:
98        raise HTTPException(status_code=404, detail="Appointment not found")
99    return appointments[appointment_id]
100 
101@app.post("/appointments")
102def create_appointment(start: str, end: str, authorization: str = Header(...)):
103    user = get_user_from_token(authorization)
104    resource_counters["appointments"] += 1
105    aid = resource_counters["appointments"]
106    appointments[aid] = {
107        "id": aid,
108        "user_id": user["id"],
109        "start": start,
110        "end": end,
111        "status": "booked"
112    }
113    return appointments[aid]
114 
115@app.get("/schedule/available")
116def get_available_schedule(authorization: str = Header(...)):
117    get_user_from_token(authorization)
118    return {
119        "available_slots": {k: v for k, v in available_slots.items() if v["available"]},
120        "booked_appointments": appointments,
121        "peak_hours": peak_hours_config["peak_hours"]
122    }

1fastapi
2uvicorn