Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import secrets
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10travelers = {}
11traveler_id_counter = 1
12user_id_counter = 1
13 
14class SignupRequest(BaseModel):
15    username: str
16    password: str
17 
18class LoginRequest(BaseModel):
19    username: str
20    password: str
21 
22class TravelerUpdate(BaseModel):
23    name: Optional[str] = None
24    passport_info: Optional[str] = None
25    travel_preferences: Optional[str] = None
26    membership_tier: Optional[str] = None
27 
28class TravelerCreate(BaseModel):
29    name: str
30    passport_info: str
31    travel_preferences: str
32    membership_tier: str
33 
34def get_current_user(authorization: str = Header(...)):
35    token = authorization.replace("Bearer ", "")
36    if token not in tokens:
37        raise HTTPException(status_code=401, detail="Invalid token")
38    return tokens[token]
39 
40@app.post("/signup")
41def signup(req: SignupRequest):
42    global user_id_counter
43    if any(u["username"] == req.username for u in users.values()):
44        raise HTTPException(status_code=400, detail="Username already exists")
45    user_id = user_id_counter
46    users[user_id] = {"id": user_id, "username": req.username, "password": req.password}
47    user_id_counter += 1
48    token = secrets.token_hex(16)
49    tokens[token] = user_id
50    return {"token": token, "user_id": user_id}
51 
52@app.post("/login")
53def login(req: LoginRequest):
54    for uid, u in users.items():
55        if u["username"] == req.username and u["password"] == req.password:
56            token = secrets.token_hex(16)
57            tokens[token] = uid
58            return {"token": token, "user_id": uid}
59    raise HTTPException(status_code=401, detail="Invalid credentials")
60 
61@app.get("/travelers/{traveler_id}")
62def get_traveler(traveler_id: int, authorization: str = Header(...)):
63    get_current_user(authorization)
64    if traveler_id not in travelers:
65        raise HTTPException(status_code=404, detail="Traveler not found")
66    return travelers[traveler_id]
67 
68@app.post("/travelers")
69def create_traveler(traveler: TravelerCreate, authorization: str = Header(...)):
70    global traveler_id_counter
71    get_current_user(authorization)
72    new_id = traveler_id_counter
73    travelers[new_id] = {
74        "id": new_id,
75        "name": traveler.name,
76        "passport_info": traveler.passport_info,
77        "travel_preferences": traveler.travel_preferences,
78        "membership_tier": traveler.membership_tier
79    }
80    traveler_id_counter += 1
81    return travelers[new_id]
82 
83@app.put("/travelers/{traveler_id}")
84def update_traveler(traveler_id: int, update: TravelerUpdate, authorization: str = Header(...)):
85    get_current_user(authorization)
86    if traveler_id not in travelers:
87        raise HTTPException(status_code=404, detail="Traveler not found")
88    t = travelers[traveler_id]
89    if update.name is not None:
90        t["name"] = update.name
91    if update.passport_info is not None:
92        t["passport_info"] = update.passport_info
93    if update.travel_preferences is not None:
94        t["travel_preferences"] = update.travel_preferences
95    if update.membership_tier is not None:
96        t["membership_tier"] = update.membership_tier
97    return t

1fastapi
2uvicorn