Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from datetime import datetime, timedelta
3import hashlib
4import secrets
5import uuid
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11courts = {}
12bookings = {}
13booking_id_counter = 1
14user_id_counter = 1
15 
16def verify_token(authorization: str = Header(...)):
17    if not authorization.startswith("Bearer "):
18        raise HTTPException(status_code=401, detail="Invalid auth header")
19    token = authorization.split(" ")[1]
20    user_id = tokens.get(token)
21    if user_id is None:
22        raise HTTPException(status_code=401, detail="Invalid token")
23    return user_id
24 
25@app.post("/signup")
26def signup(username: str, password: str):
27    global user_id_counter
28    if any(u["username"] == username for u in users.values()):
29        raise HTTPException(status_code=400, detail="Username already exists")
30    user_id = user_id_counter
31    user_id_counter += 1
32    users[user_id] = {
33        "id": user_id,
34        "username": username,
35        "password": hashlib.sha256(password.encode()).hexdigest()
36    }
37    return {"user_id": user_id, "username": username}
38 
39@app.post("/login")
40def login(username: str, password: str):
41    for uid, u in users.items():
42        if u["username"] == username and u["password"] == hashlib.sha256(password.encode()).hexdigest():
43            token = secrets.token_hex(32)
44            tokens[token] = uid
45            return {"token": token}
46    raise HTTPException(status_code=401, detail="Invalid credentials")
47 
48@app.post("/courts")
49def create_court(name: str, authorization: str = Header(...)):
50    user_id = verify_token(authorization)
51    court_id = len(courts) + 1
52    courts[court_id] = {"id": court_id, "name": name, "created_by": user_id}
53    return courts[court_id]
54 
55@app.get("/courts/{court_id}")
56def get_court(court_id: int, authorization: str = Header(...)):
57    verify_token(authorization)
58    court = courts.get(court_id)
59    if not court:
60        raise HTTPException(status_code=404, detail="Court not found")
61    return court
62 
63@app.post("/bookings")
64def create_booking(court_id: int, start_time: str, hours: int, authorization: str = Header(...)):
65    global booking_id_counter
66    user_id = verify_token(authorization)
67    court = courts.get(court_id)
68    if not court:
69        raise HTTPException(status_code=404, detail="Court not found")
70    try:
71        start_dt = datetime.fromisoformat(start_time)
72    except:
73        raise HTTPException(status_code=400, detail="Invalid datetime format")
74    end_dt = start_dt + timedelta(hours=hours)
75    for bid, b in bookings.items():
76        if b["court_id"] == court_id:
77            b_start = datetime.fromisoformat(b["start_time"])
78            b_end = datetime.fromisoformat(b["end_time"])
79            if start_dt < b_end and end_dt > b_start:
80                raise HTTPException(status_code=409, detail="Time slot conflict")
81    booking_id = booking_id_counter
82    booking_id_counter += 1
83    bookings[booking_id] = {
84        "id": booking_id,
85        "court_id": court_id,
86        "user_id": user_id,
87        "start_time": start_time,
88        "end_time": end_dt.isoformat(),
89        "hours": hours,
90        "total_cost": hours * 10
91    }
92    return bookings[booking_id]
93 
94@app.get("/bookings/{booking_id}")
95def get_booking(booking_id: int, authorization: str = Header(...)):
96    verify_token(authorization)
97    booking = bookings.get(booking_id)
98    if not booking:
99        raise HTTPException(status_code=404, detail="Booking not found")
100    return booking

1fastapi
2uvicorn