Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, List
4from datetime import date, datetime
5import hashlib
6import secrets
7 
8app = FastAPI()
9 
10users = {}
11tokens = {}
12itineraries = {}
13activities = {}
14itinerary_id_counter = 1
15activity_id_counter = 1
16user_id_counter = 1
17 
18def get_current_user(authorization: Optional[str] = Header(None)):
19    if not authorization:
20        raise HTTPException(status_code=401, detail="Missing auth header")
21    token = authorization.replace("Bearer ", "")
22    for uid, t in tokens.items():
23        if t == token:
24            return uid
25    raise HTTPException(status_code=401, detail="Invalid token")
26 
27class SignupRequest(BaseModel):
28    username: str
29    password: str
30 
31class LoginRequest(BaseModel):
32    username: str
33    password: str
34 
35class ItineraryCreate(BaseModel):
36    destination: str
37    start_date: date
38    end_date: date
39 
40class ActivityCreate(BaseModel):
41    time: datetime
42    cost: float
43 
44@app.post("/signup")
45def signup(req: SignupRequest):
46    global user_id_counter
47    for u in users.values():
48        if u["username"] == req.username:
49            raise HTTPException(status_code=400, detail="Username taken")
50    uid = user_id_counter
51    user_id_counter += 1
52    users[uid] = {"id": uid, "username": req.username, "password": hashlib.sha256(req.password.encode()).hexdigest()}
53    return {"id": uid, "username": req.username}
54 
55@app.post("/login")
56def login(req: LoginRequest):
57    for uid, u in users.items():
58        if u["username"] == req.username and u["password"] == hashlib.sha256(req.password.encode()).hexdigest():
59            token = secrets.token_hex(32)
60            tokens[uid] = token
61            return {"token": token}
62    raise HTTPException(status_code=401, detail="Invalid credentials")
63 
64@app.get("/itineraries/{itinerary_id}")
65def get_itinerary(itinerary_id: int, authorization: Optional[str] = Header(None)):
66    uid = get_current_user(authorization)
67    if itinerary_id not in itineraries:
68        raise HTTPException(status_code=404, detail="Not found")
69    itin = itineraries[itinerary_id]
70    acts = [activities[a] for a in itin["activities"]]
71    return {"itinerary": itin, "activities": acts}
72 
73@app.post("/itineraries")
74def create_itinerary(req: ItineraryCreate, authorization: Optional[str] = Header(None)):
75    global itinerary_id_counter
76    uid = get_current_user(authorization)
77    itin_id = itinerary_id_counter
78    itinerary_id_counter += 1
79    itineraries[itin_id] = {
80        "id": itin_id,
81        "destination": req.destination,
82        "start_date": req.start_date.isoformat(),
83        "end_date": req.end_date.isoformat(),
84        "activities": [],
85        "user_id": uid
86    }
87    return itineraries[itin_id]
88 
89@app.get("/itineraries")
90def list_itineraries(authorization: Optional[str] = Header(None)):
91    uid = get_current_user(authorization)
92    user_itins = {k: v for k, v in itineraries.items() if v["user_id"] == uid}
93    result = []
94    for itin_id, itin in user_itins.items():
95        acts = [activities[a] for a in itin["activities"]]
96        result.append({"itinerary": itin, "activities": acts})
97    return result
98 
99@app.post("/itineraries/{itinerary_id}/activities")
100def add_activity(itinerary_id: int, req: ActivityCreate, authorization: Optional[str] = Header(None)):
101    global activity_id_counter
102    uid = get_current_user(authorization)
103    if itinerary_id not in itineraries:
104        raise HTTPException(status_code=404, detail="Itinerary not found")
105    if itineraries[itinerary_id]["user_id"] != uid:
106        raise HTTPException(status_code=403, detail="Not your itinerary")
107    act_id = activity_id_counter
108    activity_id_counter += 1
109    activities[act_id] = {
110        "id": act_id,
111        "time": req.time.isoformat(),
112        "cost": req.cost,
113        "itinerary_id": itinerary_id
114    }
115    itineraries[itinerary_id]["activities"].append(act_id)
116    return activities[act_id]

1fastapi
2uvicorn