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 = {}
9pets = {}
10vaccinations = {}
11tokens = {}
12user_id_counter = 1
13pet_id_counter = 1
14vaccination_id_counter = 1
15 
16class SignupRequest(BaseModel):
17    username: str
18    password: str
19 
20class LoginRequest(BaseModel):
21    username: str
22    password: str
23 
24class PetCreate(BaseModel):
25    name: str
26    breed: str
27    age: int
28    medical_notes: str = ""
29    owner_contact: str = ""
30    tier: str = "standard"
31 
32class PetUpdate(BaseModel):
33    name: Optional[str] = None
34    breed: Optional[str] = None
35    age: Optional[int] = None
36    medical_notes: Optional[str] = None
37    owner_contact: Optional[str] = None
38    tier: Optional[str] = None
39 
40class VaccinationCreate(BaseModel):
41    pet_id: int
42    vaccine_name: str
43    date_administered: str
44    next_due_date: Optional[str] = None
45 
46def get_current_user(authorization: str = Header(...)):
47    token = authorization.replace("Bearer ", "")
48    if token not in tokens:
49        raise HTTPException(status_code=401, detail="Invalid token")
50    return tokens[token]
51 
52@app.post("/signup")
53def signup(req: SignupRequest):
54    global user_id_counter
55    if req.username in users:
56        raise HTTPException(status_code=400, detail="Username already exists")
57    user_id = user_id_counter
58    user_id_counter += 1
59    users[req.username] = {"id": user_id, "username": req.username, "password": req.password}
60    return {"id": user_id, "username": req.username}
61 
62@app.post("/login")
63def login(req: LoginRequest):
64    if req.username not in users or users[req.username]["password"] != req.password:
65        raise HTTPException(status_code=401, detail="Invalid credentials")
66    token = secrets.token_hex(16)
67    tokens[token] = req.username
68    return {"token": token}
69 
70@app.get("/pets/{pet_id}")
71def get_pet(pet_id: int, authorization: str = Header(...)):
72    get_current_user(authorization)
73    if pet_id not in pets:
74        raise HTTPException(status_code=404, detail="Pet not found")
75    return pets[pet_id]
76 
77@app.post("/pets")
78def create_pet(pet: PetCreate, authorization: str = Header(...)):
79    global pet_id_counter
80    get_current_user(authorization)
81    pet_id = pet_id_counter
82    pet_id_counter += 1
83    pets[pet_id] = {
84        "id": pet_id,
85        "name": pet.name,
86        "breed": pet.breed,
87        "age": pet.age,
88        "medical_notes": pet.medical_notes,
89        "owner_contact": pet.owner_contact,
90        "tier": pet.tier,
91        "vaccinations": []
92    }
93    return pets[pet_id]
94 
95@app.patch("/pets/{pet_id}")
96def update_pet(pet_id: int, pet: PetUpdate, authorization: str = Header(...)):
97    get_current_user(authorization)
98    if pet_id not in pets:
99        raise HTTPException(status_code=404, detail="Pet not found")
100    existing = pets[pet_id]
101    if pet.name is not None:
102        existing["name"] = pet.name
103    if pet.breed is not None:
104        existing["breed"] = pet.breed
105    if pet.age is not None:
106        existing["age"] = pet.age
107    if pet.medical_notes is not None:
108        existing["medical_notes"] = pet.medical_notes
109    if pet.owner_contact is not None:
110        existing["owner_contact"] = pet.owner_contact
111    if pet.tier is not None:
112        existing["tier"] = pet.tier
113    return existing
114 
115@app.get("/vaccinations/{vaccination_id}")
116def get_vaccination(vaccination_id: int, authorization: str = Header(...)):
117    get_current_user(authorization)
118    if vaccination_id not in vaccinations:
119        raise HTTPException(status_code=404, detail="Vaccination not found")
120    return vaccinations[vaccination_id]
121 
122@app.post("/vaccinations")
123def create_vaccination(vac: VaccinationCreate, authorization: str = Header(...)):
124    global vaccination_id_counter
125    get_current_user(authorization)
126    if vac.pet_id not in pets:
127        raise HTTPException(status_code=404, detail="Pet not found")
128    vac_id = vaccination_id_counter
129    vaccination_id_counter += 1
130    vaccination_entry = {
131        "id": vac_id,
132        "pet_id": vac.pet_id,
133        "vaccine_name": vac.vaccine_name,
134        "date_administered": vac.date_administered,
135        "next_due_date": vac.next_due_date
136    }
137    vaccinations[vac_id] = vaccination_entry
138    pets[vac.pet_id]["vaccinations"].append(vac_id)
139    return vaccination_entry

1fastapi
2uvicorn