|
1 | 1 | from haystack.query import SearchQuerySet, EmptySearchQuerySet |
2 | 2 |
|
3 | | -class HaystackManager(object): |
| 3 | + |
| 4 | +class SearchIndexManager(object): |
4 | 5 | def __init__(self, using=None): |
5 | | - super(HaystackManager, self).__init__() |
| 6 | + super(SearchIndexManager, self).__init__() |
6 | 7 | self.using = using |
7 | | - |
8 | | - def get_query_set(self): |
| 8 | + |
| 9 | + def get_search_queryset(self): |
9 | 10 | """Returns a new SearchQuerySet object. Subclasses can override this method |
10 | 11 | to easily customize the behavior of the Manager. |
11 | 12 | """ |
12 | 13 | return SearchQuerySet(using=self.using) |
13 | | - |
| 14 | + |
14 | 15 | def get_empty_query_set(self): |
15 | 16 | return EmptySearchQuerySet(using=self.using) |
16 | | - |
| 17 | + |
17 | 18 | def all(self): |
18 | | - return self.get_query_set() |
19 | | - |
| 19 | + return self.get_search_queryset() |
| 20 | + |
20 | 21 | def none(self): |
21 | 22 | return self.get_empty_query_set() |
22 | | - |
| 23 | + |
23 | 24 | def filter(self, *args, **kwargs): |
24 | | - return self.get_query_set().filter(*args, **kwargs) |
25 | | - |
| 25 | + return self.get_search_queryset().filter(*args, **kwargs) |
| 26 | + |
26 | 27 | def exclude(self, *args, **kwargs): |
27 | | - return self.get_query_set().exclude(*args, **kwargs) |
28 | | - |
| 28 | + return self.get_search_queryset().exclude(*args, **kwargs) |
| 29 | + |
29 | 30 | def filter_and(self, *args, **kwargs): |
30 | | - return self.get_query_set().filter_and(*args, **kwargs) |
31 | | - |
| 31 | + return self.get_search_queryset().filter_and(*args, **kwargs) |
| 32 | + |
32 | 33 | def filter_or(self, *args, **kwargs): |
33 | | - return self.get_query_set().filter_or(*args, **kwargs) |
34 | | - |
| 34 | + return self.get_search_queryset().filter_or(*args, **kwargs) |
| 35 | + |
35 | 36 | def order_by(self, *args): |
36 | | - return self.get_query_set().order_by(*args) |
37 | | - |
| 37 | + return self.get_search_queryset().order_by(*args) |
| 38 | + |
38 | 39 | def order_by_distance(self, **kwargs): |
39 | | - return self.get_query_set().order_by_distance(**kwargs) |
40 | | - |
| 40 | + return self.get_search_queryset().order_by_distance(**kwargs) |
| 41 | + |
41 | 42 | def highlight(self): |
42 | | - return self.get_query_set().highlight() |
43 | | - |
| 43 | + return self.get_search_queryset().highlight() |
| 44 | + |
44 | 45 | def boost(self, term, boost): |
45 | | - return self.get_query_set().boost(term, boost) |
46 | | - |
| 46 | + return self.get_search_queryset().boost(term, boost) |
| 47 | + |
47 | 48 | def facet(self, field): |
48 | | - return self.get_query_set().facet(field) |
49 | | - |
| 49 | + return self.get_search_queryset().facet(field) |
| 50 | + |
50 | 51 | def within(self, field, point_1, point_2): |
51 | | - return self.get_query_set().within(field, point_1, point_2) |
52 | | - |
| 52 | + return self.get_search_queryset().within(field, point_1, point_2) |
| 53 | + |
53 | 54 | def dwithin(self, field, point, distance): |
54 | | - return self.get_query_set().dwithin(field, point, distance) |
55 | | - |
| 55 | + return self.get_search_queryset().dwithin(field, point, distance) |
| 56 | + |
56 | 57 | def distance(self, field, point): |
57 | | - return self.get_query_set().distance(field, point) |
58 | | - |
| 58 | + return self.get_search_queryset().distance(field, point) |
| 59 | + |
59 | 60 | def date_facet(self, field, start_date, end_date, gap_by, gap_amount=1): |
60 | | - return self.get_query_set().date_facet(field, start_date, end_date, gap_by, gap_amount=1) |
61 | | - |
| 61 | + return self.get_search_queryset().date_facet(field, start_date, end_date, gap_by, gap_amount=1) |
| 62 | + |
62 | 63 | def query_facet(self, field, query): |
63 | | - return self.get_query_set().query_facet(field, query) |
64 | | - |
| 64 | + return self.get_search_queryset().query_facet(field, query) |
| 65 | + |
65 | 66 | def narrow(self, query): |
66 | | - return self.get_query_set().narrow(query) |
67 | | - |
| 67 | + return self.get_search_queryset().narrow(query) |
| 68 | + |
68 | 69 | def raw_search(self, query_string, **kwargs): |
69 | | - return self.get_query_set().raw_search(query_string, **kwargs) |
70 | | - |
| 70 | + return self.get_search_queryset().raw_search(query_string, **kwargs) |
| 71 | + |
71 | 72 | def load_all(self): |
72 | | - return self.get_query_set().load_all() |
73 | | - |
| 73 | + return self.get_search_queryset().load_all() |
| 74 | + |
74 | 75 | def auto_query(self, query_string, fieldname='content'): |
75 | | - return self.get_query_set().auto_query(query_string, fieldname=fieldname) |
76 | | - |
| 76 | + return self.get_search_queryset().auto_query(query_string, fieldname=fieldname) |
| 77 | + |
77 | 78 | def autocomplete(self, **kwargs): |
78 | | - return self.get_query_set().autocomplete(**kwargs) |
79 | | - |
| 79 | + return self.get_search_queryset().autocomplete(**kwargs) |
| 80 | + |
80 | 81 | def using(self, connection_name): |
81 | | - return self.get_query_set().using(connection_name) |
82 | | - |
| 82 | + return self.get_search_queryset().using(connection_name) |
| 83 | + |
83 | 84 | def count(self): |
84 | | - return self.get_query_set().count() |
85 | | - |
| 85 | + return self.get_search_queryset().count() |
| 86 | + |
86 | 87 | def best_match(self): |
87 | | - return self.get_query_set().best_match() |
88 | | - |
| 88 | + return self.get_search_queryset().best_match() |
| 89 | + |
89 | 90 | def latest(self, date_field): |
90 | | - return self.get_query_set().latest(date_field) |
91 | | - |
| 91 | + return self.get_search_queryset().latest(date_field) |
| 92 | + |
92 | 93 | def more_like_this(self, model_instance): |
93 | | - return self.get_query_set().more_like_this(model_instance) |
94 | | - |
| 94 | + return self.get_search_queryset().more_like_this(model_instance) |
| 95 | + |
95 | 96 | def facet_counts(self): |
96 | | - return self.get_query_set().facet_counts() |
97 | | - |
| 97 | + return self.get_search_queryset().facet_counts() |
| 98 | + |
98 | 99 | def spelling_suggestion(self, preferred_query=None): |
99 | | - return self.get_query_set().spelling_suggestion(preferred_query=None) |
100 | | - |
| 100 | + return self.get_search_queryset().spelling_suggestion(preferred_query=None) |
| 101 | + |
101 | 102 | def values(self, *fields): |
102 | | - return self.get_query_set().values(*fields) |
103 | | - |
104 | | - def values_list(self, *fields, **kwargs): |
105 | | - return self.get_query_set().values_list(*fields, **kwargs) |
| 103 | + return self.get_search_queryset().values(*fields) |
106 | 104 |
|
| 105 | + def values_list(self, *fields, **kwargs): |
| 106 | + return self.get_search_queryset().values_list(*fields, **kwargs) |
0 commit comments