Browse the data

1from fastapi import FastAPI, HTTPException, Header
2from pydantic import BaseModel
3from typing import Optional
4import random
5import string
6 
7app = FastAPI()
8 
9users = {}
10tokens = {}
11shelters = {}
12pets = {}
13adopters = {}
14next_user_id = 1
15next_shelter_id = 1
16next_pet_id = 1
17next_adopter_id = 1
18 
19def generate_token():
20    return ''.join(random.choices(string.ascii_letters + string.digits, k=32))
21 
22def get_current_user(authorization: Optional[str] = Header(None)):
23    if not authorization:
24        raise HTTPException(status_code=401, detail="Missing authorization header")
25    token = authorization.replace("Bearer ", "")
26    if token not in tokens:
27        raise HTTPException(status_code=401, detail="Invalid token")
28    return tokens[token]
29 
30class SignupRequest(BaseModel):
31    username: str
32    password: str
33 
34class LoginRequest(BaseModel):
35    username: str
36    password: str
37 
38class ShelterCreate(BaseModel):
39    name: str
40    location: str
41 
42class PetCreate(BaseModel):
43    name: str
44    species: str
45    age: int
46    shelter_id: int
47 
48class AdopterCreate(BaseModel):
49    name: str
50    email: str
51 
52@app.post("/signup")
53def signup(req: SignupRequest):
54    global next_user_id
55    user_id = next_user_id
56    next_user_id += 1
57    users[user_id] = {"username": req.username, "password": req.password}
58    token = generate_token()
59    tokens[token] = user_id
60    return {"user_id": user_id, "token": token}
61 
62@app.post("/login")
63def login(req: LoginRequest):
64    for uid, u in users.items():
65        if u["username"] == req.username and u["password"] == req.password:
66            token = generate_token()
67            tokens[token] = uid
68            return {"token": token}
69    raise HTTPException(status_code=401, detail="Invalid credentials")
70 
71@app.post("/shelters")
72def create_shelter(shelter: ShelterCreate, authorization: Optional[str] = Header(None)):
73    get_current_user(authorization)
74    global next_shelter_id
75    sid = next_shelter_id
76    next_shelter_id += 1
77    shelters[sid] = {"id": sid, "name": shelter.name, "location": shelter.location}
78    return shelters[sid]
79 
80@app.get("/shelters/{shelter_id}")
81def get_shelter(shelter_id: int, authorization: Optional[str] = Header(None)):
82    get_current_user(authorization)
83    if shelter_id not in shelters:
84        raise HTTPException(status_code=404, detail="Shelter not found")
85    return shelters[shelter_id]
86 
87@app.post("/pets")
88def create_pet(pet: PetCreate, authorization: Optional[str] = Header(None)):
89    get_current_user(authorization)
90    if pet.shelter_id not in shelters:
91        raise HTTPException(status_code=400, detail="Shelter does not exist")
92    global next_pet_id
93    pid = next_pet_id
94    next_pet_id += 1
95    pets[pid] = {"id": pid, "name": pet.name, "species": pet.species, "age": pet.age, "shelter_id": pet.shelter_id}
96    return pets[pid]
97 
98@app.get("/pets/{pet_id}")
99def get_pet(pet_id: int, authorization: Optional[str] = Header(None)):
100    get_current_user(authorization)
101    if pet_id not in pets:
102        raise HTTPException(status_code=404, detail="Pet not found")
103    return pets[pet_id]
104 
105@app.post("/adopters")
106def create_adopter(adopter: AdopterCreate, authorization: Optional[str] = Header(None)):
107    get_current_user(authorization)
108    global next_adopter_id
109    aid = next_adopter_id
110    next_adopter_id += 1
111    adopters[aid] = {"id": aid, "name": adopter.name, "email": adopter.email}
112    return adopters[aid]
113 
114@app.get("/adopters/{adopter_id}")
115def get_adopter(adopter_id: int, authorization: Optional[str] = Header(None)):
116    get_current_user(authorization)
117    if adopter_id not in adopters:
118        raise HTTPException(status_code=404, detail="Adopter not found")
119    return adopters[adopter_id]

1fastapi
2uvicorn