Coverage for users/filterset_mixins.py: 100%

1import django_filters 

2from django.utils.translation import gettext as _ 

3 

4from users.models import Owner, OwnerGroup 

5 

6__all__ = ( 

7 'OwnerFilterMixin', 

8) 

9 

10 

11class OwnerFilterMixin(django_filters.FilterSet): 

12 """ 

13 Adds owner & owner_id filters for models which inherit from OwnerMixin. 

14 """ 

15 owner_group_id = django_filters.ModelMultipleChoiceFilter( 

16 queryset=OwnerGroup.objects.all(), 

17 distinct=False, 

18 field_name='owner__group', 

19 label=_('Owner Group (ID)'), 

20 ) 

21 owner_group = django_filters.ModelMultipleChoiceFilter( 

22 queryset=OwnerGroup.objects.all(), 

23 distinct=False, 

24 field_name='owner__group__name', 

25 to_field_name='name', 

26 label=_('Owner Group (name)'), 

27 ) 

28 owner_id = django_filters.ModelMultipleChoiceFilter( 

29 queryset=Owner.objects.all(), 

30 distinct=False, 

31 label=_('Owner (ID)'), 

32 ) 

33 owner = django_filters.ModelMultipleChoiceFilter( 

34 field_name='owner__name', 

35 queryset=Owner.objects.all(), 

36 distinct=False, 

37 to_field_name='name', 

38 label=_('Owner (name)'), 

39 )