Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import secrets
4 
5app = FastAPI()
6 
7users = {}
8tokens = {}
9hotels = {}
10rooms = {}
11bookings = {}
12next_user_id = 1
13next_hotel_id = 1
14next_room_id = 1
15next_booking_id = 1
16 
17def get_user_from_token(authorization: Optional[str] = Header(None)):
18    if not authorization:
19        raise HTTPException(status_code=401, detail="No auth token")
20    token = authorization.replace("Bearer ", "")
21    user_id = tokens.get(token)
22    if not user_id:
23        raise HTTPException(status_code=401, detail="Invalid token")
24    return user_id
25 
26@app.post("/signup")
27def signup(username: str, password: str):
28    global next_user_id
29    for u in users.values():
30        if u["username"] == username:
31            raise HTTPException(status_code=400, detail="Username taken")
32    user_id = next_user_id
33    next_user_id += 1
34    users[user_id] = {"id": user_id, "username": username, "password": password}
35    return {"user_id": user_id}
36 
37@app.post("/login")
38def login(username: str, password: str):
39    for u in users.values():
40        if u["username"] == username and u["password"] == password:
41            token = secrets.token_hex(16)
42            tokens[token] = u["id"]
43            return {"token": token}
44    raise HTTPException(status_code=401, detail="Invalid credentials")
45 
46@app.post("/hotels")
47def create_hotel(name: str, location: str, authorization: Optional[str] = Header(None)):
48    get_user_from_token(authorization)
49    global next_hotel_id
50    hotel_id = next_hotel_id
51    next_hotel_id += 1
52    hotels[hotel_id] = {"id": hotel_id, "name": name, "location": location}
53    return {"hotel_id": hotel_id}
54 
55@app.get("/hotels/{hotel_id}")
56def get_hotel(hotel_id: int, authorization: Optional[str] = Header(None)):
57    get_user_from_token(authorization)
58    hotel = hotels.get(hotel_id)
59    if not hotel:
60        raise HTTPException(status_code=404, detail="Hotel not found")
61    return hotel
62 
63@app.post("/rooms")
64def create_room(hotel_id: int, room_number: str, price: float, authorization: Optional[str] = Header(None)):
65    get_user_from_token(authorization)
66    if hotel_id not in hotels:
67        raise HTTPException(status_code=404, detail="Hotel not found")
68    global next_room_id
69    room_id = next_room_id
70    next_room_id += 1
71    rooms[room_id] = {"id": room_id, "hotel_id": hotel_id, "room_number": room_number, "price": price, "available": True}
72    return {"room_id": room_id}
73 
74@app.get("/rooms/{room_id}")
75def get_room(room_id: int, authorization: Optional[str] = Header(None)):
76    get_user_from_token(authorization)
77    room = rooms.get(room_id)
78    if not room:
79        raise HTTPException(status_code=404, detail="Room not found")
80    return room
81 
82@app.post("/bookings")
83def create_booking(room_id: int, guest_name: str, authorization: Optional[str] = Header(None)):
84    user_id = get_user_from_token(authorization)
85    if room_id not in rooms:
86        raise HTTPException(status_code=404, detail="Room not found")
87    if not rooms[room_id]["available"]:
88        raise HTTPException(status_code=400, detail="Room not available")
89    global next_booking_id
90    booking_id = next_booking_id
91    next_booking_id += 1
92    rooms[room_id]["available"] = False
93    bookings[booking_id] = {"id": booking_id, "room_id": room_id, "guest_name": guest_name, "user_id": user_id}
94    return {"booking_id": booking_id}
95 
96@app.get("/bookings/{booking_id}")
97def get_booking(booking_id: int, authorization: Optional[str] = Header(None)):
98    get_user_from_token(authorization)
99    booking = bookings.get(booking_id)
100    if not booking:
101        raise HTTPException(status_code=404, detail="Booking not found")
102    return booking

1fastapi
2uvicorn