Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import secrets
5 
6app = FastAPI()
7 
8# In-memory stores
9users = {}
10tokens = {}
11trails = {}
12trail_id_counter = 1
13 
14# Auth helpers
15def get_current_user(authorization: Optional[str] = Header(None)):
16    if not authorization:
17        raise HTTPException(status_code=401, detail="Missing auth token")
18    token = authorization.replace("Bearer ", "")
19    if token not in tokens:
20        raise HTTPException(status_code=401, detail="Invalid auth token")
21    return tokens[token]
22 
23# User models
24class SignupRequest(BaseModel):
25    username: str
26    password: str
27 
28class LoginRequest(BaseModel):
29    username: str
30    password: str
31 
32# Trail models
33class TrailCreate(BaseModel):
34    name: str
35    difficulty: str
36    length: float
37    is_dog_friendly: Optional[bool] = False
38    region: Optional[str] = None
39 
40class TrailUpdate(BaseModel):
41    name: Optional[str] = None
42    difficulty: Optional[str] = None
43    length: Optional[float] = None
44    is_dog_friendly: Optional[bool] = None
45    region: Optional[str] = None
46 
47# Auth endpoints
48@app.post("/signup")
49def signup(req: SignupRequest):
50    if req.username in users:
51        raise HTTPException(status_code=400, detail="Username already exists")
52    users[req.username] = req.password
53    token = secrets.token_hex(16)
54    tokens[token] = req.username
55    return {"token": token}
56 
57@app.post("/login")
58def login(req: LoginRequest):
59    if req.username not in users or users[req.username] != req.password:
60        raise HTTPException(status_code=401, detail="Invalid credentials")
61    token = secrets.token_hex(16)
62    tokens[token] = req.username
63    return {"token": token}
64 
65# Trail endpoints
66@app.post("/trails")
67def create_trail(trail: TrailCreate, authorization: Optional[str] = Header(None)):
68    get_current_user(authorization)
69    global trail_id_counter
70    trail_id = trail_id_counter
71    trail_id_counter += 1
72    trails[trail_id] = {
73        "id": trail_id,
74        "name": trail.name,
75        "difficulty": trail.difficulty,
76        "length": trail.length,
77        "is_dog_friendly": trail.is_dog_friendly,
78        "region": trail.region,
79    }
80    return trails[trail_id]
81 
82@app.get("/trails/{trail_id}")
83def get_trail(trail_id: int, authorization: Optional[str] = Header(None)):
84    get_current_user(authorization)
85    if trail_id not in trails:
86        raise HTTPException(status_code=404, detail="Trail not found")
87    return trails[trail_id]
88 
89@app.patch("/trails/{trail_id}")
90def update_trail(trail_id: int, trail: TrailUpdate, authorization: Optional[str] = Header(None)):
91    get_current_user(authorization)
92    if trail_id not in trails:
93        raise HTTPException(status_code=404, detail="Trail not found")
94    existing = trails[trail_id]
95    if trail.name is not None:
96        existing["name"] = trail.name
97    if trail.difficulty is not None:
98        existing["difficulty"] = trail.difficulty
99    if trail.length is not None:
100        existing["length"] = trail.length
101    if trail.is_dog_friendly is not None:
102        existing["is_dog_friendly"] = trail.is_dog_friendly
103    if trail.region is not None:
104        existing["region"] = trail.region
105    return existing

1fastapi
2uvicorn