base_indexes.py
3.01 KB
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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
# -*- coding: utf-8 -*-
import math
from haystack import indexes
class BaseIndex(indexes.SearchIndex):
text = indexes.CharField(document=True, use_template=True, stored=False)
url = indexes.CharField(model_attr='get_absolute_url', indexed=False)
title = indexes.CharField()
description = indexes.CharField(null=True)
fullname = indexes.CharField(null=True)
author = indexes.CharField(null=True)
author_url = indexes.CharField(null=True, indexed=False)
created = indexes.DateTimeField(model_attr='created', null=True)
modified = indexes.DateTimeField(model_attr='modified', null=True)
type = indexes.CharField()
icon_name = indexes.CharField(indexed=False)
fullname_and_username = indexes.CharField(null=True, stored=False)
hits = indexes.IntegerField(model_attr='hits')
modified_by = indexes.CharField(null=True)
modified_by_url = indexes.CharField(null=True)
def get_updated_field(self):
return 'modified'
def get_boost(self, obj):
if obj.hits <= 10:
return 1
return math.log(obj.hits)
def prepare(self, obj):
self.author_obj = None
if hasattr(obj, 'get_author'):
self.author_obj = obj.get_author()
data = super(BaseIndex, self).prepare(obj)
data['boost'] = self.get_boost(obj)
return data
def prepare_author(self, obj):
if self.author_obj:
return self.author_obj.username
return obj.author
def prepare_author_url(self, obj):
if self.author_obj:
return self.author_obj.get_absolute_url()
return None
def prepare_fullname(self, obj):
if hasattr(obj, 'modified_by'):
modified_by = obj.get_modified_by()
if modified_by:
return modified_by.get_full_name()
if self.author_obj:
return self.author_obj.get_full_name()
return obj.author
def prepare_fullname_and_username(self, obj):
if hasattr(obj, 'modified_by'):
modified_by = obj.get_modified_by()
if modified_by:
return u'{}\n{}'.format(
modified_by.get_full_name(),
modified_by.username,
)
if not self.author_obj:
return obj.author
return u'{}\n{}'.format(
self.author_obj.get_full_name(),
self.author_obj.username,
)
def prepare_modified_by(self, obj):
if hasattr(obj, 'modified_by'):
modified_by = obj.get_modified_by()
if modified_by:
return modified_by.get_full_name()
if self.author_obj:
return self.author_obj.get_full_name()
return obj.author
def prepare_modified_by_url(self, obj):
if hasattr(obj, 'modified_by'):
modified_by = obj.get_modified_by()
if modified_by:
return modified_by.get_absolute_url()
if self.author_obj:
return self.author_obj.get_absolute_url()
return None