def select_all():
treatments = []
sql = "SELECT * FROM treatments"
results = run_sql(sql)
for row in results:
animal = animal_repository.select(row["animal_id"])
treatment = Treatment(
row["decription"],
row["duration"],
row["recovery"],
row["cost"],
animal,
row["id"],
)
treatment.start = row["start"]
treatments.append(treatment)
return treatments
def select(animal):
treatment = None
sql = "SELECT * FROM treatments WHERE animal_id = %s"
values = [animal.id]
row = run_sql(sql, values)
if len(row) > 0:
row = row[0]
treatment = Treatment(
row["description"],
row["duration"],
row["recovery"],
row["cost"],
animal,
row["id"],
)
treatment.start = row["start"]
return treatment