Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from datetime import datetime, timedelta
3import random
4import string
5import hashlib
6import uvicorn
7 
8app = FastAPI()
9 
10users = {}
11tokens = {}
12appointments = {}
13specialty_index = {}
14next_appointment_id = 1
15 
16def generate_token():
17    return ''.join(random.choices(string.ascii_letters + string.digits, k=32))
18 
19def get_current_user(authorization: str = Header(None)):
20    if not authorization:
21        raise HTTPException(status_code=401, detail="Missing authorization header")
22    token = authorization.replace("Bearer ", "")
23    if token not in tokens:
24        raise HTTPException(status_code=401, detail="Invalid token")
25    return tokens[token]
26 
27@app.post("/signup")
28def signup(username: str, password: str):
29    if username in users:
30        raise HTTPException(status_code=400, detail="User already exists")
31    users[username] = hashlib.sha256(password.encode()).hexdigest()
32    token = generate_token()
33    tokens[token] = username
34    return {"token": token}
35 
36@app.post("/login")
37def login(username: str, password: str):
38    if username not in users:
39        raise HTTPException(status_code=401, detail="Invalid credentials")
40    if users[username] != hashlib.sha256(password.encode()).hexdigest():
41        raise HTTPException(status_code=401, detail="Invalid credentials")
42    token = generate_token()
43    tokens[token] = username
44    return {"token": token}
45 
46@app.get("/appointments/{appointment_id}")
47def get_appointment(appointment_id: int, authorization: str = Header(None)):
48    get_current_user(authorization)
49    if appointment_id not in appointments:
50        raise HTTPException(status_code=404, detail="Appointment not found")
51    return appointments[appointment_id]
52 
53@app.post("/appointments")
54def create_appointment(doctor_name: str, clinic_address: str, specialty: str, insurance: str, telehealth: bool, next_available_date: str, authorization: str = Header(None)):
55    get_current_user(authorization)
56    global next_appointment_id
57    appointment = {
58        "id": next_appointment_id,
59        "doctor_name": doctor_name,
60        "clinic_address": clinic_address,
61        "specialty": specialty,
62        "insurance": insurance,
63        "telehealth": telehealth,
64        "next_available_date": next_available_date
65    }
66    appointments[next_appointment_id] = appointment
67    spec_key = specialty.lower()
68    if spec_key not in specialty_index:
69        specialty_index[spec_key] = []
70    specialty_index[spec_key].append(next_appointment_id)
71    next_appointment_id += 1
72    return appointment
73 
74@app.get("/appointments")
75def get_appointments(specialty: str, insurance: str = None, telehealth: bool = None, next_available_date: str = None, authorization: str = Header(None)):
76    get_current_user(authorization)
77    spec_key = specialty.lower()
78    if spec_key not in specialty_index:
79        return []
80    result = []
81    for appt_id in specialty_index[spec_key]:
82        appt = appointments[appt_id]
83        if insurance and appt["insurance"].lower() != insurance.lower():
84            continue
85        if telehealth is not None and appt["telehealth"] != telehealth:
86            continue
87        if next_available_date and appt["next_available_date"] < next_available_date:
88            continue
89        result.append({
90            "doctor_name": appt["doctor_name"],
91            "clinic_address": appt["clinic_address"]
92        })
93    return result

1fastapi
2uvicorn