phzhik/gault-millau
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add comments

Browse Source
This commit is contained in:
littlewolf 2019-09-27 12:23:12 +03:00
parent 910322273e
commit b55a8ffda6
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
apps/utils/querysets.py
View File

@ -22,6 +22,7 @@ class RelatedObjectsCountMixin(models.QuerySet):
"""QuerySet for ManyToMany related count filter""" """QuerySet for ManyToMany related count filter"""
def get_related_objects_names(self): def get_related_objects_names(self):
"""Get all related objects (with reversed)"""
related_objects_names = [] related_objects_names = []
for related_object in self.model._meta.related_objects: for related_object in self.model._meta.related_objects:
@ -31,7 +32,7 @@ class RelatedObjectsCountMixin(models.QuerySet):
return related_objects_names return related_objects_names
def annotate_related_objects_count(self): def annotate_related_objects_count(self):
"""Annotate all related objects to queryset"""
annotations = {} annotations = {}
for related_object in self.get_related_objects_names(): for related_object in self.get_related_objects_names():
annotations[f"{related_object}_count"] = models.Count(f"{related_object}") annotations[f"{related_object}_count"] = models.Count(f"{related_object}")
@ -39,6 +40,7 @@ class RelatedObjectsCountMixin(models.QuerySet):
return self.annotate(**annotations) return self.annotate(**annotations)
def filter_related_gt(self, count): def filter_related_gt(self, count):
"""QuerySet filter by related objects count"""
q_objects = Q() q_objects = Q()
for related_object in self.get_related_objects_names(): for related_object in self.get_related_objects_names():
q_objects.add(Q(**{f"{related_object}_count__gt": count}), Q.OR) q_objects.add(Q(**{f"{related_object}_count__gt": count}), Q.OR)