class ProductFilter(django_filters.FilterSet):
name = django_filters.CharFilter(lookup_expr='icontains')
price = RangeFilter()
class Meta:
model = Product
fields = ['name', 'price']
class DestinationFilter(django_filters.FilterSet):
description = django_filters.CharFilter(lookup_expr='icontains')
price = RangeFilter()
check_in_date = DateFromToRangeFilter()
class Meta:
model = Destination
fields = ('check_in_date', 'price', 'description')
class ProductFilter(django_filters.FilterSet):
name = django_filters.CharFilter(lookup_expr='icontains')
price = RangeFilter()
class Meta:
model = Product
fields = ['name', 'price']
# In this file, we are creating a class called ProductFilter which inherits
# FilterSet from django filter. Then we have created two fields one is for name which is Charfield.
# Another one is price which I used RangeField cause I think people can filter products based on their budget.
def test_filtering_ignores_lookup_type(self):
qs = mock.Mock()
value = mock.Mock(start=20, stop=30)
f = RangeFilter(lookup_type='gte')
f.filter(qs, value)
qs.filter.assert_called_once_with(None__range=(20, 30))
def test_filtering_skipped_with_none_value(self):
qs = mock.Mock(spec=['filter'])
f = RangeFilter()
result = f.filter(qs, None)
self.assertEqual(qs, result)
def test_filtering(self):
qs = mock.Mock(spec=['filter'])
value = mock.Mock(start=20, stop=30)
f = RangeFilter()
f.filter(qs, value)
qs.filter.assert_called_once_with(None__range=(20, 30))
class ImpressoEnderecamentoContatoFilterSet(FilterSet):
FEMININO = 'F'
MASCULINO = 'M'
AMBOS = ''
SEXO_CHOICE = ((AMBOS, _('Ambos')), (FEMININO, _('Feminino')),
(MASCULINO, _('Masculino')))
DEPOIS_PRONOME = 'DP'
LINHA_NOME = 'LN'
DEPOIS_NOME = 'DN'
LOCAL_CARGO_CHOICE = ((DEPOIS_PRONOME,
_('Depois do Pronome '
'de tratamento')),
(LINHA_NOME, _('Antes do Nome do Contato')),
(DEPOIS_NOME,
_('Entre o Nome do Contato '
'e seu Endereço')))
FILHOS_CHOICE = [(None, _('Ambos'))] + YES_NO_CHOICES
search = CharFilter(method='filter_search')
sexo = ChoiceFilter(choices=SEXO_CHOICE)
tem_filhos = ChoiceFilter(label=_('Com filhos?'), choices=FILHOS_CHOICE)
idade = RangeFilter(label=_('Idade entre:'),
widget=RangeWidgetNumber,
method='filter_idade')
impresso = ModelChoiceFilter(
label=ImpressoEnderecamento._meta.verbose_name_plural,
required=True,
queryset=ImpressoEnderecamento.objects.all(),
method='filter_impresso')
grupo = ChoiceFilter(label=_('Contatos Agrupados'),
required=False,
method='filter_grupo')
imprimir_pronome = ChoiceFilter(label=_('Imprimir Pronome?'),
choices=YES_NO_CHOICES,
required=True,
initial=False,
method='filter_imprimir_pronome')
imprimir_cargo = ChoiceFilter(label=_('Imprimir Cargo?'),
required=True,
empty_label=None,
choices=YES_NO_CHOICES,
initial=False,
method='filter_imprimir_cargo')
fontsize = NumberFilter(label=_('Tamanho da Fonte'),
initial='',
max_value=100,
min_value=0,
max_digits=3,
decimal_places=0,
method='filter_fontsize')
nome_maiusculo = ChoiceFilter(label=_('Nome Maiúsculo'),
choices=YES_NO_CHOICES,
initial=False,
method='filter_nome_maiusculo')
local_cargo = ChoiceFilter(label=_('Local para imprimir o Cargo'),
choices=LOCAL_CARGO_CHOICE,
initial=False,
method='filter_local_cargo')
def filter_impresso(self, queryset, field_name, value):
return queryset
def filter_fontsize(self, queryset, field_name, value):
return queryset
def filter_grupo(self, queryset, field_name, value):
if value == '0':
queryset = queryset.filter(grupodecontatos_set__isnull=True)
elif value != '':
queryset = queryset.filter(grupodecontatos_set=value)
return queryset
def filter_local_cargo(self, queryset, field_name, value):
return queryset
def filter_imprimir_pronome(self, queryset, field_name, value):
return queryset
def filter_imprimir_cargo(self, queryset, field_name, value):
return queryset
def filter_nome_maiusculo(self, queryset, field_name, value):
return queryset
def filter_idade(self, queryset, field_name, value):
idi = int(value.start) if value.start is not None else 0
idf = int(
value.stop) if value.stop is not None else date.today().year - 2
if idi > idf:
a = idi
idi = idf
idf = a
# lim inicial-dt.mais antiga
li = date.today() - relativedelta(years=idf + 1)
# lim final - dt. mais nova
lf = date.today() - relativedelta(years=idi)
return queryset.filter(data_nascimento__gt=li, data_nascimento__lte=lf)
def filter_search(self, queryset, field_name, value):
query = normalize(value)
query = query.split(' ')
if query:
q = Q()
for item in query:
if not item:
continue
q = q & Q(search__icontains=item)
if q:
queryset = queryset.filter(q)
return queryset
def filter_data_nascimento(self, queryset, field_name, value):
#_where = "date_part('year', age(timestamp '%s', data_nascimento)) != date_part('year', age(timestamp '%s', data_nascimento))"
# return queryset.extra(where=_where, params=value)
if not value[0] or not value[1]:
return queryset
now = datetime.datetime.strptime(value[0], "%d/%m/%Y").date()
then = datetime.datetime.strptime(value[1], "%d/%m/%Y").date()
if now > then:
a = now
now = then
then = a
# Build the list of month/day tuples.
monthdays = [(now.month, now.day)]
while now <= then:
monthdays.append((now.month, now.day))
now += timedelta(days=1)
# Tranform each into queryset keyword args.
monthdays = (dict(
zip(("data_nascimento__month", "data_nascimento__day"), t))
for t in monthdays)
# Compose the djano.db.models.Q objects together for a single query.
query = reduce(operator.or_, (Q(**d) for d in monthdays))
# Run the query.
return queryset.extra(
select={
'month': 'extract( month from data_nascimento )',
'day': 'extract( day from data_nascimento )',
}).order_by('month', 'day', 'nome').filter(query)
data_nascimento = DataRangeFilter(widget=RangeWidgetOverride,
method='filter_data_nascimento')
class Meta:
model = Contato
fields = [
'search', 'sexo', 'tem_filhos', 'data_nascimento',
'tipo_autoridade'
]
"""filter_overrides = {
models.DateField: {
'filter_class': DataRangeFilter,
'extra': lambda f: {
'label': '%s (%s)' % (f.verbose_name, _('Inicial - Final')),
'widget': RangeWidgetOverride
}
}
}"""
def __init__(self, data=None, queryset=None, prefix=None, **kwargs):
workspace = kwargs.pop('workspace')
super(ImpressoEnderecamentoContatoFilterSet,
self).__init__(data=data,
queryset=queryset,
prefix=prefix,
**kwargs)
col1 = to_row([
('search', 6),
('sexo', 3),
('tem_filhos', 3),
('data_nascimento', 6),
('idade', 6),
('tipo_autoridade', 6),
('grupo', 6),
])
col2 = to_row([
('impresso', 12),
('fontsize', 4),
('nome_maiusculo', 4),
('imprimir_pronome', 4),
('imprimir_cargo', 5),
('local_cargo', 7),
])
row = to_row([
(Fieldset(
_('Informações para Seleção de Contatos'), col1,
to_row([(SubmitFilterPrint(
'filter',
value=_('Filtrar'),
css_class='btn-outline-primary pull-right',
type='submit'), 12)])), 6),
(Fieldset(
_('Informações para Impressão'), col2,
to_row([(SubmitFilterPrint('print',
value=_('Imprimir'),
css_class='btn-primary pull-right',
type='submit'), 12)])), 6)
])
self.form.helper = FormHelper()
self.form.helper.form_method = 'GET'
self.form.helper.layout = Layout(row, )
self.form.fields['search'].label = _('Nome/Nome Social/Apelido')
self.form.fields['data_nascimento'].label = '%s (%s)' % (
_('Aniversário'), _('Inicial - Final'))
self.form.fields['tem_filhos'].choices.choices[0] = (None, _('Ambos'))
self.form.fields['grupo'].choices = [
('0', _('Apenas Contatos sem Grupo')),
] + [(g.pk, str(g))
for g in GrupoDeContatos.objects.filter(workspace=workspace)]
class F(FilterSet):
price = RangeFilter()
class Meta:
model = Book
fields = ['price']
def test_filtering_start(self):
qs = mock.Mock(spec=['filter'])
value = mock.Mock(start=20, stop=None)
f = RangeFilter()
f.filter(qs, value)
qs.filter.assert_called_once_with(None__gte=20)
def test_filtering_exclude(self):
qs = mock.Mock(spec=['exclude'])
value = mock.Mock(start=20, stop=30)
f = RangeFilter(exclude=True)
f.filter(qs, value)
qs.exclude.assert_called_once_with(None__range=(20, 30))
def test_filtering_ignores_lookup_type(self):
qs = mock.Mock()
value = mock.Mock(start=20, stop=30)
f = RangeFilter(lookup_type='gte')
f.filter(qs, value)
qs.filter.assert_called_once_with(None__range=(20, 30))
def test_filtering_skipped_with_none_value(self):
qs = mock.Mock(spec=['filter'])
f = RangeFilter()
result = f.filter(qs, None)
self.assertEqual(qs, result)
def test_filtering_stop(self):
qs = mock.Mock(spec=['filter'])
value = mock.Mock(start=None, stop=30)
f = RangeFilter()
f.filter(qs, value)
qs.filter.assert_called_once_with(None__lte=30)
def test_filtering_exclude(self):
qs = mock.Mock(spec=['exclude'])
value = mock.Mock(start=20, stop=30)
f = RangeFilter(exclude=True)
f.filter(qs, value)
qs.exclude.assert_called_once_with(None__range=(20, 30))
def test_default_field(self):
f = RangeFilter()
field = f.field
self.assertIsInstance(field, RangeField)
