Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import hashlib
5import random
6import string
7 
8app = FastAPI()
9 
10users = {}
11tokens = {}
12bookings = {}
13booking_id_counter = 1
14 
15def generate_token():
16    return ''.join(random.choices(string.ascii_letters + string.digits, k=32))
17 
18def hash_password(password: str):
19    return hashlib.sha256(password.encode()).hexdigest()
20 
21def get_current_user(authorization: str = Header(None)):
22    if not authorization:
23        raise HTTPException(status_code=401, detail="Missing token")
24    token = authorization.replace("Bearer ", "")
25    if token not in tokens:
26        raise HTTPException(status_code=401, detail="Invalid token")
27    return tokens[token]
28 
29class SignupRequest(BaseModel):
30    username: str
31    password: str
32 
33class LoginRequest(BaseModel):
34    username: str
35    password: str
36 
37class BookingCreate(BaseModel):
38    destination: str
39    travel_dates: str
40    passengers: int
41    tier: str = "economy"
42 
43class BookingUpdate(BaseModel):
44    destination: Optional[str] = None
45    travel_dates: Optional[str] = None
46    passengers: Optional[int] = None
47    tier: Optional[str] = None
48 
49@app.post("/signup")
50def signup(req: SignupRequest):
51    if req.username in users:
52        raise HTTPException(status_code=400, detail="User exists")
53    users[req.username] = hash_password(req.password)
54    return {"message": "User created"}
55 
56@app.post("/login")
57def login(req: LoginRequest):
58    if req.username not in users or users[req.username] != hash_password(req.password):
59        raise HTTPException(status_code=401, detail="Invalid credentials")
60    token = generate_token()
61    tokens[token] = req.username
62    return {"token": token}
63 
64@app.post("/bookings")
65def create_booking(req: BookingCreate, authorization: str = Header(None)):
66    user = get_current_user(authorization)
67    global booking_id_counter
68    price = 100 if req.tier == "economy" else 250
69    price *= req.passengers
70    booking = {
71        "id": booking_id_counter,
72        "user": user,
73        "destination": req.destination,
74        "travel_dates": req.travel_dates,
75        "passengers": req.passengers,
76        "tier": req.tier,
77        "price": price
78    }
79    bookings[booking_id_counter] = booking
80    booking_id_counter += 1
81    return booking
82 
83@app.get("/bookings/{booking_id}")
84def get_booking(booking_id: int, authorization: str = Header(None)):
85    user = get_current_user(authorization)
86    if booking_id not in bookings:
87        raise HTTPException(status_code=404, detail="Booking not found")
88    return bookings[booking_id]
89 
90@app.put("/bookings/{booking_id}")
91def update_booking(booking_id: int, req: BookingUpdate, authorization: str = Header(None)):
92    user = get_current_user(authorization)
93    if booking_id not in bookings:
94        raise HTTPException(status_code=404, detail="Booking not found")
95    booking = bookings[booking_id]
96    if req.destination is not None:
97        booking["destination"] = req.destination
98    if req.travel_dates is not None:
99        booking["travel_dates"] = req.travel_dates
100    if req.passengers is not None:
101        booking["passengers"] = req.passengers
102    if req.tier is not None:
103        booking["tier"] = req.tier
104    price = 100 if booking["tier"] == "economy" else 250
105    price *= booking["passengers"]
106    booking["price"] = price
107    return booking

1fastapi
2uvicorn