forked from mirror/MrDoc
109 lines
3.9 KiB
Python
109 lines
3.9 KiB
Python
# coding:utf-8
|
||
# @文件: views_search.py
|
||
# @创建者：州的先生
|
||
# #日期：2020/11/22
|
||
# 博客地址：zmister.com
|
||
|
||
|
||
from haystack.generic_views import SearchView as BaseSearchView
|
||
from django.db.models import Q
|
||
from haystack.views import SearchView
|
||
from haystack.query import SearchQuerySet
|
||
from app_doc.models import *
|
||
import datetime
|
||
|
||
|
||
class DocSearchView2(BaseSearchView):
|
||
|
||
def get_queryset(self):
|
||
queryset = super(DocSearchView, self).get_queryset()
|
||
# further filter queryset based on some set of criteria
|
||
return queryset.filter(pub_date__gte=date(2015, 1, 1))
|
||
|
||
def get_context_data(self, *args, **kwargs):
|
||
context = super(DocSearchView, self).get_context_data(*args, **kwargs)
|
||
# do something
|
||
return context
|
||
|
||
# 文档搜索 - 基于Haystack全文搜索
|
||
class DocSearchView(SearchView):
|
||
results_per_page = 10
|
||
|
||
def __call__(self, request):
|
||
self.request = request
|
||
date_type = self.request.GET.get('d_type', 'recent')
|
||
date_range = self.request.GET.get('d_range', 'all') # 时间范围，默认不限，all
|
||
|
||
# 处理时间范围
|
||
if date_type == 'recent':
|
||
if date_range == 'recent1': # 最近1天
|
||
start_date = datetime.datetime.now() - datetime.timedelta(days=1)
|
||
elif date_range == 'recent7': # 最近7天
|
||
start_date = datetime.datetime.now() - datetime.timedelta(days=7)
|
||
elif date_range == 'recent30': # 最近30天
|
||
start_date = datetime.datetime.now() - datetime.timedelta(days=30)
|
||
elif date_range == 'recent365': # 最近一年
|
||
start_date = datetime.datetime.now() - datetime.timedelta(days=365)
|
||
else:
|
||
start_date = datetime.datetime.strptime('1900-01-01', '%Y-%m-%d')
|
||
end_date = datetime.datetime.now()
|
||
elif date_type == 'day':
|
||
try:
|
||
date_list = date_range.split('|')
|
||
start_date = datetime.datetime.strptime(date_list[0], '%Y-%m-%d')
|
||
end_date = datetime.datetime.strptime(date_list[1], '%Y-%m-%d')
|
||
except:
|
||
start_date = datetime.datetime.now() - datetime.timedelta(days=1)
|
||
end_date = datetime.datetime.now()
|
||
|
||
# 是否时间筛选
|
||
if date_range == 'all':
|
||
is_date_range = False
|
||
else:
|
||
is_date_range = True
|
||
|
||
# 是否认证
|
||
if self.request.user.is_authenticated:
|
||
is_auth = True
|
||
else:
|
||
is_auth = False
|
||
|
||
# 获取可搜索的文集列表
|
||
if is_auth:
|
||
colla_list = [i.project.id for i in
|
||
ProjectCollaborator.objects.filter(user=self.request.user)] # 用户的协作文集
|
||
open_list = [i.id for i in Project.objects.filter(
|
||
Q(role=0) | Q(create_user=self.request.user)
|
||
)] # 公开文集
|
||
|
||
view_list = list(set(open_list).union(set(colla_list))) # 合并上述两个文集ID列表
|
||
else:
|
||
view_list = [i.id for i in Project.objects.filter(role=0)] # 公开文集
|
||
|
||
if len(view_list) > 0:
|
||
sqs = SearchQuerySet().filter(
|
||
top_doc__in=view_list
|
||
).filter(
|
||
modify_time__gte=start_date,
|
||
modify_time__lte=end_date)
|
||
else:
|
||
sqs = SearchQuerySet().filter(
|
||
top_doc__in=None
|
||
).filter(
|
||
modify_time__gte=start_date,
|
||
modify_time__lte=end_date)
|
||
self.form = self.build_form(form_kwargs={'searchqueryset': sqs})
|
||
self.query = self.get_query().replace("\n",'').replace("\r",'')
|
||
self.results = self.get_results()
|
||
return self.create_response()
|
||
|
||
def extra_context(self):
|
||
context = {
|
||
'date_range':self.request.GET.get('d_range', 'all')
|
||
}
|
||
return context
|
||
|
||
|
||
|
||
|