Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, List
4from datetime import date, datetime
5import uuid
6import copy
7 
8app = FastAPI()
9 
10users = {}
11tokens = {}
12bookings = {}
13booking_id_counter = 1
14cancellation_history = {}
15 
16def get_current_user(authorization: str = Header(None)):
17    if not authorization:
18        raise HTTPException(status_code=401, detail="Missing auth header")
19    token = authorization.replace("Bearer ", "")
20    if token not in tokens:
21        raise HTTPException(status_code=401, detail="Invalid token")
22    return tokens[token]
23 
24class SignupRequest(BaseModel):
25    username: str
26    password: str
27 
28class LoginRequest(BaseModel):
29    username: str
30    password: str
31 
32class BookingCreate(BaseModel):
33    check_in: date
34    check_out: date
35    room_type: str
36    special_requests: Optional[str] = ""
37 
38class BookingUpdate(BaseModel):
39    check_in: Optional[date] = None
40    check_out: Optional[date] = None
41    room_type: Optional[str] = None
42    special_requests: Optional[str] = None
43 
44class CancelRequest(BaseModel):
45    cancel: bool = True
46 
47@app.post("/signup")
48def signup(req: SignupRequest):
49    if req.username in users:
50        raise HTTPException(status_code=400, detail="User already exists")
51    users[req.username] = {"username": req.username, "password": req.password}
52    return {"message": "User created"}
53 
54@app.post("/login")
55def login(req: LoginRequest):
56    if req.username not in users or users[req.username]["password"] != req.password:
57        raise HTTPException(status_code=401, detail="Invalid credentials")
58    token = str(uuid.uuid4())
59    tokens[token] = req.username
60    return {"token": token}
61 
62@app.post("/bookings")
63def create_booking(booking: BookingCreate, authorization: str = Header(None)):
64    user = get_current_user(authorization)
65    global booking_id_counter
66    bid = booking_id_counter
67    booking_id_counter += 1
68    bookings[bid] = {
69        "id": bid,
70        "user": user,
71        "check_in": str(booking.check_in),
72        "check_out": str(booking.check_out),
73        "room_type": booking.room_type,
74        "special_requests": booking.special_requests,
75        "status": "active",
76        "created_at": datetime.utcnow().isoformat()
77    }
78    return bookings[bid]
79 
80@app.get("/bookings/{booking_id}")
81def get_booking(booking_id: int, authorization: str = Header(None)):
82    user = get_current_user(authorization)
83    if booking_id not in bookings:
84        raise HTTPException(status_code=404, detail="Booking not found")
85    return bookings[booking_id]
86 
87@app.patch("/bookings/{booking_id}")
88def update_booking(booking_id: int, update: BookingUpdate, authorization: str = Header(None)):
89    user = get_current_user(authorization)
90    if booking_id not in bookings:
91        raise HTTPException(status_code=404, detail="Booking not found")
92    if bookings[booking_id]["status"] == "cancelled":
93        raise HTTPException(status_code=400, detail="Cannot update cancelled booking")
94    b = bookings[booking_id]
95    if update.check_in is not None:
96        b["check_in"] = str(update.check_in)
97    if update.check_out is not None:
98        b["check_out"] = str(update.check_out)
99    if update.room_type is not None:
100        b["room_type"] = update.room_type
101    if update.special_requests is not None:
102        b["special_requests"] = update.special_requests
103    return b
104 
105@app.post("/bookings/{booking_id}/cancel")
106def cancel_booking(booking_id: int, req: CancelRequest, authorization: str = Header(None)):
107    user = get_current_user(authorization)
108    if booking_id not in bookings:
109        raise HTTPException(status_code=404, detail="Booking not found")
110    if bookings[booking_id]["status"] == "cancelled":
111        raise HTTPException(status_code=400, detail="Already cancelled")
112    bookings[booking_id]["status"] = "cancelled"
113    bookings[booking_id]["cancelled_at"] = datetime.utcnow().isoformat()
114    # Partial refund: 50%
115    refund_amount = 50.0
116    history_entry = {
117        "booking_id": booking_id,
118        "action": "cancelled",
119        "refund_percent": 50,
120        "refund_amount": refund_amount,
121        "timestamp": datetime.utcnow().isoformat()
122    }
123    if booking_id not in cancellation_history:
124        cancellation_history[booking_id] = []
125    cancellation_history[booking_id].append(history_entry)
126    return {
127        "message": "Booking cancelled",
128        "partial_refund": refund_amount,
129        "history": cancellation_history[booking_id]
130    }
131 
132@app.get("/bookings/{booking_id}/history")
133def get_cancellation_history(booking_id: int, authorization: str = Header(None)):
134    user = get_current_user(authorization)
135    if booking_id not in bookings:
136        raise HTTPException(status_code=404, detail="Booking not found")
137    return cancellation_history.get(booking_id, [])

1fastapi
2uvicorn