Skip to content

Commit 2bcee04

Browse files
authored
Update filters.py
1 parent e9cd2cf commit 2bcee04

1 file changed

Lines changed: 8 additions & 3 deletions

File tree

users/filters.py

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,8 @@
1+
import datetime
12
from django.contrib.auth import get_user_model
23
from django.db.models import Q
34
from django_filters import rest_framework as filters
5+
from dateutil.relativedelta import relativedelta
46

57
from partner_programs.models import PartnerProgram, PartnerProgramUserProfile
68

@@ -44,9 +46,12 @@ def filter_by_partner_program(cls, queryset, name, value):
4446
return User.objects.none()
4547

4648
@classmethod
47-
def filter_by_age(cls, queryset, name, value):
48-
start, stop = value
49-
return queryset.filter(Q(birthday__lte=stop) & Q(birthday__gte=start))
49+
def filter_age__gte(cls, queryset, name, value):
50+
return queryset.filter(birthday__gte=datetime.datetime.now() - relativedelta(years=int(value)))
51+
52+
@classmethod
53+
def filter_age__lte(cls, queryset, name, value):
54+
return queryset.filter(birthday__lte=datetime.datetime.now() - relativedelta(years=int(value)))
5055

5156
@classmethod
5257
def filter_by_fullname(cls, queryset, name, value):

0 commit comments

Comments
 (0)