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 = {}
10trips = {}
11trip_id_counter = 1
12 
13class SignupRequest(BaseModel):
14    username: str
15    password: str
16 
17class LoginRequest(BaseModel):
18    username: str
19    password: str
20 
21class TripCreate(BaseModel):
22    destination: str
23    dates: str
24    accommodation: str = ""
25    activities: list[str] = []
26    sharing: bool = False
27 
28class TripUpdate(BaseModel):
29    destination: Optional[str] = None
30    dates: Optional[str] = None
31    accommodation: Optional[str] = None
32    activities: Optional[list[str]] = None
33    sharing: Optional[bool] = None
34    tier: Optional[str] = None
35 
36def get_user_from_token(authorization: str = Header(...)):
37    if not authorization.startswith("Bearer "):
38        raise HTTPException(status_code=401, detail="Invalid auth header")
39    token = authorization[7:]
40    if token not in tokens:
41        raise HTTPException(status_code=401, detail="Invalid token")
42    return tokens[token]
43 
44@app.post("/signup")
45def signup(req: SignupRequest):
46    if req.username in users:
47        raise HTTPException(status_code=400, detail="User already exists")
48    users[req.username] = {"password": req.password}
49    token = secrets.token_hex(16)
50    tokens[token] = req.username
51    return {"token": token}
52 
53@app.post("/login")
54def login(req: LoginRequest):
55    if req.username not in users or users[req.username]["password"] != req.password:
56        raise HTTPException(status_code=401, detail="Invalid credentials")
57    token = secrets.token_hex(16)
58    tokens[token] = req.username
59    return {"token": token}
60 
61@app.post("/trips")
62def create_trip(trip: TripCreate, authorization: str = Header(...)):
63    user = get_user_from_token(authorization)
64    global trip_id_counter
65    trip_id = trip_id_counter
66    trip_id_counter += 1
67    trips[trip_id] = {
68        "id": trip_id,
69        "destination": trip.destination,
70        "dates": trip.dates,
71        "accommodation": trip.accommodation,
72        "activities": trip.activities,
73        "sharing": trip.sharing,
74        "tier": "free",
75        "owner": user
76    }
77    return trips[trip_id]
78 
79@app.get("/trips/{trip_id}")
80def get_trip(trip_id: int, authorization: str = Header(...)):
81    user = get_user_from_token(authorization)
82    if trip_id not in trips:
83        raise HTTPException(status_code=404, detail="Trip not found")
84    return trips[trip_id]
85 
86@app.put("/trips/{trip_id}")
87def update_trip(trip_id: int, trip: TripUpdate, authorization: str = Header(...)):
88    user = get_user_from_token(authorization)
89    if trip_id not in trips:
90        raise HTTPException(status_code=404, detail="Trip not found")
91    if trips[trip_id]["owner"] != user:
92        raise HTTPException(status_code=403, detail="Not your trip")
93    data = trips[trip_id]
94    if trip.destination is not None:
95        data["destination"] = trip.destination
96    if trip.dates is not None:
97        data["dates"] = trip.dates
98    if trip.accommodation is not None:
99        data["accommodation"] = trip.accommodation
100    if trip.activities is not None:
101        data["activities"] = trip.activities
102    if trip.sharing is not None:
103        data["sharing"] = trip.sharing
104    if trip.tier is not None:
105        data["tier"] = trip.tier
106    return data

1fastapi
2uvicorn