Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, Dict
4 
5app = FastAPI()
6 
7users = {}
8user_id_counter = 1
9tokens = {}  # token -> user_id
10 
11shelters = {}
12shelter_id_counter = 1
13 
14listings = {}
15listing_id_counter = 1
16 
17class SignupRequest(BaseModel):
18    username: str
19    password: str
20    role: str = "user"  # user, shelter, superadmin
21 
22class LoginRequest(BaseModel):
23    username: str
24    password: str
25 
26class ShelterCreate(BaseModel):
27    name: str
28    address: str
29 
30class ListingCreate(BaseModel):
31    species: str
32    age: int
33    medical_history: str
34    shelter_id: int
35 
36class PromoteRequest(BaseModel):
37    user_id: int
38    shelter_id: int
39 
40import secrets
41 
42def get_current_user(authorization: str = Header(None)):
43    if not authorization:
44        raise HTTPException(status_code=401, detail="No auth header")
45    token = authorization.replace("Bearer ", "")
46    user_id = tokens.get(token)
47    if not user_id:
48        raise HTTPException(status_code=401, detail="Invalid token")
49    return user_id
50 
51@app.post("/signup")
52def signup(req: SignupRequest):
53    global user_id_counter
54    for u in users.values():
55        if u["username"] == req.username:
56            raise HTTPException(status_code=400, detail="Username taken")
57    uid = user_id_counter
58    user_id_counter += 1
59    users[uid] = {
60        "id": uid,
61        "username": req.username,
62        "password": req.password,
63        "role": req.role
64    }
65    token = secrets.token_hex(16)
66    tokens[token] = uid
67    # If first shelter to sign up, make them superadmin
68    if req.role == "shelter" and not any(u["role"] == "superadmin" for u in users.values()):
69        users[uid]["role"] = "superadmin"
70    return {"user_id": uid, "token": token}
71 
72@app.post("/login")
73def login(req: LoginRequest):
74    for uid, u in users.items():
75        if u["username"] == req.username and u["password"] == req.password:
76            token = secrets.token_hex(16)
77            tokens[token] = uid
78            return {"token": token}
79    raise HTTPException(status_code=401, detail="Invalid credentials")
80 
81@app.post("/shelters")
82def create_shelter(req: ShelterCreate, authorization: str = Header(None)):
83    user_id = get_current_user(authorization)
84    global shelter_id_counter
85    sid = shelter_id_counter
86    shelter_id_counter += 1
87    shelters[sid] = {
88        "id": sid,
89        "name": req.name,
90        "address": req.address,
91        "created_by": user_id
92    }
93    return shelters[sid]
94 
95@app.get("/shelters/{shelter_id}")
96def get_shelter(shelter_id: int):
97    s = shelters.get(shelter_id)
98    if not s:
99        raise HTTPException(status_code=404, detail="Shelter not found")
100    return s
101 
102@app.post("/listings")
103def create_listing(req: ListingCreate, authorization: str = Header(None)):
104    user_id = get_current_user(authorization)
105    if req.shelter_id not in shelters:
106        raise HTTPException(status_code=400, detail="Shelter not found")
107    global listing_id_counter
108    lid = listing_id_counter
109    listing_id_counter += 1
110    listings[lid] = {
111        "id": lid,
112        "species": req.species,
113        "age": req.age,
114        "medical_history": req.medical_history,
115        "shelter_id": req.shelter_id,
116        "created_by": user_id
117    }
118    return listings[lid]
119 
120@app.get("/listings/{listing_id}")
121def get_listing(listing_id: int):
122    l = listings.get(listing_id)
123    if not l:
124        raise HTTPException(status_code=404, detail="Listing not found")
125    return l
126 
127@app.post("/shelters/promote")
128def promote_user(req: PromoteRequest, authorization: str = Header(None)):
129    admin_id = get_current_user(authorization)
130    admin = users.get(admin_id)
131    if not admin or admin["role"] != "superadmin":
132        raise HTTPException(status_code=403, detail="Only superadmin can promote")
133    target = users.get(req.user_id)
134    if not target:
135        raise HTTPException(status_code=404, detail="User not found")
136    if req.shelter_id not in shelters:
137        raise HTTPException(status_code=404, detail="Shelter not found")
138    users[req.user_id]["role"] = "adopter_coordinator"
139    return {"message": f"User {req.user_id} promoted to adopter_coordinator"}

1fastapi
2uvicorn