Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import secrets
5 
6app = FastAPI()
7 
8users = {}
9user_tokens = {}
10bookings = {}
11next_user_id = 1
12next_booking_id = 1
13 
14class SignupRequest(BaseModel):
15    username: str
16    password: str
17 
18class LoginRequest(BaseModel):
19    username: str
20    password: str
21 
22class BookingCreate(BaseModel):
23    destination: str
24    travel_dates: str
25    accommodation_type: str
26    loyalty_tier: str
27    traveler_name: str
28    traveler_contact: str
29 
30class BookingUpdate(BaseModel):
31    destination: Optional[str] = None
32    travel_dates: Optional[str] = None
33    accommodation_type: Optional[str] = None
34    loyalty_tier: Optional[str] = None
35    traveler_name: Optional[str] = None
36    traveler_contact: Optional[str] = None
37 
38def get_current_user(authorization: str = Header(...)):
39    token = authorization.replace("Bearer ", "")
40    for uid, t in user_tokens.items():
41        if t == token:
42            return uid
43    raise HTTPException(status_code=401, detail="Invalid token")
44 
45@app.post("/signup")
46def signup(req: SignupRequest):
47    global next_user_id
48    uid = next_user_id
49    next_user_id += 1
50    users[uid] = {"username": req.username, "password": req.password}
51    return {"id": uid, "username": req.username}
52 
53@app.post("/login")
54def login(req: LoginRequest):
55    for uid, u in users.items():
56        if u["username"] == req.username and u["password"] == req.password:
57            token = secrets.token_hex(16)
58            user_tokens[uid] = token
59            return {"token": token}
60    raise HTTPException(status_code=401, detail="Invalid credentials")
61 
62@app.post("/bookings")
63def create_booking(req: BookingCreate, authorization: str = Header(...)):
64    global next_booking_id
65    uid = get_current_user(authorization)
66    bid = next_booking_id
67    next_booking_id += 1
68    bookings[bid] = {
69        "id": bid,
70        "user_id": uid,
71        "destination": req.destination,
72        "travel_dates": req.travel_dates,
73        "accommodation_type": req.accommodation_type,
74        "loyalty_tier": req.loyalty_tier,
75        "traveler_name": req.traveler_name,
76        "traveler_contact": req.traveler_contact,
77    }
78    return bookings[bid]
79 
80@app.get("/bookings/{booking_id}")
81def get_booking(booking_id: int, authorization: str = Header(...)):
82    uid = 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.put("/bookings/{booking_id}")
88def update_booking(booking_id: int, req: BookingUpdate, authorization: str = Header(...)):
89    uid = get_current_user(authorization)
90    if booking_id not in bookings:
91        raise HTTPException(status_code=404, detail="Booking not found")
92    b = bookings[booking_id]
93    if req.destination is not None:
94        b["destination"] = req.destination
95    if req.travel_dates is not None:
96        b["travel_dates"] = req.travel_dates
97    if req.accommodation_type is not None:
98        b["accommodation_type"] = req.accommodation_type
99    if req.loyalty_tier is not None:
100        b["loyalty_tier"] = req.loyalty_tier
101    if req.traveler_name is not None:
102        b["traveler_name"] = req.traveler_name
103    if req.traveler_contact is not None:
104        b["traveler_contact"] = req.traveler_contact
105    return b

1fastapi
2uvicorn