You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
70 lines
2.4 KiB
70 lines
2.4 KiB
import django_filters
|
|
from django.db.models import Q
|
|
from django.contrib.auth import get_user_model
|
|
from django.contrib import admin
|
|
from django.utils import timezone
|
|
|
|
from dateutil.relativedelta import relativedelta
|
|
|
|
from .models import Activity, Prospect
|
|
|
|
User = get_user_model()
|
|
|
|
class ProspectFilter(django_filters.FilterSet):
|
|
zip_code = django_filters.CharFilter(lookup_expr='istartswith', label='Code postal')
|
|
activities = django_filters.ModelMultipleChoiceFilter(
|
|
queryset=Activity.objects.all(),
|
|
field_name='activities',
|
|
)
|
|
city = django_filters.CharFilter(lookup_expr='icontains', label='Ville')
|
|
name = django_filters.CharFilter(method='filter_name', label='Nom')
|
|
|
|
def filter_name(self, queryset, name, value):
|
|
return queryset.filter(
|
|
Q(first_name__icontains=value) | Q(last_name__icontains=value) | Q(entity_name__icontains=value)
|
|
)
|
|
|
|
class Meta:
|
|
model = Prospect
|
|
fields = ['name', 'city', 'activities', 'zip_code']
|
|
|
|
class StaffUserFilter(admin.SimpleListFilter):
|
|
title = 'staff user'
|
|
parameter_name = 'user'
|
|
|
|
def lookups(self, request, model_admin):
|
|
staff_users = User.objects.filter(is_staff=True)
|
|
return [(user.id, user.username) for user in staff_users]
|
|
|
|
def queryset(self, request, queryset):
|
|
# Filter the queryset based on the selected user ID
|
|
if self.value():
|
|
return queryset.filter(related_user__id=self.value())
|
|
return queryset
|
|
|
|
class ProspectProfileFilter(admin.SimpleListFilter):
|
|
title = 'Prospect profiles' # displayed in the admin UI
|
|
parameter_name = 'profile' # URL parameter
|
|
|
|
def lookups(self, request, model_admin):
|
|
return (
|
|
('tournament_at_least_1_month_old', 'tournaments > 1 month old'),
|
|
('no_tournaments', 'No tournaments'),
|
|
)
|
|
|
|
def queryset(self, request, queryset):
|
|
if not self.value():
|
|
return queryset
|
|
|
|
two_months_ago = timezone.now().date() - relativedelta(months=2)
|
|
|
|
if self.value() == 'tournament_at_least_2_month_old':
|
|
return queryset.filter(
|
|
official_user__isnull=False,
|
|
official_user__events__creation_date__lte=two_months_ago
|
|
)
|
|
elif self.value() == 'no_tournaments':
|
|
return queryset.filter(
|
|
official_user__isnull=False,
|
|
official_user__events__isnull=True
|
|
)
|
|
|