Django / haystack - 仅基于 DateField 的年份元素的分面

Django / haystack - faceting based only on the year element of a DateField

假设我在 Django 实现中有以下模型:

class Broadcast(models.Model):
    broadcast_date = models.DateField()
    ...

在我的 haystack/Solr-driven 搜索结果中,我希望能够仅按广播年份而不是确切的日期来分类。

我可以为此创建一个新的模型属性,但似乎我应该能够以某种方式在 search_indexes 模型定义中执行此操作。这是我目前在 search_indexes.py 中的内容,仅按完整日期显示:

class BroadcastIndex(indexes.SearchIndex, indexes.Indexable):
    text = [...]
    broadcast_date = indexes.DateField(model_attr="broadcast_date", faceted=True)

有什么想法吗?

您应该可以使用 date_facet()。如果将它的 gap_by 参数设置为 year,它将按年分面。

只是概述一下我最后做了什么。

首先,在我的 urls.py 文件中,我包含了一个使用 date_facet() 的自定义 SearchQuerySet(感谢 Adrian Ghiuta!),我将其传递到我的搜索视图:

...
from haystack.forms import FacetedSearchForm
from haystack.query import SearchQuerySet

sqs = SearchQuerySet().date_facet('broadcast_date', start_date=date(1955, 1, 1), end_date=date(2015, 1, 1), gap_by="year")

urlpatterns = [
    ...
    url(r'^search/', views.SarkSearch(form_class=FacetedSearchForm, searchqueryset=sqs), name='haystack_search'),
]

date_facet() 函数确实给出了带有年份计数的多面日期字典,但返回的键的格式为 yyyy-01-01:00:00:00,所以我最终将该字符串中的年份。我使用列表理解来创建一个只包含我想要的数据的新列表,然后我将其添加到 facets 上下文变量中。我使用的是基于 class 的视图子 classing FacetedSearchView,因此我需要将所有处理代码放在 extra_context() 函数中:

class FooSearch(FacetedSearchView):
    def extra_content(self):
        extra = super(FooSearch, self).extra_context()
        dates = extra['facets']['dates']['broadcast_date']
        dates = sorted([[year[:4], count] for year, count in dates.items() if count > 0])
        extra['facets']['dates']['broadcast_date'] = dates
        return extra

既然我已经有了返回的年份方面和计数,我需要想出一种方法将实际搜索结果缩小到给定年份内的那些。在我的搜索模板中,我使用 GET 变量 "year_facet" 将每个返回的分面日期绑定到一个分面 link。看起来像这样:

<dl>
    {% if facets.dates.broadcast_date %}
        <dt>Broadcast dates</dt>
        {% for date in facets.dates.broadcast_date %}
            <dd><a href="{{ request.get_full_path }}&amp;year_facet={{ date.0 }}">{{ date.0 }}</a> ({{ date.1 }})</dd>
        {% endfor %}
    {% endif %}
    {% if 'year_facet' in request.get_full_path %}
        <!-- I want a way to clear the facet if one is present -->
        <dd><a href="?q={{ query }}"><em>see all</em></a></dd>
    {% endif %}
</dl>

既然已经设置了 link,我需要一种方法让我的视图在选择基于年份的日期方面时实际缩小结果。为此,回到我的搜索视图中,我重写了 get_results() 函数以包含给定年份内所有日期的日期过滤器:

def get_results(self):
    if 'year_facet' in self.request.GET:
        year = int(self.request.GET['year_facet'])
        return self.form.search().filter(broadcast_date__lte=datetime.date(year, 12, 31)).filter(broadcast_date__gte=datetime.date(year, 1, 1))

    return self.form.search()

所有这一切都非常有效。