Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import secrets
4import datetime
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10airlines = {}
11passes = {}
12bookings = {}
13next_user_id = 1
14next_airline_id = 1
15next_pass_id = 1
16next_booking_id = 1
17 
18def get_current_user(authorization: Optional[str] = Header(None)):
19    if not authorization:
20        raise HTTPException(status_code=401, detail="Missing auth token")
21    token = authorization.replace("Bearer ", "")
22    if token not in tokens:
23        raise HTTPException(status_code=401, detail="Invalid token")
24    user_id = tokens[token]
25    if user_id not in users:
26        raise HTTPException(status_code=401, detail="User not found")
27    return user_id
28 
29@app.post("/signup")
30def signup(username: str, password: str):
31    global next_user_id
32    user_id = next_user_id
33    next_user_id += 1
34    users[user_id] = {"username": username, "password": password}
35    return {"user_id": user_id}
36 
37@app.post("/login")
38def login(username: str, password: str):
39    for uid, u in users.items():
40        if u["username"] == username and u["password"] == password:
41            token = secrets.token_hex(16)
42            tokens[token] = uid
43            return {"token": token, "user_id": uid}
44    raise HTTPException(status_code=401, detail="Invalid credentials")
45 
46@app.post("/airlines")
47def create_airline(name: str, authorization: Optional[str] = Header(None)):
48    get_current_user(authorization)
49    global next_airline_id
50    airline_id = next_airline_id
51    next_airline_id += 1
52    airlines[airline_id] = {"id": airline_id, "name": name}
53    return airlines[airline_id]
54 
55@app.get("/airlines/{airline_id}")
56def get_airline(airline_id: int, authorization: Optional[str] = Header(None)):
57    get_current_user(authorization)
58    if airline_id not in airlines:
59        raise HTTPException(status_code=404, detail="Airline not found")
60    return airlines[airline_id]
61 
62@app.post("/bookings")
63def create_booking(airline_id: int, passenger_name: str, flight_number: str, authorization: Optional[str] = Header(None)):
64    user_id = get_current_user(authorization)
65    global next_booking_id
66    booking_id = next_booking_id
67    next_booking_id += 1
68    bookings[booking_id] = {
69        "id": booking_id,
70        "airline_id": airline_id,
71        "passenger_name": passenger_name,
72        "flight_number": flight_number,
73        "user_id": user_id
74    }
75    return bookings[booking_id]
76 
77@app.get("/bookings/{booking_id}")
78def get_booking(booking_id: int, authorization: Optional[str] = Header(None)):
79    get_current_user(authorization)
80    if booking_id not in bookings:
81        raise HTTPException(status_code=404, detail="Booking not found")
82    return bookings[booking_id]
83 
84@app.post("/passes")
85def create_pass(booking_id: int, seat: str, gate: str, boarding_time: str, authorization: Optional[str] = Header(None)):
86    user_id = get_current_user(authorization)
87    if booking_id not in bookings:
88        raise HTTPException(status_code=404, detail="Booking not found")
89    if bookings[booking_id]["user_id"] != user_id:
90        raise HTTPException(status_code=403, detail="Not your booking")
91    global next_pass_id
92    pass_id = next_pass_id
93    next_pass_id += 1
94    passes[pass_id] = {
95        "id": pass_id,
96        "booking_id": booking_id,
97        "seat": seat,
98        "gate": gate,
99        "boarding_time": boarding_time,
100        "airline_id": bookings[booking_id]["airline_id"],
101        "passenger_name": bookings[booking_id]["passenger_name"],
102        "flight_number": bookings[booking_id]["flight_number"]
103    }
104    return passes[pass_id]
105 
106@app.get("/passes/{pass_id}")
107def get_pass(pass_id: int, authorization: Optional[str] = Header(None)):
108    get_current_user(authorization)
109    if pass_id not in passes:
110        raise HTTPException(status_code=404, detail="Pass not found")
111    return passes[pass_id]
112 
113@app.get("/passes/by-booking/{booking_id}")
114def get_pass_by_booking(booking_id: int, authorization: Optional[str] = Header(None)):
115    get_current_user(authorization)
116    for p in passes.values():
117        if p["booking_id"] == booking_id:
118            return p
119    raise HTTPException(status_code=404, detail="No pass found for this booking")

1fastapi
2uvicorn