Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, List
4from datetime import date
5import secrets
6 
7app = FastAPI()
8 
9# In-memory stores
10users = {}
11tokens = {}
12sitters = {}
13bookings = {}
14sitter_id_counter = 1
15booking_id_counter = 1
16user_id_counter = 1
17 
18# Auth helpers
19def get_user_from_token(authorization: str = Header(None)):
20    if not authorization:
21        raise HTTPException(status_code=401, detail="Missing auth header")
22    token = authorization.replace("Bearer ", "")
23    user_id = tokens.get(token)
24    if not user_id:
25        raise HTTPException(status_code=401, detail="Invalid token")
26    return user_id
27 
28# Models
29class SignupRequest(BaseModel):
30    username: str
31    password: str
32 
33class LoginRequest(BaseModel):
34    username: str
35    password: str
36 
37class SitterCreate(BaseModel):
38    name: str
39    hourly_rate: float
40    distance: float
41    pet_type_preferences: List[str]
42 
43class BookingCreate(BaseModel):
44    sitter_id: int
45    start_date: date
46    end_date: date
47    pet_details: str
48    special_instructions: Optional[str] = ""
49 
50# Endpoints
51@app.post("/signup")
52def signup(req: SignupRequest):
53    global user_id_counter
54    user_id = user_id_counter
55    user_id_counter += 1
56    users[user_id] = {"username": req.username, "password": req.password}
57    return {"user_id": user_id, "message": "User created"}
58 
59@app.post("/login")
60def login(req: LoginRequest):
61    for uid, u in users.items():
62        if u["username"] == req.username and u["password"] == req.password:
63            token = secrets.token_hex(16)
64            tokens[token] = uid
65            return {"token": token}
66    raise HTTPException(status_code=401, detail="Invalid credentials")
67 
68@app.get("/sitters")
69def get_sitters():
70    return list(sitters.values())
71 
72@app.get("/sitters/{sitter_id}")
73def get_sitter(sitter_id: int):
74    sitter = sitters.get(sitter_id)
75    if not sitter:
76        raise HTTPException(status_code=404, detail="Sitter not found")
77    return sitter
78 
79@app.post("/sitters")
80def create_sitter(sitter: SitterCreate, user_id: int = Header(None)):
81    global sitter_id_counter
82    sitter_id = sitter_id_counter
83    sitter_id_counter += 1
84    sitters[sitter_id] = {
85        "id": sitter_id,
86        "name": sitter.name,
87        "hourly_rate": sitter.hourly_rate,
88        "distance": sitter.distance,
89        "pet_type_preferences": sitter.pet_type_preferences
90    }
91    return {"sitter_id": sitter_id}
92 
93@app.get("/bookings/{booking_id}")
94def get_booking(booking_id: int, user_id: int = Header(None)):
95    booking = bookings.get(booking_id)
96    if not booking:
97        raise HTTPException(status_code=404, detail="Booking not found")
98    return booking
99 
100@app.post("/booking")
101def create_booking(booking: BookingCreate, user_id: int = Header(None)):
102    global booking_id_counter
103    booking_id = booking_id_counter
104    booking_id_counter += 1
105    bookings[booking_id] = {
106        "id": booking_id,
107        "sitter_id": booking.sitter_id,
108        "start_date": booking.start_date.isoformat(),
109        "end_date": booking.end_date.isoformat(),
110        "pet_details": booking.pet_details,
111        "special_instructions": booking.special_instructions
112    }
113    return {"booking_id": booking_id}

1fastapi
2uvicorn