-
Notifications
You must be signed in to change notification settings - Fork 0
/
simple_search.py
80 lines (69 loc) · 2.78 KB
/
simple_search.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
import re
from django.db.models import Q
from django import forms
from django.utils.text import smart_split
from django.conf import settings
from django.core.exceptions import FieldError
from blog.models import Entry
from django.contrib.sites.models import Site
class BaseSearchForm(forms.Form):
q = forms.CharField(label='Suchbegriffe', required=False)
def clean_q(self):
return self.cleaned_data['q'].strip()
# order_by = forms.CharField(
# widget=forms.HiddenInput(),
# required=False,
# )
class Meta:
abstract = True
base_qs = None
search_fields = None
fulltext_fields = None
def get_text_search_query(self, query_string):
filters = []
def construct_search(field_name):
if field_name.startswith('^'):
return "%s__istartswith" % field_name[1:]
elif field_name.startswith('='):
return "%s__iexact" % field_name[1:]
elif field_name.startswith('@'):
if settings.DATABASE_ENGINE == 'mysql':
return "%s__search" % field_name[1:]
else:
return "%s__icontains" % field_name[1:]
else:
return "%s__icontains" % field_name
for bit in smart_split(query_string):
or_queries = [Q(**{construct_search(str(field_name)): bit}) for field_name in self.Meta.search_fields]
filters.append(reduce(Q.__or__, or_queries))
return reduce(Q.__and__, filters)
def construct_filter_args(self, cleaned_data=None):
args = []
if not cleaned_data:
cleaned_data = self.cleaned_data
# construct text search
if cleaned_data['q']:
args.append(self.get_text_search_query(cleaned_data.pop('q')))
# if its an instance of Q, append to filter args
# otherwise assume an exact match lookup
for field in cleaned_data:
if isinstance(cleaned_data[field], Q):
args.append(cleaned_data[field])
# elif field == 'order_by':
# pass # special case - ordering handled in get_result_queryset
elif cleaned_data[field]:
args.append(Q(**{field: cleaned_data[field]}))
return args
def get_result_queryset(self):
qs = self.Meta.base_qs.filter(*self.construct_filter_args())
for field_name in self.Meta.search_fields:
if '__' in field_name:
qs = qs.distinct()
break
# if self.cleaned_data['order_by']:
# qs = qs.order_by(self.cleaned_data['order_by'])
return qs
class EntrySearchForm(BaseSearchForm):
class Meta:
base_qs = Entry.objects.filter(site=Site.objects.get_current())
search_fields = ['title','body',]