Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import uuid
4 
5app = FastAPI()
6 
7users = {}
8trips = {}
9trip_participants = {}
10tokens = {}
11next_user_id = 1
12next_trip_id = 1
13next_participant_id = 1
14 
15def get_current_user(authorization: Optional[str] = Header(None)):
16    if not authorization or not authorization.startswith("Bearer "):
17        raise HTTPException(status_code=401, detail="Invalid token")
18    token = authorization.split(" ")[1]
19    if token not in tokens:
20        raise HTTPException(status_code=401, detail="Invalid token")
21    return tokens[token]
22 
23@app.post("/signup")
24def signup(username: str, password: str):
25    global next_user_id
26    for u in users.values():
27        if u["username"] == username:
28            raise HTTPException(status_code=400, detail="Username taken")
29    user_id = next_user_id
30    next_user_id += 1
31    users[user_id] = {"id": user_id, "username": username, "password": password}
32    return {"id": user_id, "username": username}
33 
34@app.post("/login")
35def login(username: str, password: str):
36    for u in users.values():
37        if u["username"] == username and u["password"] == password:
38            token = str(uuid.uuid4())
39            tokens[token] = u["id"]
40            return {"token": token}
41    raise HTTPException(status_code=401, detail="Invalid credentials")
42 
43@app.get("/trips/{trip_id}")
44def get_trip(trip_id: int, authorization: Optional[str] = Header(None)):
45    user_id = get_current_user(authorization)
46    if trip_id not in trips:
47        raise HTTPException(status_code=404, detail="Trip not found")
48    trip = trips[trip_id].copy()
49    trip["participants"] = [p for p in trip_participants.values() if p["trip_id"] == trip_id]
50    return trip
51 
52@app.post("/trips")
53def create_trip(name: str, destination: str, start_date: str, end_date: str, authorization: Optional[str] = Header(None)):
54    user_id = get_current_user(authorization)
55    global next_trip_id
56    trip_id = next_trip_id
57    next_trip_id += 1
58    trips[trip_id] = {
59        "id": trip_id,
60        "name": name,
61        "destination": destination,
62        "start_date": start_date,
63        "end_date": end_date,
64        "created_by": user_id
65    }
66    # Add creator as participant with role "organizer"
67    global next_participant_id
68    participant_id = next_participant_id
69    next_participant_id += 1
70    trip_participants[participant_id] = {
71        "id": participant_id,
72        "trip_id": trip_id,
73        "user_id": user_id,
74        "role": "organizer",
75        "budget": 0.0
76    }
77    return trips[trip_id]
78 
79@app.patch("/trips/{trip_id}")
80def update_trip(trip_id: int, name: Optional[str] = None, destination: Optional[str] = None, 
81                start_date: Optional[str] = None, end_date: Optional[str] = None, 
82                participant_id: Optional[int] = None, role: Optional[str] = None, 
83                budget: Optional[float] = None, authorization: Optional[str] = Header(None)):
84    user_id = get_current_user(authorization)
85    if trip_id not in trips:
86        raise HTTPException(status_code=404, detail="Trip not found")
87    trip = trips[trip_id]
88    if name is not None:
89        trip["name"] = name
90    if destination is not None:
91        trip["destination"] = destination
92    if start_date is not None:
93        trip["start_date"] = start_date
94    if end_date is not None:
95        trip["end_date"] = end_date
96    if participant_id is not None and participant_id in trip_participants and trip_participants[participant_id]["trip_id"] == trip_id:
97        if role is not None:
98            trip_participants[participant_id]["role"] = role
99        if budget is not None:
100            trip_participants[participant_id]["budget"] = budget
101    return trip

1fastapi
2uvicorn