Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, List
4import secrets
5import uvicorn
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11trips = {}
12trip_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 TripCreate(BaseModel):
23    name: str
24    destination: str
25    start_date: str
26    end_date: str
27 
28class ParticipantUpdate(BaseModel):
29    role: Optional[str] = None
30    budget: Optional[float] = None
31 
32class TripUpdate(BaseModel):
33    name: Optional[str] = None
34    destination: Optional[str] = None
35    start_date: Optional[str] = None
36    end_date: Optional[str] = None
37    participants: Optional[List[ParticipantUpdate]] = None
38 
39def get_current_user(authorization: str = Header(None)):
40    if not authorization:
41        raise HTTPException(status_code=401, detail="Missing auth header")
42    token = authorization.replace("Bearer ", "")
43    if token not in tokens:
44        raise HTTPException(status_code=401, detail="Invalid token")
45    return tokens[token]
46 
47@app.post("/signup")
48def signup(req: SignupRequest):
49    if req.username in users:
50        raise HTTPException(status_code=400, detail="User exists")
51    users[req.username] = {"password": req.password, "id": len(users) + 1}
52    token = secrets.token_hex(16)
53    tokens[token] = req.username
54    return {"token": token, "user_id": users[req.username]["id"]}
55 
56@app.post("/login")
57def login(req: LoginRequest):
58    if req.username not in users or users[req.username]["password"] != req.password:
59        raise HTTPException(status_code=401, detail="Invalid credentials")
60    token = secrets.token_hex(16)
61    tokens[token] = req.username
62    return {"token": token}
63 
64@app.get("/trips/{trip_id}")
65def get_trip(trip_id: int, authorization: str = Header(None)):
66    user = get_current_user(authorization)
67    if trip_id not in trips:
68        raise HTTPException(status_code=404, detail="Trip not found")
69    return trips[trip_id]
70 
71@app.post("/trips")
72def create_trip(trip: TripCreate, authorization: str = Header(None)):
73    user = get_current_user(authorization)
74    global trip_id_counter
75    trip_id = trip_id_counter
76    trip_id_counter += 1
77    trips[trip_id] = {
78        "id": trip_id,
79        "name": trip.name,
80        "destination": trip.destination,
81        "start_date": trip.start_date,
82        "end_date": trip.end_date,
83        "participants": []
84    }
85    return trips[trip_id]
86 
87@app.patch("/trips/{trip_id}")
88def update_trip(trip_id: int, update: TripUpdate, authorization: str = Header(None)):
89    user = get_current_user(authorization)
90    if trip_id not in trips:
91        raise HTTPException(status_code=404, detail="Trip not found")
92    trip = trips[trip_id]
93    if update.name is not None:
94        trip["name"] = update.name
95    if update.destination is not None:
96        trip["destination"] = update.destination
97    if update.start_date is not None:
98        trip["start_date"] = update.start_date
99    if update.end_date is not None:
100        trip["end_date"] = update.end_date
101    if update.participants is not None:
102        for p in update.participants:
103            if p.role is not None:
104                trip["participants"].append({"role": p.role, "budget": p.budget})
105            elif p.budget is not None:
106                trip["participants"].append({"role": "member", "budget": p.budget})
107            else:
108                trip["participants"].append({"role": "member", "budget": 0.0})
109    return trip

1fastapi
2uvicorn
3pydantic