Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, Dict, List
4import secrets
5import datetime
6 
7app = FastAPI()
8 
9# In-memory stores
10users: Dict[int, dict] = {}
11tokens: Dict[str, int] = {}  # token -> user_id
12profiles: Dict[int, dict] = {}
13pets: Dict[int, dict] = {}
14next_user_id = 1
15next_profile_id = 1
16next_pet_id = 1
17 
18class SignupRequest(BaseModel):
19    username: str
20    password: str
21 
22class LoginRequest(BaseModel):
23    username: str
24    password: str
25 
26class ProfileCreate(BaseModel):
27    name: str
28    bio: Optional[str] = ""
29 
30class PetCreate(BaseModel):
31    name: str
32    feeding_schedule: str
33    stay_duration_days: int
34    photo_url: Optional[str] = ""
35 
36def get_user_id_from_token(authorization: str = Header(None)):
37    if not authorization:
38        raise HTTPException(status_code=401, detail="Missing auth token")
39    token = authorization.replace("Bearer ", "")
40    user_id = tokens.get(token)
41    if user_id is None:
42        raise HTTPException(status_code=401, detail="Invalid token")
43    return user_id
44 
45@app.post("/signup")
46def signup(req: SignupRequest):
47    global next_user_id
48    for u in users.values():
49        if u["username"] == req.username:
50            raise HTTPException(status_code=400, detail="Username taken")
51    user_id = next_user_id
52    next_user_id += 1
53    users[user_id] = {
54        "id": user_id,
55        "username": req.username,
56        "password": req.password  # plaintext because we ship fast
57    }
58    return {"user_id": user_id, "message": "User created"}
59 
60@app.post("/login")
61def login(req: LoginRequest):
62    for u in users.values():
63        if u["username"] == req.username and u["password"] == req.password:
64            token = secrets.token_hex(16)
65            tokens[token] = u["id"]
66            return {"token": token}
67    raise HTTPException(status_code=401, detail="Invalid credentials")
68 
69@app.post("/profile")
70def create_profile(req: ProfileCreate, authorization: str = Header(None)):
71    global next_profile_id
72    user_id = get_user_id_from_token(authorization)
73    # First sitter becomes host
74    if not profiles:
75        role = "host"
76    else:
77        role = "sitter"
78    profile_id = next_profile_id
79    next_profile_id += 1
80    profiles[profile_id] = {
81        "id": profile_id,
82        "user_id": user_id,
83        "name": req.name,
84        "bio": req.bio,
85        "role": role,
86        "co_hosts": []
87    }
88    return {"profile_id": profile_id, "role": role}
89 
90@app.get("/profile/{profile_id}")
91def get_profile(profile_id: int, authorization: str = Header(None)):
92    get_user_id_from_token(authorization)
93    profile = profiles.get(profile_id)
94    if not profile:
95        raise HTTPException(status_code=404, detail="Profile not found")
96    return profile
97 
98@app.post("/profile/{profile_id}/promote")
99def promote_to_co_host(profile_id: int, target_user_id: int, authorization: str = Header(None)):
100    user_id = get_user_id_from_token(authorization)
101    profile = profiles.get(profile_id)
102    if not profile:
103        raise HTTPException(status_code=404, detail="Profile not found")
104    if profile["user_id"] != user_id:
105        raise HTTPException(status_code=403, detail="Only the host can promote")
106    if profile["role"] != "host":
107        raise HTTPException(status_code=400, detail="Only a host profile can promote")
108    # Find target user's profile
109    target_profile = None
110    for p in profiles.values():
111        if p["user_id"] == target_user_id:
112            target_profile = p
113            break
114    if not target_profile:
115        raise HTTPException(status_code=404, detail="Target user has no profile")
116    if target_user_id not in profile["co_hosts"]:
117        profile["co_hosts"].append(target_user_id)
118    target_profile["role"] = "co_host"
119    return {"message": f"User {target_user_id} promoted to co-host"}
120 
121@app.post("/pet")
122def create_pet(req: PetCreate, authorization: str = Header(None)):
123    global next_pet_id
124    user_id = get_user_id_from_token(authorization)
125    pet_id = next_pet_id
126    next_pet_id += 1
127    pets[pet_id] = {
128        "id": pet_id,
129        "owner_user_id": user_id,
130        "name": req.name,
131        "feeding_schedule": req.feeding_schedule,
132        "stay_duration_days": req.stay_duration_days,
133        "photo_url": req.photo_url,
134        "photo_updates": []
135    }
136    return {"pet_id": pet_id}
137 
138@app.get("/pet/{pet_id}")
139def get_pet(pet_id: int, authorization: str = Header(None)):
140    get_user_id_from_token(authorization)
141    pet = pets.get(pet_id)
142    if not pet:
143        raise HTTPException(status_code=404, detail="Pet not found")
144    return pet
145 
146@app.post("/pet/{pet_id}/photo")
147def update_pet_photo(pet_id: int, photo_url: str, authorization: str = Header(None)):
148    user_id = get_user_id_from_token(authorization)
149    pet = pets.get(pet_id)
150    if not pet:
151        raise HTTPException(status_code=404, detail="Pet not found")
152    pet["photo_url"] = photo_url
153    pet["photo_updates"].append({
154        "url": photo_url,
155        "timestamp": datetime.datetime.utcnow().isoformat(),
156        "updated_by": user_id
157    })
158    return {"message": "Photo updated"}

1fastapi
2uvicorn