Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import random
5import string
6import time
7from datetime import datetime
8 
9app = FastAPI()
10 
11users = {}
12tokens = {}
13trails = {}
14checkins = {}
15next_user_id = 1
16next_trail_id = 1
17next_checkin_id = 1
18 
19def generate_token():
20    return ''.join(random.choices(string.ascii_letters + string.digits, k=32))
21 
22def get_user_from_token(authorization: str = Header(None)):
23    if not authorization:
24        raise HTTPException(status_code=401, detail="Missing authorization header")
25    token = authorization.replace("Bearer ", "")
26    if token not in tokens:
27        raise HTTPException(status_code=401, detail="Invalid token")
28    return tokens[token]
29 
30class SignupRequest(BaseModel):
31    username: str
32    password: str
33 
34class LoginRequest(BaseModel):
35    username: str
36    password: str
37 
38class TrailCreate(BaseModel):
39    name: str
40    difficulty: str
41    elevation_gain: int
42    description: Optional[str] = None
43 
44class CheckinCreate(BaseModel):
45    trail_id: int
46    photos: list[str] = []
47    conditions: Optional[str] = None
48 
49@app.post("/signup")
50def signup(req: SignupRequest):
51    global next_user_id
52    user_id = next_user_id
53    next_user_id += 1
54    users[user_id] = {"id": user_id, "username": req.username, "password": req.password}
55    return {"id": user_id, "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 = generate_token()
62            tokens[token] = uid
63            return {"token": token}
64    raise HTTPException(status_code=401, detail="Invalid credentials")
65 
66@app.get("/trails/nearby")
67def get_trails_nearby():
68    result = []
69    for tid, t in trails.items():
70        recent_reviews = []
71        for cid, c in checkins.items():
72            if c["trail_id"] == tid:
73                recent_reviews.append({
74                    "id": cid,
75                    "conditions": c["conditions"],
76                    "photos": c["photos"],
77                    "timestamp": c["timestamp"]
78                })
79        result.append({
80            "id": tid,
81            "name": t["name"],
82            "difficulty": t["difficulty"],
83            "elevation_gain": t["elevation_gain"],
84            "description": t.get("description"),
85            "recent_reviews": recent_reviews[-5:]
86        })
87    return result
88 
89@app.post("/trails/checkin")
90def checkin_trail(req: CheckinCreate, authorization: str = Header(None)):
91    user_id = get_user_from_token(authorization)
92    if req.trail_id not in trails:
93        raise HTTPException(status_code=404, detail="Trail not found")
94    global next_checkin_id
95    checkin_id = next_checkin_id
96    next_checkin_id += 1
97    checkins[checkin_id] = {
98        "id": checkin_id,
99        "trail_id": req.trail_id,
100        "user_id": user_id,
101        "photos": req.photos,
102        "conditions": req.conditions,
103        "timestamp": datetime.now().isoformat()
104    }
105    return {"id": checkin_id}
106 
107@app.get("/trails/{trail_id}")
108def get_trail(trail_id: int):
109    if trail_id not in trails:
110        raise HTTPException(status_code=404, detail="Trail not found")
111    t = trails[trail_id]
112    trail_reviews = []
113    for cid, c in checkins.items():
114        if c["trail_id"] == trail_id:
115            trail_reviews.append({
116                "id": cid,
117                "conditions": c["conditions"],
118                "photos": c["photos"],
119                "timestamp": c["timestamp"]
120            })
121    return {
122        "id": trail_id,
123        "name": t["name"],
124        "difficulty": t["difficulty"],
125        "elevation_gain": t["elevation_gain"],
126        "description": t.get("description"),
127        "reviews": trail_reviews
128    }
129 
130@app.post("/trails")
131def create_trail(req: TrailCreate, authorization: str = Header(None)):
132    get_user_from_token(authorization)
133    global next_trail_id
134    trail_id = next_trail_id
135    next_trail_id += 1
136    trails[trail_id] = {
137        "id": trail_id,
138        "name": req.name,
139        "difficulty": req.difficulty,
140        "elevation_gain": req.elevation_gain,
141        "description": req.description
142    }
143    return {"id": trail_id}
144 
145@app.get("/checkins/{checkin_id}")
146def get_checkin(checkin_id: int):
147    if checkin_id not in checkins:
148        raise HTTPException(status_code=404, detail="Checkin not found")
149    return checkins[checkin_id]

1fastapi
2uvicorn