taiga
/
taiga-back
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Reimplemented search api using postgresql full-text-search capability.

remotes/origin/enhancement/email-actions
Andrey Antukh 2013-10-15 23:07:49 +02:00
parent 235a5df443
commit 63e53fec5d
1 changed files with 46 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

58
greenmine/base/searches/api.py
View File

@ -5,11 +5,10 @@ from django.db.models.loading import get_model
from rest_framework.response import Response
from rest_framework import viewsets
from haystack import query, inputs
from greenmine.base import exceptions as excp
from .serializers import SearchSerializer
from greenmine.projects.userstories.serializers import UserStorySerializer
from greenmine.projects.tasks.serializers import TaskSerializer
from greenmine.projects.issues.serializers import IssueSerializer
class SearchViewSet(viewsets.ViewSet):
@ -23,15 +22,14 @@ class SearchViewSet(viewsets.ViewSet):
except (project_model.DoesNotExist, TypeError):
raise excp.PermissionDenied({"detail": "Wrong project id"})
#if not text:
# raise excp.BadRequest("text parameter must be contains text")
result = {
"userstories": self._search_user_stories(project, text),
"tasks": self._search_tasks(project, text),
"issues": self._search_issues(project, text)
}
queryset = query.SearchQuerySet()
queryset = queryset.filter(text=inputs.AutoQuery(text))
queryset = queryset.filter(project_id=project_id)
return_data = SearchSerializer(queryset, many=True)
return Response(return_data.data)
result["count"] = sum(map(lambda x: len(x), result.values()))
return Response(result)
def _get_project(self, project_id):
project_model = get_model("projects", "Project")
@ -39,3 +37,39 @@ class SearchViewSet(viewsets.ViewSet):
.filter(members=self.request.user))
return own_projects.get(pk=project_id)
def _search_user_stories(self, project, text):
where_clause = ("to_tsvector(userstories_userstory.subject || "
"userstories_userstory.description) @@ plainto_tsquery(%s)")
model_cls = get_model("userstories", "UserStory")
queryset = (model_cls.objects
.extra(where=[where_clause], params=[text])
.filter(project_id=project.pk)[:50])
serializer = UserStorySerializer(queryset, many=True)
return serializer.data
def _search_tasks(self, project, text):
where_clause = ("to_tsvector(tasks_task.subject || tasks_task.description) "
"@@ plainto_tsquery(%s)")
model_cls = get_model("tasks", "Task")
queryset = (model_cls.objects
.extra(where=[where_clause], params=[text])
.filter(project_id=project.pk)[:50])
serializer = TaskSerializer(queryset, many=True)
return serializer.data
def _search_issues(self, project, text):
where_clause = ("to_tsvector(issues_issue.subject || issues_issue.description) "
"@@ plainto_tsquery(%s)")
model_cls = get_model("issues", "Issue")
queryset = (model_cls.objects
.extra(where=[where_clause], params=[text])
.filter(project_id=project.pk)[:50])
serializer = IssueSerializer(queryset, many=True)
return serializer.data