Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from typing import Optional
3import secrets
4 
5app = FastAPI()
6 
7users = {}
8tokens = {}
9pets = {}
10sittings = {}
11sitters = {}
12preferred_sitters = {}
13availabilities = {}
14bookings = {}
15 
16next_user_id = 1
17next_pet_id = 1
18next_sitting_id = 1
19next_sitter_id = 1
20next_availability_id = 1
21next_booking_id = 1
22 
23def get_user_from_token(authorization: Optional[str] = Header(None)):
24    if not authorization:
25        raise HTTPException(status_code=401, detail="Missing Authorization header")
26    token = authorization.replace("Bearer ", "")
27    for uid, t in tokens.items():
28        if t == token:
29            return users[uid]
30    raise HTTPException(status_code=401, detail="Invalid token")
31 
32@app.post("/signup")
33def signup(username: str, password: str):
34    global next_user_id
35    for u in users.values():
36        if u["username"] == username:
37            raise HTTPException(status_code=400, detail="Username taken")
38    uid = next_user_id
39    next_user_id += 1
40    users[uid] = {"id": uid, "username": username, "password": password}
41    return {"id": uid, "username": username}
42 
43@app.post("/login")
44def login(username: str, password: str):
45    for uid, u in users.items():
46        if u["username"] == username and u["password"] == password:
47            token = secrets.token_hex(16)
48            tokens[uid] = token
49            return {"token": token, "user_id": uid}
50    raise HTTPException(status_code=401, detail="Invalid credentials")
51 
52@app.post("/pets")
53def create_pet(name: str, species: str, owner_id: int, authorization: Optional[str] = Header(None)):
54    user = get_user_from_token(authorization)
55    global next_pet_id
56    pid = next_pet_id
57    next_pet_id += 1
58    pets[pid] = {"id": pid, "name": name, "species": species, "owner_id": owner_id}
59    return pets[pid]
60 
61@app.get("/pets/{pet_id}")
62def get_pet(pet_id: int, authorization: Optional[str] = Header(None)):
63    user = get_user_from_token(authorization)
64    if pet_id not in pets:
65        raise HTTPException(status_code=404, detail="Pet not found")
66    return pets[pet_id]
67 
68@app.post("/sitters")
69def create_sitter(name: str, authorization: Optional[str] = Header(None)):
70    user = get_user_from_token(authorization)
71    global next_sitter_id
72    sid = next_sitter_id
73    next_sitter_id += 1
74    sitters[sid] = {"id": sid, "name": name, "user_id": user["id"]}
75    return sitters[sid]
76 
77@app.get("/sitters/{sitter_id}")
78def get_sitter(sitter_id: int, authorization: Optional[str] = Header(None)):
79    user = get_user_from_token(authorization)
80    if sitter_id not in sitters:
81        raise HTTPException(status_code=404, detail="Sitter not found")
82    return sitters[sitter_id]
83 
84@app.post("/sittings")
85def create_sitting(pet_id: int, sitter_id: int, start_time: str, end_time: str, authorization: Optional[str] = Header(None)):
86    user = get_user_from_token(authorization)
87    global next_sitting_id
88    sid = next_sitting_id
89    next_sitting_id += 1
90    sittings[sid] = {
91        "id": sid,
92        "pet_id": pet_id,
93        "sitter_id": sitter_id,
94        "start_time": start_time,
95        "end_time": end_time,
96        "owner_id": user["id"]
97    }
98    return sittings[sid]
99 
100@app.get("/sittings/{sitting_id}")
101def get_sitting(sitting_id: int, authorization: Optional[str] = Header(None)):
102    user = get_user_from_token(authorization)
103    if sitting_id not in sittings:
104        raise HTTPException(status_code=404, detail="Sitting not found")
105    return sittings[sitting_id]
106 
107@app.post("/preferred_sitters")
108def set_preferred_sitter(sitter_id: int, owner_id: int, authorization: Optional[str] = Header(None)):
109    user = get_user_from_token(authorization)
110    if sitter_id not in sitters:
111        raise HTTPException(status_code=404, detail="Sitter not found")
112    if owner_id != user["id"]:
113        raise HTTPException(status_code=403, detail="You can only set preferred sitters for yourself")
114    preferred_sitters[owner_id] = sitter_id
115    return {"owner_id": owner_id, "preferred_sitter_id": sitter_id}
116 
117@app.get("/preferred_sitters/{owner_id}")
118def get_preferred_sitter(owner_id: int, authorization: Optional[str] = Header(None)):
119    user = get_user_from_token(authorization)
120    if owner_id not in preferred_sitters:
121        raise HTTPException(status_code=404, detail="No preferred sitter set")
122    return {"owner_id": owner_id, "preferred_sitter_id": preferred_sitters[owner_id]}
123 
124@app.post("/availabilities")
125def set_availability(sitter_id: int, day_of_week: str, start_time: str, end_time: str, authorization: Optional[str] = Header(None)):
126    user = get_user_from_token(authorization)
127    if sitter_id not in sitters:
128        raise HTTPException(status_code=404, detail="Sitter not found")
129    if sitters[sitter_id]["user_id"] != user["id"]:
130        raise HTTPException(status_code=403, detail="You can only set your own availability")
131    global next_availability_id
132    aid = next_availability_id
133    next_availability_id += 1
134    availabilities[aid] = {
135        "id": aid,
136        "sitter_id": sitter_id,
137        "day_of_week": day_of_week,
138        "start_time": start_time,
139        "end_time": end_time
140    }
141    return availabilities[aid]
142 
143@app.get("/availabilities/{availability_id}")
144def get_availability(availability_id: int, authorization: Optional[str] = Header(None)):
145    user = get_user_from_token(authorization)
146    if availability_id not in availabilities:
147        raise HTTPException(status_code=404, detail="Availability not found")
148    return availabilities[availability_id]
149 
150@app.post("/bookings")
151def create_booking(pet_id: int, sitter_id: int, start_time: str, end_time: str, authorization: Optional[str] = Header(None)):
152    user = get_user_from_token(authorization)
153    global next_booking_id
154    bid = next_booking_id
155    next_booking_id += 1
156    bookings[bid] = {
157        "id": bid,
158        "pet_id": pet_id,
159        "sitter_id": sitter_id,
160        "start_time": start_time,
161        "end_time": end_time,
162        "owner_id": user["id"]
163    }
164    return bookings[bid]
165 
166@app.get("/bookings/{booking_id}")
167def get_booking(booking_id: int, authorization: Optional[str] = Header(None)):
168    user = get_user_from_token(authorization)
169    if booking_id not in bookings:
170        raise HTTPException(status_code=404, detail="Booking not found")
171    return bookings[booking_id]
172 
173@app.get("/users/{user_id}")
174def get_user(user_id: int, authorization: Optional[str] = Header(None)):
175    user = get_user_from_token(authorization)
176    if user_id not in users:
177        raise HTTPException(status_code=404, detail="User not found")
178    return {"id": users[user_id]["id"], "username": users[user_id]["username"]}

1fastapi
2uvicorn