Merge pull request #247 from PROCOLLAB-github/afarut-patch-1

users > filters > create filter for age

Author: VeryBigSad

poetry.lock

@@ -1,3 +1,4 @@
+import datetime
 from django.contrib.auth import get_user_model
 from django.db.models import Q
 from django_filters import rest_framework as filters
@@ -43,6 +44,21 @@ def filter_by_partner_program(cls, queryset, name, value):
         except PartnerProgram.DoesNotExist:
             return User.objects.none()
 
+    @classmethod
+    def filter_age(cls, queryset, name, value):
+        start, stop = map(int, value.split(","))
+        start, stop = min(start, stop), max(start, stop)
+        return queryset.filter(
+            Q(
+                birthday__gte=datetime.datetime.now()
+                - datetime.timedelta(days=365.24 * int(stop))
+            )
+            & Q(
+                birthday__lte=datetime.datetime.now()
+                - datetime.timedelta(days=365.24 * int(start))
+            )
+        )
+
     @classmethod
     def filter_by_fullname(cls, queryset, name, value):
         words = value.split()
@@ -77,8 +93,7 @@ def filter_by_fullname(cls, queryset, name, value):
     )
     fullname = filters.CharFilter(method="filter_by_fullname")
 
-    age__gte = filters.NumberFilter(field_name="age", lookup_expr="gte")
-    age__lte = filters.NumberFilter(field_name="age", lookup_expr="lte")
+    age = filters.Filter(field_name="age", method="filter_age")
 
     class Meta:
         model = User