has_treatment.py

FIELDS = [
    "treatment_category",
    "urgent_treatment",
    "regulation_11_replacement_appliance",
    "prescription_only",
    "denture_repairs",
    "bridge_repairs",
    "arrest_of_bleeding",
    "removal_of_sutures",
]


def get_multiple_treatments(episode):
    treatment = episode.fp17treatmentcategory_set.get()
    result = []
    count = 0

    for field in FIELDS:
        if getattr(treatment, field) == True:
            result.append(field)

    if len(result) > 1:
        return (episode, result,)