Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from fastapi.responses import Response
3from pydantic import BaseModel
4import json
5import datetime
6from reportlab.pdfgen import canvas
7from io import BytesIO
8 
9app = FastAPI()
10 
11users = {}
12user_tokens = {}
13next_user_id = 1
14trips = {}
15next_trip_id = 1
16itineraries = {}
17next_itinerary_id = 1
18 
19class SignupRequest(BaseModel):
20    username: str
21    password: str
22 
23class LoginRequest(BaseModel):
24    username: str
25    password: str
26 
27class TripCreate(BaseModel):
28    name: str
29    destination: str
30    start_date: str
31    end_date: str
32 
33class ItineraryCreate(BaseModel):
34    trip_id: int
35    day: int
36    activity: str
37    location: str
38    notes: str = ""
39 
40def get_user_from_token(authorization: str = Header(None)):
41    if not authorization:
42        raise HTTPException(status_code=401, detail="Missing auth token")
43    token = authorization.replace("Bearer ", "")
44    for uid, t in user_tokens.items():
45        if t == token:
46            return uid
47    raise HTTPException(status_code=401, detail="Invalid token")
48 
49@app.post("/signup")
50def signup(req: SignupRequest):
51    global next_user_id
52    uid = next_user_id
53    next_user_id += 1
54    users[uid] = {"id": uid, "username": req.username, "password": req.password}
55    return {"id": uid, "username": req.username}
56 
57@app.post("/login")
58def login(req: LoginRequest):
59    for uid, u in users.items():
60        if u["username"] == req.username and u["password"] == req.password:
61            token = f"tok_{uid}_{datetime.datetime.now().timestamp()}"
62            user_tokens[uid] = token
63            return {"token": token}
64    raise HTTPException(status_code=401, detail="Invalid credentials")
65 
66@app.post("/trips")
67def create_trip(trip: TripCreate, authorization: str = Header(None)):
68    uid = get_user_from_token(authorization)
69    global next_trip_id
70    tid = next_trip_id
71    next_trip_id += 1
72    trips[tid] = {
73        "id": tid,
74        "user_id": uid,
75        "name": trip.name,
76        "destination": trip.destination,
77        "start_date": trip.start_date,
78        "end_date": trip.end_date
79    }
80    return trips[tid]
81 
82@app.get("/trips/{trip_id}")
83def get_trip(trip_id: int, authorization: str = Header(None)):
84    uid = get_user_from_token(authorization)
85    if trip_id not in trips:
86        raise HTTPException(status_code=404, detail="Trip not found")
87    return trips[trip_id]
88 
89@app.post("/itineraries")
90def create_itinerary(it: ItineraryCreate, authorization: str = Header(None)):
91    uid = get_user_from_token(authorization)
92    if it.trip_id not in trips:
93        raise HTTPException(status_code=404, detail="Trip not found")
94    global next_itinerary_id
95    iid = next_itinerary_id
96    next_itinerary_id += 1
97    itineraries[iid] = {
98        "id": iid,
99        "trip_id": it.trip_id,
100        "day": it.day,
101        "activity": it.activity,
102        "location": it.location,
103        "notes": it.notes
104    }
105    return itineraries[iid]
106 
107@app.get("/itineraries/{trip_id}")
108def get_itineraries(trip_id: int, authorization: str = Header(None)):
109    uid = get_user_from_token(authorization)
110    if trip_id not in trips:
111        raise HTTPException(status_code=404, detail="Trip not found")
112    trip_itineraries = [v for v in itineraries.values() if v["trip_id"] == trip_id]
113    return trip_itineraries
114 
115@app.get("/itineraries/{trip_id}/export")
116def export_itinerary(trip_id: int, format: str = "json", authorization: str = Header(None)):
117    uid = get_user_from_token(authorization)
118    if trip_id not in trips:
119        raise HTTPException(status_code=404, detail="Trip not found")
120    trip = trips[trip_id]
121    trip_itineraries = [v for v in itineraries.values() if v["trip_id"] == trip_id]
122 
123    if format == "json":
124        data = {"trip": trip, "itineraries": trip_itineraries}
125        return Response(content=json.dumps(data, indent=2), media_type="application/json")
126 
127    elif format == "pdf":
128        buffer = BytesIO()
129        p = canvas.Canvas(buffer)
130        p.drawString(100, 800, f"Trip: {trip['name']}")
131        p.drawString(100, 780, f"Destination: {trip['destination']}")
132        p.drawString(100, 760, f"Dates: {trip['start_date']} - {trip['end_date']}")
133        y = 730
134        for it in trip_itineraries:
135            p.drawString(100, y, f"Day {it['day']}: {it['activity']} @ {it['location']}")
136            if it["notes"]:
137                p.drawString(120, y - 15, f"Notes: {it['notes']}")
138                y -= 30
139            else:
140                y -= 20
141        p.save()
142        buffer.seek(0)
143        return Response(content=buffer.read(), media_type="application/pdf", headers={"Content-Disposition": f"attachment; filename=itinerary_{trip_id}.pdf"})
144 
145    raise HTTPException(status_code=400, detail="Invalid format")

1fastapi
2uvicorn
3reportlab