Coverage for apps/campaigns/filters.py: 88%

1import django_filters 

2 

3from base.utils import get_subclass_as_dict 

4 

5from .models import Campaign 

6 

7 

8class CampaignFilter(django_filters.rest_framework.FilterSet): 

9 ctype_dict = get_subclass_as_dict(Campaign) 

10 

11 campaign_type = django_filters.ChoiceFilter( 

12 method="campaign_type_filter", 

13 choices=list(ctype_dict.items()), 

14 ) 

15 

16 def campaign_type_filter(self, queryset, name, value): 

17 selected_campaign_type = self.ctype_dict[value] if value in self.ctype_dict.keys() else None 

18 return queryset.instance_of(selected_campaign_type) 

19 

20 class Meta: 

21 model = Campaign 

22 fields = ["campaign_type", "status", "organization"]