Browse the data

1from fastapi import FastAPI, HTTPException, Header, Depends
2from datetime import datetime, timedelta
3import secrets
4import hashlib
5from typing import Optional
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11doctors = {
12    1: {"name": "Dr. Smith", "specialty": "Cardiology"},
13    2: {"name": "Dr. Jones", "specialty": "Dermatology"},
14    3: {"name": "Dr. Lee", "specialty": "Pediatrics"},
15    4: {"name": "Dr. Patel", "specialty": "Orthopedics"}
16}
17slots = {}
18bookings = {}
19booking_id_counter = 1
20 
21def get_current_user(authorization: Optional[str] = Header(None)):
22    if not authorization:
23        raise HTTPException(status_code=401, detail="Missing Authorization header")
24    token = authorization.replace("Bearer ", "")
25    if token not in tokens:
26        raise HTTPException(status_code=401, detail="Invalid token")
27    return tokens[token]
28 
29@app.post("/signup")
30def signup(username: str, password: str):
31    if username in users:
32        raise HTTPException(status_code=400, detail="User already exists")
33    users[username] = hashlib.sha256(password.encode()).hexdigest()
34    return {"message": "User created"}
35 
36@app.post("/login")
37def login(username: str, password: str):
38    if username not in users or users[username] != hashlib.sha256(password.encode()).hexdigest():
39        raise HTTPException(status_code=401, detail="Invalid credentials")
40    token = secrets.token_hex(16)
41    tokens[token] = username
42    return {"token": token}
43 
44@app.get("/available-slots")
45def get_available_slots(user: str = Depends(get_current_user)):
46    now = datetime.now()
47    available = []
48    for doc_id, doc in doctors.items():
49        for day_offset in range(7):
50            day = now + timedelta(days=day_offset)
51            if day.weekday() >= 5:
52                continue
53            for hour in range(9, 17):
54                slot_time = day.replace(hour=hour, minute=0, second=0, microsecond=0)
55                if slot_time <= now:
56                    continue
57                slot_id = f"{doc_id}_{slot_time.isoformat()}"
58                if slot_id not in slots:
59                    slots[slot_id] = {"doctor_id": doc_id, "time": slot_time, "booked": False}
60                if not slots[slot_id]["booked"]:
61                    available.append({
62                        "slot_id": slot_id,
63                        "doctor_name": doc["name"],
64                        "specialty": doc["specialty"],
65                        "time": slot_time.isoformat()
66                    })
67    return available
68 
69@app.post("/book")
70def book_slot(slot_id: str, patient_name: str, reason: str, user: str = Depends(get_current_user)):
71    global booking_id_counter
72    if slot_id not in slots:
73        raise HTTPException(status_code=404, detail="Slot not found")
74    if slots[slot_id]["booked"]:
75        raise HTTPException(status_code=400, detail="Slot already booked")
76    slots[slot_id]["booked"] = True
77    booking = {
78        "id": booking_id_counter,
79        "slot_id": slot_id,
80        "doctor_id": slots[slot_id]["doctor_id"],
81        "doctor_name": doctors[slots[slot_id]["doctor_id"]]["name"],
82        "time": slots[slot_id]["time"].isoformat(),
83        "patient_name": patient_name,
84        "reason": reason,
85        "booked_by": user
86    }
87    bookings[booking_id_counter] = booking
88    booking_id_counter += 1
89    return booking
90 
91@app.get("/bookings/{booking_id}")
92def get_booking(booking_id: int, user: str = Depends(get_current_user)):
93    if booking_id not in bookings:
94        raise HTTPException(status_code=404, detail="Booking not found")
95    return bookings[booking_id]
96 
97@app.get("/doctors/{doctor_id}")
98def get_doctor(doctor_id: int, user: str = Depends(get_current_user)):
99    if doctor_id not in doctors:
100        raise HTTPException(status_code=404, detail="Doctor not found")
101    return doctors[doctor_id]
102 
103@app.get("/users/{username}")
104def get_user(username: str, user: str = Depends(get_current_user)):
105    if username not in users:
106        raise HTTPException(status_code=404, detail="User not found")
107    return {"username": username}

1fastapi
2uvicorn