Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import uuid
5import time
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11bookings = {}
12booking_id_counter = 1
13 
14def get_current_user(authorization: str = Header(None)):
15    if not authorization:
16        raise HTTPException(status_code=401, detail="Missing auth token")
17    token = authorization.replace("Bearer ", "")
18    if token not in tokens:
19        raise HTTPException(status_code=401, detail="Invalid token")
20    return tokens[token]
21 
22class SignupRequest(BaseModel):
23    username: str
24    password: str
25 
26class LoginRequest(BaseModel):
27    username: str
28    password: str
29 
30class CreateBookingRequest(BaseModel):
31    check_in: str
32    check_out: str
33    room_type: str
34    guest_count: int
35    special_requests: Optional[str] = ""
36 
37class UpdateBookingRequest(BaseModel):
38    check_in: Optional[str] = None
39    check_out: Optional[str] = None
40    room_type: Optional[str] = None
41    guest_count: Optional[int] = None
42    special_requests: Optional[str] = None
43 
44BASE_PRICES = {"single": 100, "double": 150, "suite": 250}
45 
46def calculate_price(check_in: str, check_out: str, room_type: str, guest_count: int):
47    try:
48        in_dt = time.strptime(check_in, "%Y-%m-%d")
49        out_dt = time.strptime(check_out, "%Y-%m-%d")
50        nights = max(1, (time.mktime(out_dt) - time.mktime(in_dt)) // 86400)
51    except:
52        nights = 1
53    base = BASE_PRICES.get(room_type, 100)
54    guest_mult = 1 + max(0, guest_count - 2) * 0.25
55    return int(nights * base * guest_mult)
56 
57@app.post("/signup")
58def signup(req: SignupRequest):
59    if req.username in users:
60        raise HTTPException(status_code=400, detail="User already exists")
61    users[req.username] = {"password": req.password}
62    return {"msg": "User created"}
63 
64@app.post("/login")
65def login(req: LoginRequest):
66    user = users.get(req.username)
67    if not user or user["password"] != req.password:
68        raise HTTPException(status_code=401, detail="Invalid credentials")
69    token = str(uuid.uuid4())
70    tokens[token] = req.username
71    return {"token": token}
72 
73@app.post("/bookings")
74def create_booking(req: CreateBookingRequest, authorization: str = Header(None)):
75    user = get_current_user(authorization)
76    global booking_id_counter
77    bid = booking_id_counter
78    booking_id_counter += 1
79    price = calculate_price(req.check_in, req.check_out, req.room_type, req.guest_count)
80    bookings[bid] = {
81        "id": bid,
82        "user": user,
83        "check_in": req.check_in,
84        "check_out": req.check_out,
85        "room_type": req.room_type,
86        "guest_count": req.guest_count,
87        "special_requests": req.special_requests,
88        "total_price": price
89    }
90    return bookings[bid]
91 
92@app.get("/bookings/{booking_id}")
93def get_booking(booking_id: int, authorization: str = Header(None)):
94    get_current_user(authorization)
95    if booking_id not in bookings:
96        raise HTTPException(status_code=404, detail="Booking not found")
97    return bookings[booking_id]
98 
99@app.put("/bookings/{booking_id}")
100def update_booking(booking_id: int, req: UpdateBookingRequest, authorization: str = Header(None)):
101    user = get_current_user(authorization)
102    if booking_id not in bookings:
103        raise HTTPException(status_code=404, detail="Booking not found")
104    booking = bookings[booking_id]
105    if booking["user"] != user:
106        raise HTTPException(status_code=403, detail="Not your booking")
107    if req.check_in is not None:
108        booking["check_in"] = req.check_in
109    if req.check_out is not None:
110        booking["check_out"] = req.check_out
111    if req.room_type is not None:
112        booking["room_type"] = req.room_type
113    if req.guest_count is not None:
114        booking["guest_count"] = req.guest_count
115    if req.special_requests is not None:
116        booking["special_requests"] = req.special_requests
117    booking["total_price"] = calculate_price(booking["check_in"], booking["check_out"], booking["room_type"], booking["guest_count"])
118    return booking

1fastapi
2uvicorn