class ArticleDateRangeFilterSet(django_filters.FilterSet):
"""Allows a list of news items to be filtered by a date range, an author, or many tags."""
projects = django_filters.ModelMultipleChoiceFilter(
name="projects",
queryset=Project.objects.get_public(),
widget=autocomplete_light.MultipleChoiceWidget('ProjectAutocomplete'),
)
authors = django_filters.ModelMultipleChoiceFilter(
queryset=(User.objects.annotate(
article_count=Count('authored_articles')).filter(
article_count__gt=0)),
widget=autocomplete_light.MultipleChoiceWidget(
'UserAuthorAutocomplete'))
pub_date = django_filters.DateFromToRangeFilter(
label='Date Range',
lookup_expr='contains',
widget=RangeWidget(attrs={
'class': 'datepicker',
'placeholder': 'MM/DD/YYYY',
}),
)
tags = django_filters.ModelMultipleChoiceFilter(
name='tags__name',
queryset=Tag.objects.all(),
label='Tags',
widget=autocomplete_light.MultipleChoiceWidget('TagAutocomplete'),
)
class Meta:
model = Article
fields = ['projects', 'authors', 'pub_date']
class QuestionFilterSet(django_filters.FilterSet):
"""Allows question to be filtered by user, date, or if it's unanswered."""
user = django_filters.ModelChoiceFilter(
queryset=User.objects.all(),
widget=autocomplete_light.ChoiceWidget('UserAutocomplete')
)
date = django_filters.DateFromToRangeFilter(
label='Date Range',
lookup_expr='contains',
widget=RangeWidget(attrs={
'class': 'datepicker',
'placeholder': 'MM/DD/YYYY',
}),
)
unanswered = django_filters.MethodFilter(
action='unanswered_filter',
widget=forms.CheckboxInput()
)
tags = django_filters.ModelMultipleChoiceFilter(
name='tags__name',
queryset=Tag.objects.all(),
widget=autocomplete_light.MultipleChoiceWidget('TagAutocomplete'),
)
def unanswered_filter(self, queryset, value):
"""Filter to show either only unanswered questions or all questions"""
# pylint: disable=no-self-use
if value:
return queryset.filter(answers__isnull=True)
else:
return queryset
class Meta:
model = Question
fields = ['user', 'date']
class FOIARequestFilterSet(django_filters.FilterSet):
"""Allows filtering a request by status, agency, jurisdiction, user, or tags."""
status = django_filters.ChoiceFilter(choices=BLANK_STATUS)
user = django_filters.ModelMultipleChoiceFilter(
queryset=User.objects.all(),
widget=autocomplete_light.MultipleChoiceWidget('UserAutocomplete')
)
agency = django_filters.ModelMultipleChoiceFilter(
queryset=Agency.objects.get_approved(),
widget=autocomplete_light.MultipleChoiceWidget('AgencyAutocomplete')
)
jurisdiction = django_filters.ModelMultipleChoiceFilter(
queryset=Jurisdiction.objects.filter(hidden=False),
widget=autocomplete_light.MultipleChoiceWidget('JurisdictionAutocomplete')
)
projects = django_filters.ModelMultipleChoiceFilter(
name="projects",
queryset=Project.objects.get_public(),
widget=autocomplete_light.MultipleChoiceWidget('ProjectAutocomplete'),
)
tags = django_filters.ModelMultipleChoiceFilter(
name='tags__name',
queryset=Tag.objects.all(),
widget=autocomplete_light.MultipleChoiceWidget('TagAutocomplete'),
)
has_embargo = django_filters.BooleanFilter(
name='embargo',
widget=forms.Select(choices=NULL_BOOLEAN_CHOICES),
)
has_crowdfund = django_filters.BooleanFilter(
name='crowdfund',
lookup_expr='isnull',
exclude=True,
widget=forms.Select(choices=NULL_BOOLEAN_CHOICES),
)
minimum_pages = django_filters.NumberFilter(
name='communications__files__pages',
lookup_expr='gte',
label='Min. Pages',
distinct=True,
widget=forms.NumberInput(),
)
date_range = django_filters.DateFromToRangeFilter(
name='communications__date',
label='Date Range',
lookup_expr='contains',
widget=RangeWidget(attrs={
'class': 'datepicker',
'placeholder': 'MM/DD/YYYY',
}),
)
class Meta:
model = FOIARequest
fields = ['status', 'user', 'agency', 'jurisdiction', 'projects', 'tags']
class ArticleAuthorFilterSet(django_filters.FilterSet):
"""Allows a list of articles by an author to be filtered by a date range or many tags."""
pub_date = django_filters.DateFromToRangeFilter(
label='Date Range',
lookup_expr='contains',
widget=RangeWidget(attrs={
'class': 'datepicker',
'placeholder': 'MM/DD/YYYY',
}),
)
tags = django_filters.ModelMultipleChoiceFilter(
name='tags__name',
queryset=Tag.objects.all(),
widget=autocomplete_light.MultipleChoiceWidget('TagAutocomplete'),
)
class Meta:
model = Article
fields = ['pub_date', 'tags']
class QuestionFilterSet(django_filters.FilterSet):
"""Allows question to be filtered by user, date, or if it's unanswered."""
user = django_filters.ModelChoiceFilter(
queryset=User.objects.all(),
widget=autocomplete_light.ChoiceWidget('UserAutocomplete'))
date = django_filters.DateFromToRangeFilter(
label='Date Range',
lookup_expr='contains',
widget=RangeWidget(attrs={
'class': 'datepicker',
'placeholder': 'MM/DD/YYYY',
}),
)
unanswered = django_filters.BooleanFilter(name='answers',
lookup_expr='isnull',
widget=forms.CheckboxInput())
class Meta:
model = Question
fields = ['user', 'date']
class TaskFilterSet(django_filters.FilterSet):
"""Allows tasks to be filtered by whether they're resolved, and by who resolved them."""
resolved = django_filters.BooleanFilter(
label='Show Resolved',
widget=forms.CheckboxInput())
resolved_by = django_filters.ModelMultipleChoiceFilter(
queryset=User.objects.all(),
widget=autocomplete_light.MultipleChoiceWidget('UserTaskAutocomplete')
)
date_created = django_filters.DateFromToRangeFilter(
label='Date Range',
lookup_expr='contains',
widget=RangeWidget(attrs={
'class': 'datepicker',
'placeholder': 'MM/DD/YYYY',
}),
)
class Meta:
model = Task
fields = ['resolved', 'resolved_by']
class AgencyFOIARequestFilterSet(django_filters.FilterSet):
"""Filters for agency users"""
user = django_filters.ModelMultipleChoiceFilter(
queryset=User.objects.all(),
widget=autocomplete_light.MultipleChoiceWidget('UserAutocomplete'))
tags = django_filters.ModelMultipleChoiceFilter(
name='tags__name',
queryset=Tag.objects.all(),
widget=autocomplete_light.MultipleChoiceWidget('TagAutocomplete'),
)
date_range = django_filters.DateFromToRangeFilter(
name='communications__date',
label='Date Range',
lookup_expr='contains',
widget=RangeWidget(attrs={
'class': 'datepicker',
'placeholder': 'MM/DD/YYYY',
}),
)
class Meta:
model = FOIARequest
fields = ['user', 'tags']
