Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import hashlib
4import random
5import string
6 
7app = FastAPI()
8 
9users = {}
10walkers = {}
11orders = {}
12tokens = {}
13walker_id_counter = 1
14order_id_counter = 1
15user_id_counter = 1
16 
17def generate_token():
18    return ''.join(random.choices(string.ascii_letters + string.digits, k=32))
19 
20def get_current_user(authorization: str = Header(None)):
21    if not authorization:
22        raise HTTPException(status_code=401, detail="Missing token")
23    token = authorization.replace("Bearer ", "")
24    user_id = tokens.get(token)
25    if not user_id:
26        raise HTTPException(status_code=401, detail="Invalid token")
27    return user_id
28 
29@app.post("/signup")
30def signup(username: str, password: str):
31    global user_id_counter
32    if any(u["username"] == username for u in users.values()):
33        raise HTTPException(status_code=400, detail="Username taken")
34    user_id = user_id_counter
35    user_id_counter += 1
36    password_hash = hashlib.sha256(password.encode()).hexdigest()
37    users[user_id] = {"id": user_id, "username": username, "password_hash": password_hash}
38    token = generate_token()
39    tokens[token] = user_id
40    return {"user_id": user_id, "token": token}
41 
42@app.post("/login")
43def login(username: str, password: str):
44    for user_id, user in users.items():
45        if user["username"] == username and user["password_hash"] == hashlib.sha256(password.encode()).hexdigest():
46            token = generate_token()
47            tokens[token] = user_id
48            return {"token": token}
49    raise HTTPException(status_code=401, detail="Invalid credentials")
50 
51@app.get("/walkers/{walker_id}")
52def get_walker(walker_id: int, authorization: str = Header(None)):
53    get_current_user(authorization)
54    walker = walkers.get(walker_id)
55    if not walker:
56        raise HTTPException(status_code=404, detail="Walker not found")
57    return walker
58 
59@app.post("/walkers")
60def create_walker(name: str, zip_code: str, available_today: bool, rating: float, price_per_walk: float, insured: bool, authorization: str = Header(None)):
61    global walker_id_counter
62    get_current_user(authorization)
63    walker_id = walker_id_counter
64    walker_id_counter += 1
65    walkers[walker_id] = {
66        "id": walker_id,
67        "name": name,
68        "zip_code": zip_code,
69        "available_today": available_today,
70        "rating": rating,
71        "price_per_walk": price_per_walk,
72        "insured": insured,
73        "walk_count": 0
74    }
75    return walkers[walker_id]
76 
77@app.get("/walkers")
78def list_walkers(zip: Optional[str] = None, available_today: Optional[bool] = None, rating_min: Optional[float] = None, max_price_per_walk: Optional[float] = None, authorization: str = Header(None)):
79    get_current_user(authorization)
80    result = []
81    for w in walkers.values():
82        if zip and w["zip_code"] != zip:
83            continue
84        if available_today is not None and w["available_today"] != available_today:
85            continue
86        if rating_min is not None and w["rating"] < rating_min:
87            continue
88        if max_price_per_walk is not None and w["price_per_walk"] > max_price_per_walk:
89            continue
90        result.append({
91            "id": w["id"],
92            "name": w["name"],
93            "zip_code": w["zip_code"],
94            "available_today": w["available_today"],
95            "rating": w["rating"],
96            "price_per_walk": w["price_per_walk"],
97            "walk_count": w["walk_count"],
98            "insured": w["insured"]
99        })
100    return result
101 
102@app.get("/orders/{order_id}")
103def get_order(order_id: int, authorization: str = Header(None)):
104    get_current_user(authorization)
105    order = orders.get(order_id)
106    if not order:
107        raise HTTPException(status_code=404, detail="Order not found")
108    return order
109 
110@app.post("/orders")
111def create_order(walker_id: int, walker_name: str, owner_name: str, dog_name: str, scheduled_time: str, authorization: str = Header(None)):
112    global order_id_counter
113    user_id = get_current_user(authorization)
114    if walker_id not in walkers:
115        raise HTTPException(status_code=404, detail="Walker not found")
116    order_id = order_id_counter
117    order_id_counter += 1
118    orders[order_id] = {
119        "id": order_id,
120        "walker_id": walker_id,
121        "walker_name": walker_name,
122        "owner_name": owner_name,
123        "dog_name": dog_name,
124        "scheduled_time": scheduled_time,
125        "user_id": user_id
126    }
127    walkers[walker_id]["walk_count"] += 1
128    return orders[order_id]

1fastapi
2uvicorn