Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, List
4import random
5import string
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11hotels = {}
12hotel_id_counter = 1
13 
14def generate_token():
15    return ''.join(random.choices(string.ascii_letters + string.digits, k=32))
16 
17class SignupRequest(BaseModel):
18    username: str
19    password: str
20 
21class LoginRequest(BaseModel):
22    username: str
23    password: str
24 
25class HotelCreate(BaseModel):
26    name: str
27    star_rating: int
28    city: str
29    amenities: List[str] = []
30    price_per_night: float
31 
32@app.post("/signup")
33def signup(req: SignupRequest):
34    if req.username in users:
35        raise HTTPException(status_code=400, detail="User exists")
36    users[req.username] = req.password
37    token = generate_token()
38    tokens[token] = req.username
39    return {"token": token}
40 
41@app.post("/login")
42def login(req: LoginRequest):
43    if req.username not in users or users[req.username] != req.password:
44        raise HTTPException(status_code=401, detail="Invalid credentials")
45    token = generate_token()
46    tokens[token] = req.username
47    return {"token": token}
48 
49def get_current_user(authorization: str = Header(None)):
50    if not authorization:
51        raise HTTPException(status_code=401, detail="Missing auth header")
52    token = authorization.replace("Bearer ", "")
53    if token not in tokens:
54        raise HTTPException(status_code=401, detail="Invalid token")
55    return tokens[token]
56 
57@app.get("/hotels/{hotel_id}")
58def get_hotel(hotel_id: int, authorization: str = Header(None)):
59    get_current_user(authorization)
60    if hotel_id not in hotels:
61        raise HTTPException(status_code=404, detail="Hotel not found")
62    return hotels[hotel_id]
63 
64@app.post("/hotels")
65def create_hotel(hotel: HotelCreate, authorization: str = Header(None)):
66    get_current_user(authorization)
67    global hotel_id_counter
68    hotel_id = hotel_id_counter
69    hotel_id_counter += 1
70    hotels[hotel_id] = {
71        "id": hotel_id,
72        "name": hotel.name,
73        "star_rating": hotel.star_rating,
74        "city": hotel.city,
75        "amenities": hotel.amenities,
76        "price_per_night": hotel.price_per_night
77    }
78    return hotels[hotel_id]
79 
80@app.get("/hotels")
81def list_hotels(
82    city: Optional[str] = None,
83    amenities: Optional[str] = None,
84    min_price: Optional[float] = None,
85    max_price: Optional[float] = None,
86    authorization: str = Header(None)
87):
88    get_current_user(authorization)
89    results = []
90    for h in hotels.values():
91        if city and h["city"].lower() != city.lower():
92            continue
93        if amenities:
94            req_amenities = [a.strip().lower() for a in amenities.split(",")]
95            hotel_amenities = [a.lower() for a in h["amenities"]]
96            if not all(a in hotel_amenities for a in req_amenities):
97                continue
98        if min_price is not None and h["price_per_night"] < min_price:
99            continue
100        if max_price is not None and h["price_per_night"] > max_price:
101            continue
102        results.append(h)
103    return results

1fastapi
2uvicorn