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 = {}
11activities = {}
12notes = {}
13budgets = {}
14 
15user_id_counter = 1
16trip_id_counter = 1
17activity_id_counter = 1
18note_id_counter = 1
19budget_id_counter = 1
20 
21class SignupRequest(BaseModel):
22    username: str
23    password: str
24 
25class LoginRequest(BaseModel):
26    username: str
27    password: str
28 
29class TripCreate(BaseModel):
30    name: str
31    destination: str
32    start_date: str
33    end_date: str
34 
35class ActivityCreate(BaseModel):
36    trip_id: int
37    day: int
38    description: str
39    time: Optional[str] = None
40    location: Optional[str] = None
41 
42class NoteCreate(BaseModel):
43    trip_id: int
44    day: int
45    content: str
46 
47class BudgetCreate(BaseModel):
48    trip_id: int
49    day: int
50    category: str
51    amount: float
52    description: Optional[str] = None
53 
54def get_current_user(authorization: str = Header(None)):
55    if not authorization:
56        raise HTTPException(status_code=401, detail="Missing authorization header")
57    token = authorization.replace("Bearer ", "")
58    if token not in tokens:
59        raise HTTPException(status_code=401, detail="Invalid token")
60    return tokens[token]
61 
62@app.post("/signup")
63def signup(req: SignupRequest):
64    global user_id_counter
65    for u in users.values():
66        if u["username"] == req.username:
67            raise HTTPException(status_code=400, detail="Username already exists")
68    user_id = user_id_counter
69    user_id_counter += 1
70    users[user_id] = {"id": user_id, "username": req.username, "password": req.password}
71    token = secrets.token_hex(16)
72    tokens[token] = user_id
73    return {"user_id": user_id, "token": token}
74 
75@app.post("/login")
76def login(req: LoginRequest):
77    for uid, u in users.items():
78        if u["username"] == req.username and u["password"] == req.password:
79            token = secrets.token_hex(16)
80            tokens[token] = uid
81            return {"user_id": uid, "token": token}
82    raise HTTPException(status_code=401, detail="Invalid credentials")
83 
84@app.post("/trip")
85def create_trip(trip: TripCreate, authorization: str = Header(None)):
86    user_id = get_current_user(authorization)
87    global trip_id_counter
88    trip_id = trip_id_counter
89    trip_id_counter += 1
90    trips[trip_id] = {
91        "id": trip_id,
92        "name": trip.name,
93        "destination": trip.destination,
94        "start_date": trip.start_date,
95        "end_date": trip.end_date,
96        "organizer_id": user_id,
97        "co_organizers": []
98    }
99    return trips[trip_id]
100 
101@app.get("/trip/{trip_id}")
102def get_trip(trip_id: int, authorization: str = Header(None)):
103    get_current_user(authorization)
104    if trip_id not in trips:
105        raise HTTPException(status_code=404, detail="Trip not found")
106    trip = trips[trip_id].copy()
107    trip["activities"] = [a for a in activities.values() if a["trip_id"] == trip_id]
108    trip["notes"] = [n for n in notes.values() if n["trip_id"] == trip_id]
109    trip["budgets"] = [b for b in budgets.values() if b["trip_id"] == trip_id]
110    return trip
111 
112@app.post("/trip/{trip_id}/promote")
113def promote_to_co_organizer(trip_id: int, user_id: int, authorization: str = Header(None)):
114    current_user_id = get_current_user(authorization)
115    if trip_id not in trips:
116        raise HTTPException(status_code=404, detail="Trip not found")
117    trip = trips[trip_id]
118    if trip["organizer_id"] != current_user_id:
119        raise HTTPException(status_code=403, detail="Only the organizer can promote")
120    if user_id not in users:
121        raise HTTPException(status_code=404, detail="User not found")
122    if user_id not in trip["co_organizers"]:
123        trip["co_organizers"].append(user_id)
124    return {"message": f"User {user_id} promoted to co-organizer"}
125 
126@app.post("/activity")
127def create_activity(activity: ActivityCreate, authorization: str = Header(None)):
128    get_current_user(authorization)
129    if activity.trip_id not in trips:
130        raise HTTPException(status_code=404, detail="Trip not found")
131    global activity_id_counter
132    activity_id = activity_id_counter
133    activity_id_counter += 1
134    activities[activity_id] = {
135        "id": activity_id,
136        "trip_id": activity.trip_id,
137        "day": activity.day,
138        "description": activity.description,
139        "time": activity.time,
140        "location": activity.location
141    }
142    return activities[activity_id]
143 
144@app.get("/activity/{activity_id}")
145def get_activity(activity_id: int, authorization: str = Header(None)):
146    get_current_user(authorization)
147    if activity_id not in activities:
148        raise HTTPException(status_code=404, detail="Activity not found")
149    return activities[activity_id]
150 
151@app.post("/note")
152def create_note(note: NoteCreate, authorization: str = Header(None)):
153    get_current_user(authorization)
154    if note.trip_id not in trips:
155        raise HTTPException(status_code=404, detail="Trip not found")
156    global note_id_counter
157    note_id = note_id_counter
158    note_id_counter += 1
159    notes[note_id] = {
160        "id": note_id,
161        "trip_id": note.trip_id,
162        "day": note.day,
163        "content": note.content
164    }
165    return notes[note_id]
166 
167@app.get("/note/{note_id}")
168def get_note(note_id: int, authorization: str = Header(None)):
169    get_current_user(authorization)
170    if note_id not in notes:
171        raise HTTPException(status_code=404, detail="Note not found")
172    return notes[note_id]
173 
174@app.post("/budget")
175def create_budget(budget: BudgetCreate, authorization: str = Header(None)):
176    get_current_user(authorization)
177    if budget.trip_id not in trips:
178        raise HTTPException(status_code=404, detail="Trip not found")
179    global budget_id_counter
180    budget_id = budget_id_counter
181    budget_id_counter += 1
182    budgets[budget_id] = {
183        "id": budget_id,
184        "trip_id": budget.trip_id,
185        "day": budget.day,
186        "category": budget.category,
187        "amount": budget.amount,
188        "description": budget.description
189    }
190    return budgets[budget_id]
191 
192@app.get("/budget/{budget_id}")
193def get_budget(budget_id: int, authorization: str = Header(None)):
194    get_current_user(authorization)
195    if budget_id not in budgets:
196        raise HTTPException(status_code=404, detail="Budget not found")
197    return budgets[budget_id]

1fastapi
2uvicorn