322 lines
11 KiB
Python
322 lines
11 KiB
Python
"""
|
|
In-memory data store for the Veterinary Clinic example.
|
|
|
|
This module is NOT thread-safe and is intended for demos and scaffolds only.
|
|
|
|
It provides minimal, process-local data stores for the five veterinary
|
|
clinic entities. Each store exposes standard CRUD operations backed by
|
|
a simple dictionary.
|
|
|
|
This module intentionally avoids:
|
|
- persistence
|
|
- concurrency guarantees
|
|
- transactional semantics
|
|
- validation beyond what Pydantic provides
|
|
|
|
This module is not part of the ``openapi_first`` library API surface.
|
|
"""
|
|
|
|
from datetime import date, datetime, timezone
|
|
|
|
from models import (
|
|
Parent, ParentCreate,
|
|
Vet, VetCreate,
|
|
Treatment, TreatmentCreate,
|
|
Pet, PetCreate,
|
|
Appointment, AppointmentCreate,
|
|
)
|
|
|
|
|
|
def _now():
|
|
return datetime.now(timezone.utc).isoformat()
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Parents
|
|
# ---------------------------------------------------------------------------
|
|
|
|
_parents: dict[int, Parent] = {}
|
|
_parents_next_id = 1
|
|
|
|
|
|
def list_parents() -> list[Parent]:
|
|
return list(_parents.values())
|
|
|
|
|
|
def get_parent(parent_id: int) -> Parent:
|
|
return _parents[parent_id]
|
|
|
|
|
|
def create_parent(payload: ParentCreate) -> Parent:
|
|
global _parents_next_id
|
|
now = _now()
|
|
parent = Parent(
|
|
id=_parents_next_id,
|
|
**payload.model_dump(exclude={"id", "metadata"}),
|
|
metadata={"createdOn": now, "updatedOn": now} if payload.metadata else None,
|
|
)
|
|
_parents[_parents_next_id] = parent
|
|
_parents_next_id += 1
|
|
return parent
|
|
|
|
|
|
def update_parent(parent_id: int, payload: ParentCreate) -> Parent:
|
|
if parent_id not in _parents:
|
|
raise KeyError(parent_id)
|
|
now = _now()
|
|
parent = _parents[parent_id]
|
|
updated = parent.model_copy(
|
|
update={
|
|
**payload.model_dump(exclude={"id", "metadata"}),
|
|
"metadata": {"createdOn": parent.metadata.get("createdOn", now), "updatedOn": now}
|
|
if parent.metadata else None,
|
|
}
|
|
)
|
|
_parents[parent_id] = updated
|
|
return updated
|
|
|
|
|
|
def delete_parent(parent_id: int) -> None:
|
|
del _parents[parent_id]
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Vets
|
|
# ---------------------------------------------------------------------------
|
|
|
|
_vets: dict[int, Vet] = {}
|
|
_vets_next_id = 1
|
|
|
|
|
|
def list_vets() -> list[Vet]:
|
|
return list(_vets.values())
|
|
|
|
|
|
def get_vet(vet_id: int) -> Vet:
|
|
return _vets[vet_id]
|
|
|
|
|
|
def create_vet(payload: VetCreate) -> Vet:
|
|
global _vets_next_id
|
|
now = _now()
|
|
vet = Vet(
|
|
id=_vets_next_id,
|
|
**payload.model_dump(exclude={"id", "metadata"}),
|
|
metadata={"createdOn": now, "updatedOn": now} if payload.metadata else None,
|
|
)
|
|
_vets[_vets_next_id] = vet
|
|
_vets_next_id += 1
|
|
return vet
|
|
|
|
|
|
def update_vet(vet_id: int, payload: VetCreate) -> Vet:
|
|
if vet_id not in _vets:
|
|
raise KeyError(vet_id)
|
|
now = _now()
|
|
vet = _vets[vet_id]
|
|
updated = vet.model_copy(
|
|
update={
|
|
**payload.model_dump(exclude={"id", "metadata"}),
|
|
"metadata": {"createdOn": vet.metadata.get("createdOn", now), "updatedOn": now}
|
|
if vet.metadata else None,
|
|
}
|
|
)
|
|
_vets[vet_id] = updated
|
|
return updated
|
|
|
|
|
|
def delete_vet(vet_id: int) -> None:
|
|
del _vets[vet_id]
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Treatments
|
|
# ---------------------------------------------------------------------------
|
|
|
|
_treatments: dict[int, Treatment] = {}
|
|
_treatments_next_id = 1
|
|
|
|
|
|
def list_treatments() -> list[Treatment]:
|
|
return list(_treatments.values())
|
|
|
|
|
|
def get_treatment(treatment_id: int) -> Treatment:
|
|
return _treatments[treatment_id]
|
|
|
|
|
|
def create_treatment(payload: TreatmentCreate) -> Treatment:
|
|
global _treatments_next_id
|
|
now = _now()
|
|
treatment = Treatment(
|
|
id=_treatments_next_id,
|
|
**payload.model_dump(exclude={"id", "metadata"}),
|
|
metadata={"createdOn": now, "updatedOn": now} if payload.metadata else None,
|
|
)
|
|
_treatments[_treatments_next_id] = treatment
|
|
_treatments_next_id += 1
|
|
return treatment
|
|
|
|
|
|
def update_treatment(treatment_id: int, payload: TreatmentCreate) -> Treatment:
|
|
if treatment_id not in _treatments:
|
|
raise KeyError(treatment_id)
|
|
now = _now()
|
|
treatment = _treatments[treatment_id]
|
|
updated = treatment.model_copy(
|
|
update={
|
|
**payload.model_dump(exclude={"id", "metadata"}),
|
|
"metadata": {"createdOn": treatment.metadata.get("createdOn", now), "updatedOn": now}
|
|
if treatment.metadata else None,
|
|
}
|
|
)
|
|
_treatments[treatment_id] = updated
|
|
return updated
|
|
|
|
|
|
def delete_treatment(treatment_id: int) -> None:
|
|
del _treatments[treatment_id]
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Pets
|
|
# ---------------------------------------------------------------------------
|
|
|
|
_pets: dict[int, Pet] = {}
|
|
_pets_next_id = 1
|
|
|
|
|
|
def list_pets() -> list[Pet]:
|
|
return list(_pets.values())
|
|
|
|
|
|
def get_pet(pet_id: int) -> Pet:
|
|
return _pets[pet_id]
|
|
|
|
|
|
def create_pet(payload: PetCreate) -> Pet:
|
|
global _pets_next_id
|
|
now = _now()
|
|
pet = Pet(
|
|
id=_pets_next_id,
|
|
**payload.model_dump(exclude={"id", "metadata"}),
|
|
metadata={"createdOn": now, "updatedOn": now} if payload.metadata else None,
|
|
)
|
|
_pets[_pets_next_id] = pet
|
|
_pets_next_id += 1
|
|
return pet
|
|
|
|
|
|
def update_pet(pet_id: int, payload: PetCreate) -> Pet:
|
|
if pet_id not in _pets:
|
|
raise KeyError(pet_id)
|
|
now = _now()
|
|
pet = _pets[pet_id]
|
|
updated = pet.model_copy(
|
|
update={
|
|
**payload.model_dump(exclude={"id", "metadata"}),
|
|
"metadata": {"createdOn": pet.metadata.get("createdOn", now), "updatedOn": now}
|
|
if pet.metadata else None,
|
|
}
|
|
)
|
|
_pets[pet_id] = updated
|
|
return updated
|
|
|
|
|
|
def delete_pet(pet_id: int) -> None:
|
|
del _pets[pet_id]
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Appointments
|
|
# ---------------------------------------------------------------------------
|
|
|
|
_appointments: dict[int, Appointment] = {}
|
|
_appointments_next_id = 1
|
|
|
|
|
|
def list_appointments() -> list[Appointment]:
|
|
return list(_appointments.values())
|
|
|
|
|
|
def get_appointment(appointment_id: int) -> Appointment:
|
|
return _appointments[appointment_id]
|
|
|
|
|
|
def create_appointment(payload: AppointmentCreate) -> Appointment:
|
|
global _appointments_next_id
|
|
now = _now()
|
|
appointment = Appointment(
|
|
id=_appointments_next_id,
|
|
**payload.model_dump(exclude={"id", "metadata"}),
|
|
metadata={"createdOn": now, "updatedOn": now} if payload.metadata else None,
|
|
)
|
|
_appointments[_appointments_next_id] = appointment
|
|
_appointments_next_id += 1
|
|
return appointment
|
|
|
|
|
|
def update_appointment(appointment_id: int, payload: AppointmentCreate) -> Appointment:
|
|
if appointment_id not in _appointments:
|
|
raise KeyError(appointment_id)
|
|
now = _now()
|
|
appointment = _appointments[appointment_id]
|
|
updated = appointment.model_copy(
|
|
update={
|
|
**payload.model_dump(exclude={"id", "metadata"}),
|
|
"metadata": {"createdOn": appointment.metadata.get("createdOn", now), "updatedOn": now}
|
|
if appointment.metadata else None,
|
|
}
|
|
)
|
|
_appointments[appointment_id] = updated
|
|
return updated
|
|
|
|
|
|
def delete_appointment(appointment_id: int) -> None:
|
|
del _appointments[appointment_id]
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Seed data — populate stores so the UI isn't empty on startup
|
|
# ---------------------------------------------------------------------------
|
|
|
|
def _seed_data():
|
|
now = _now()
|
|
meta = {"createdOn": now, "updatedOn": now}
|
|
global _parents_next_id, _vets_next_id, _treatments_next_id
|
|
global _pets_next_id, _appointments_next_id
|
|
|
|
_parents[1] = Parent(id=1, name="Alice Johnson", email="alice@example.com", phone="555-0101", metadata=meta)
|
|
_parents[2] = Parent(id=2, name="Bob Smith", email="bob@example.com", phone="555-0102", metadata=meta)
|
|
_parents[3] = Parent(id=3, name="Carol Williams", email="carol@example.com", phone="555-0103", metadata=meta)
|
|
_parents[4] = Parent(id=4, name="Dave Brown", email="dave@example.com", phone="555-0104", metadata=meta)
|
|
_parents_next_id = 5
|
|
|
|
_vets[1] = Vet(id=1, name="Sarah Connor", specialty="Surgery", email="sarah@clinic.com", phone="555-0201", metadata=meta)
|
|
_vets[2] = Vet(id=2, name="James Wilson", specialty="Dentistry", email="james@clinic.com", phone="555-0202", metadata=meta)
|
|
_vets[3] = Vet(id=3, name="Emily Davis", specialty="General Practice", email="emily@clinic.com", phone="555-0203", metadata=meta)
|
|
_vets_next_id = 4
|
|
|
|
_treatments[1] = Treatment(id=1, label="Annual Checkup", description="Full physical examination", metadata=meta)
|
|
_treatments[2] = Treatment(id=2, label="Vaccination", description="Core vaccines for common diseases", metadata=meta)
|
|
_treatments[3] = Treatment(id=3, label="Dental Cleaning", description="Scaling, polishing, and oral exam", metadata=meta)
|
|
_treatments[4] = Treatment(id=4, label="Spay/Neuter", description="Surgical sterilization", metadata=meta)
|
|
_treatments[5] = Treatment(id=5, label="Blood Panel", description="Complete blood count and chemistry", metadata=meta)
|
|
_treatments_next_id = 6
|
|
|
|
_pets[1] = Pet(id=1, name="Max", species="dog", age=4, weight=25.5, birthDate=date(2022, 3, 15), parent_ids=[1], metadata=meta)
|
|
_pets[2] = Pet(id=2, name="Luna", species="cat", age=2, weight=4.2, birthDate=date(2024, 1, 10), parent_ids=[1, 2], metadata=meta)
|
|
_pets[3] = Pet(id=3, name="Charlie", species="dog", age=7, weight=18.0, birthDate=date(2019, 8, 22), parent_ids=[2], metadata=meta)
|
|
_pets[4] = Pet(id=4, name="Bella", species="bird", age=1, weight=0.3, birthDate=date(2025, 5, 1), parent_ids=[3], metadata=meta)
|
|
_pets[5] = Pet(id=5, name="Rocky", species="dog", age=3, weight=30.0, birthDate=date(2023, 11, 5), parent_ids=[4], metadata=meta)
|
|
_pets_next_id = 6
|
|
|
|
_appointments[1] = Appointment(id=1, date=datetime(2026, 6, 18, 9, 0, tzinfo=timezone.utc), notes="Annual checkup", pet_id=1, vet_id=1, treatment_id=1, metadata=meta)
|
|
_appointments[2] = Appointment(id=2, date=datetime(2026, 6, 18, 10, 30, tzinfo=timezone.utc), notes="Dental cleaning", pet_id=2, vet_id=2, treatment_id=3, metadata=meta)
|
|
_appointments[3] = Appointment(id=3, date=datetime(2026, 6, 19, 11, 0, tzinfo=timezone.utc), notes="Vaccination booster", pet_id=3, vet_id=3, treatment_id=2, metadata=meta)
|
|
_appointments[4] = Appointment(id=4, date=datetime(2026, 6, 20, 14, 0, tzinfo=timezone.utc), notes="Follow-up after surgery", pet_id=5, vet_id=1, treatment_id=4, metadata=meta)
|
|
_appointments_next_id = 5
|
|
|
|
|
|
_seed_data()
|