Django product count per category - django

Question

Have below models
class TreatmentCategory(models.Model):
name = models.CharField(max_length=255, unique=True)
def __str__(self):
return self.name
class Treatment(models.Model):
unique_id = models.UUIDField('Unique Record ID', default=uuid.uuid4, editable=False)
name = models.CharField('Treatment Name', max_length=255, db_index=True, unique=True)
category = models.ForeignKey(TreatmentCategory, on_delete=models.PROTECT)
duration = models.PositiveSmallIntegerField('Treatment Duration (min)')
price = models.DecimalField('Treatment Cost', max_digits=9, decimal_places=2)
description = models.TextField(null=True, blank=True)
active = models.BooleanField(default=True)
def __str__(self):
return self.name
class Therapist(models.Model):
day_options = [
("Off Day", "Off Day"),
("On Leave", "On Leave"),
("On Duty", "On Duty"),
]
unique_id = models.UUIDField('Unique Record ID', default=uuid.uuid4, editable=False)
name = models.CharField('Therapist Name', max_length=255)
contact = models.CharField('Phone Number', max_length=12, blank=True, null=True)
day_status = models.CharField(
'Duty Status',
max_length=255,
choices=day_options,
default='On Duty'
)
status = models.BooleanField(
'Status of Therapist',
default=True,
)
treatments = models.ManyToManyField(Treatment, related_name='therapists')
def __str__(self):
return self.name
Would like to output all therapists and a total count of treatments per category they have records i.e each therapist may have different treatments per category thus they wont have same number of treatments
therapist1 | treatmentcategory1 (number of treatments)
| treatmentcategory2 (number of treatments)
| treatmentcategory3 (number of treatments)
therapist2 | treatmentcategory1 (number of treatments)
My current code as below, its displaying therapists however, instead of treatmentcategories, it showing treatments
category = (TreatmentCategory.objects.filter(treatment__active=True)
.prefetch_related('treatment_set')
.annotate(treatment_count=Count('treatment__name'))
)
queryset = (Therapist.objects.filter(status=True)
.prefetch_related(
Prefetch('mytreatments', queryset=category, to_attr='treatment_details')
)
)
Testing category alone gives the list of treatmentcategories, treatment count and all treatments in each treatmentcategory.
Queryset returns the therapists and treatments only, no categories nor treatment count per category.
Kindly assist in refining/rewriting the queries to extract the data as per sample above.
Thanks in advance.