Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, Dict, Any
4import secrets
5 
6app = FastAPI()
7 
8users = {}
9pets = {}
10applications = {}
11tokens = {}
12next_user_id = 1
13next_pet_id = 1
14next_app_id = 1
15next_token_id = 1
16 
17class SignupRequest(BaseModel):
18    username: str
19    password: str
20    role: str = "adopter"
21 
22class LoginRequest(BaseModel):
23    username: str
24    password: str
25 
26class PetCreate(BaseModel):
27    breed: str
28    age: int
29    shelter_id: int
30 
31class ApplicationCreate(BaseModel):
32    pet_id: int
33    adopter_id: int
34 
35def get_current_user(authorization: str = Header(None)):
36    if not authorization:
37        raise HTTPException(status_code=401, detail="Missing auth header")
38    token = authorization.replace("Bearer ", "")
39    user_id = tokens.get(token)
40    if not user_id:
41        raise HTTPException(status_code=401, detail="Invalid token")
42    return user_id
43 
44@app.post("/signup")
45def signup(req: SignupRequest):
46    global next_user_id
47    user_id = next_user_id
48    next_user_id += 1
49    users[user_id] = {"username": req.username, "password": req.password, "role": req.role}
50    return {"user_id": user_id, "message": "User created"}
51 
52@app.post("/login")
53def login(req: LoginRequest):
54    for uid, u in users.items():
55        if u["username"] == req.username and u["password"] == req.password:
56            global next_token_id
57            token = secrets.token_hex(16)
58            tokens[token] = uid
59            return {"token": token}
60    raise HTTPException(status_code=401, detail="Invalid credentials")
61 
62@app.get("/users/{user_id}")
63def get_user(user_id: int):
64    user = users.get(user_id)
65    if not user:
66        raise HTTPException(status_code=404, detail="User not found")
67    return {"user_id": user_id, **user}
68 
69@app.get("/pets/{pet_id}")
70def get_pet(pet_id: int):
71    pet = pets.get(pet_id)
72    if not pet:
73        raise HTTPException(status_code=404, detail="Pet not found")
74    return {"pet_id": pet_id, **pet}
75 
76@app.get("/applications/{application_id}")
77def get_application(application_id: int):
78    app = applications.get(application_id)
79    if not app:
80        raise HTTPException(status_code=404, detail="Application not found")
81    return {"application_id": application_id, **app}
82 
83@app.post("/pets")
84def create_pet(pet: PetCreate, authorization: str = Header(None)):
85    user_id = get_current_user(authorization)
86    user = users.get(user_id)
87    if not user or user["role"] != "shelter":
88        raise HTTPException(status_code=403, detail="Only shelters can create pets")
89    global next_pet_id
90    pet_id = next_pet_id
91    next_pet_id += 1
92    pets[pet_id] = {"breed": pet.breed, "age": pet.age, "shelter_id": pet.shelter_id}
93    return {"pet_id": pet_id, "message": "Pet created"}
94 
95@app.post("/applications")
96def create_application(app: ApplicationCreate, authorization: str = Header(None)):
97    user_id = get_current_user(authorization)
98    user = users.get(user_id)
99    if not user or user["role"] != "adopter":
100        raise HTTPException(status_code=403, detail="Only adopters can apply")
101    if app.pet_id not in pets:
102        raise HTTPException(status_code=404, detail="Pet not found")
103    global next_app_id
104    app_id = next_app_id
105    next_app_id += 1
106    applications[app_id] = {"pet_id": app.pet_id, "adopter_id": app.adopter_id, "status": "pending"}
107    return {"application_id": app_id, "message": "Application created"}
108 
109@app.put("/applications/{application_id}/status")
110def update_application_status(application_id: int, status: str, authorization: str = Header(None)):
111    user_id = get_current_user(authorization)
112    user = users.get(user_id)
113    if not user or user["role"] != "admin":
114        raise HTTPException(status_code=403, detail="Only admins can update status")
115    if application_id not in applications:
116        raise HTTPException(status_code=404, detail="Application not found")
117    applications[application_id]["status"] = status
118    return {"application_id": application_id, "status": status, "message": "Status updated"}

1fastapi
2uvicorn
3pydantic