Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional, List
4import secrets
5 
6app = FastAPI()
7 
8users = {}
9tokens = {}
10shelters = {}
11pets = {}
12applications = {}
13next_user_id = 1
14next_shelter_id = 1
15next_pet_id = 1
16next_app_id = 1
17 
18class SignupRequest(BaseModel):
19    username: str
20    password: str
21    lifestyle: Optional[str] = None
22 
23class LoginRequest(BaseModel):
24    username: str
25    password: str
26 
27class ShelterCreate(BaseModel):
28    name: str
29 
30class PetCreate(BaseModel):
31    shelter_id: int
32    breed: str
33    temperament: str
34 
35class ApplicationCreate(BaseModel):
36    pet_id: int
37    user_id: int
38 
39def get_current_user(authorization: str = Header(None)):
40    if not authorization or not authorization.startswith("Bearer "):
41        raise HTTPException(status_code=401, detail="Invalid auth")
42    token = authorization.split(" ")[1]
43    for uid, t in tokens.items():
44        if t == token:
45            return uid
46    raise HTTPException(status_code=401, detail="Invalid token")
47 
48@app.post("/signup")
49def signup(req: SignupRequest):
50    global next_user_id
51    uid = next_user_id
52    next_user_id += 1
53    users[uid] = {"username": req.username, "password": req.password, "lifestyle": req.lifestyle}
54    return {"user_id": uid}
55 
56@app.post("/login")
57def login(req: LoginRequest):
58    for uid, u in users.items():
59        if u["username"] == req.username and u["password"] == req.password:
60            token = secrets.token_hex(16)
61            tokens[uid] = token
62            return {"token": token}
63    raise HTTPException(status_code=401, detail="Invalid credentials")
64 
65@app.get("/users/{user_id}")
66def get_user(user_id: int, authorization: str = Header(None)):
67    get_current_user(authorization)
68    if user_id not in users:
69        raise HTTPException(status_code=404, detail="User not found")
70    return users[user_id]
71 
72@app.post("/shelters")
73def create_shelter(shelter: ShelterCreate, authorization: str = Header(None)):
74    get_current_user(authorization)
75    global next_shelter_id
76    sid = next_shelter_id
77    next_shelter_id += 1
78    shelters[sid] = {"id": sid, "name": shelter.name}
79    return shelters[sid]
80 
81@app.get("/shelters/{shelter_id}")
82def get_shelter(shelter_id: int, authorization: str = Header(None)):
83    get_current_user(authorization)
84    if shelter_id not in shelters:
85        raise HTTPException(status_code=404, detail="Shelter not found")
86    return shelters[shelter_id]
87 
88@app.post("/pets")
89def create_pet(pet: PetCreate, authorization: str = Header(None)):
90    get_current_user(authorization)
91    if pet.shelter_id not in shelters:
92        raise HTTPException(status_code=400, detail="Shelter not found")
93    global next_pet_id
94    pid = next_pet_id
95    next_pet_id += 1
96    pets[pid] = {"id": pid, "shelter_id": pet.shelter_id, "breed": pet.breed, "temperament": pet.temperament, "adopted": False}
97    return pets[pid]
98 
99@app.get("/pets/{pet_id}")
100def get_pet(pet_id: int, authorization: str = Header(None)):
101    get_current_user(authorization)
102    if pet_id not in pets:
103        raise HTTPException(status_code=404, detail="Pet not found")
104    return pets[pet_id]
105 
106@app.post("/applications")
107def create_application(app_req: ApplicationCreate, authorization: str = Header(None)):
108    get_current_user(authorization)
109    if app_req.pet_id not in pets:
110        raise HTTPException(status_code=400, detail="Pet not found")
111    if app_req.user_id not in users:
112        raise HTTPException(status_code=400, detail="User not found")
113    if pets[app_req.pet_id]["adopted"]:
114        raise HTTPException(status_code=400, detail="Pet already adopted")
115    global next_app_id
116    aid = next_app_id
117    next_app_id += 1
118    user = users[app_req.user_id]
119    pet = pets[app_req.pet_id]
120    score = 0
121    if user.get("lifestyle") and pet["temperament"]:
122        if user["lifestyle"] == "active" and pet["temperament"] in ["energetic", "playful"]:
123            score = 10
124        elif user["lifestyle"] == "calm" and pet["temperament"] in ["calm", "lazy"]:
125            score = 10
126        elif user["lifestyle"] == "active" and pet["temperament"] in ["calm", "lazy"]:
127            score = 5
128        elif user["lifestyle"] == "calm" and pet["temperament"] in ["energetic", "playful"]:
129            score = 2
130        else:
131            score = 5
132    applications[aid] = {"id": aid, "pet_id": app_req.pet_id, "user_id": app_req.user_id, "compatibility_score": score}
133    return applications[aid]
134 
135@app.get("/applications/{app_id}")
136def get_application(app_id: int, authorization: str = Header(None)):
137    get_current_user(authorization)
138    if app_id not in applications:
139        raise HTTPException(status_code=404, detail="Application not found")
140    return applications[app_id]

1fastapi
2uvicorn
3pydantic