Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, List
4import uuid
5import random
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11pets = {}
12applications = {}
13pet_id_counter = 1
14app_id_counter = 1
15 
16breeds = ["Labrador", "Poodle", "Beagle", "German Shepherd", "Bulldog", "Golden Retriever", "Persian Cat", "Siamese Cat", "Tabby Cat"]
17ages = ["puppy", "young", "adult", "senior"]
18 
19class SignupRequest(BaseModel):
20    username: str
21    password: str
22 
23class LoginRequest(BaseModel):
24    username: str
25    password: str
26 
27class PetCreate(BaseModel):
28    breed: Optional[str] = None
29    age: Optional[str] = None
30 
31class ApplicationCreate(BaseModel):
32    pet_id: int
33    adopter_name: str
34    adopter_email: str
35    has_yard: bool = False
36    has_other_pets: bool = False
37    has_children: bool = False
38 
39def get_current_user(authorization: str = Header(None)):
40    if not authorization:
41        raise HTTPException(status_code=401, detail="Missing auth token")
42    token = authorization.replace("Bearer ", "")
43    if token not in tokens:
44        raise HTTPException(status_code=401, detail="Invalid token")
45    return tokens[token]
46 
47@app.post("/signup")
48def signup(req: SignupRequest):
49    if req.username in users:
50        raise HTTPException(status_code=400, detail="User already exists")
51    users[req.username] = {"password": req.password}
52    return {"message": "User created"}
53 
54@app.post("/login")
55def login(req: LoginRequest):
56    if req.username not in users or users[req.username]["password"] != req.password:
57        raise HTTPException(status_code=401, detail="Invalid credentials")
58    token = str(uuid.uuid4())
59    tokens[token] = req.username
60    return {"token": token}
61 
62@app.get("/pets/available")
63def get_available_pets(authorization: str = Header(None)):
64    get_current_user(authorization)
65    available = []
66    for pid, pet in pets.items():
67        if not pet.get("adopted"):
68            compatibility = random.randint(1, 10)
69            pet["compatibility_score"] = compatibility
70            available.append({"id": pid, **pet})
71    return available
72 
73@app.post("/pets")
74def create_pet(pet: PetCreate, authorization: str = Header(None)):
75    get_current_user(authorization)
76    global pet_id_counter
77    new_pet = {
78        "id": pet_id_counter,
79        "breed": pet.breed if pet.breed else random.choice(breeds),
80        "age": pet.age if pet.age else random.choice(ages),
81        "adopted": False
82    }
83    pets[pet_id_counter] = new_pet
84    pet_id_counter += 1
85    return new_pet
86 
87@app.get("/pets/{pet_id}")
88def get_pet(pet_id: int, authorization: str = Header(None)):
89    get_current_user(authorization)
90    if pet_id not in pets:
91        raise HTTPException(status_code=404, detail="Pet not found")
92    return pets[pet_id]
93 
94@app.post("/applications/submit")
95def submit_application(app_data: ApplicationCreate, authorization: str = Header(None)):
96    user = get_current_user(authorization)
97    global app_id_counter
98    if app_data.pet_id not in pets:
99        raise HTTPException(status_code=404, detail="Pet not found")
100    if pets[app_data.pet_id]["adopted"]:
101        raise HTTPException(status_code=400, detail="Pet already adopted")
102    
103    score = 0
104    if app_data.has_yard:
105        score += 3
106    if app_data.has_other_pets:
107        score += 2
108    if app_data.has_children:
109        score += 1
110    
111    new_app = {
112        "id": app_id_counter,
113        "pet_id": app_data.pet_id,
114        "adopter_name": app_data.adopter_name,
115        "adopter_email": app_data.adopter_email,
116        "has_yard": app_data.has_yard,
117        "has_other_pets": app_data.has_other_pets,
118        "has_children": app_data.has_children,
119        "match_score": score,
120        "status": "pending",
121        "submitted_by": user
122    }
123    applications[app_id_counter] = new_app
124    app_id_counter += 1
125    return new_app
126 
127@app.get("/applications/{app_id}")
128def get_application(app_id: int, authorization: str = Header(None)):
129    get_current_user(authorization)
130    if app_id not in applications:
131        raise HTTPException(status_code=404, detail="Application not found")
132    return applications[app_id]

1fastapi
2uvicorn