Merge pull request #419 from taigaio/issue/3125/searchs-api

Issue/3125/searchs api
remotes/origin/enhancement/email-actions
Alejandro 2015-08-04 14:01:51 +02:00
commit 65d1db1c33
5 changed files with 16 additions and 72 deletions

View File

@ -56,11 +56,12 @@ class SearchViewSet(viewsets.ViewSet):
futures_list.append(wiki_pages_future)
for future in futures.as_completed(futures_list):
data = []
try:
data = future.result()
except Exception as exc:
print('%s generated an exception: %s' % (future.result_key, exc))
else:
finally:
result[future.result_key] = data
result["count"] = sum(map(lambda x: len(x), result.values()))

View File

@ -1,41 +0,0 @@
# -*- coding: utf-8 -*-
from __future__ import unicode_literals
from django.db import models, migrations
class Migration(migrations.Migration):
dependencies = [
('wiki', '0001_initial'),
('userstories', '0009_remove_userstory_is_archived'),
('issues', '0005_auto_20150623_1923'),
('tasks', '0006_auto_20150623_1923'),
]
operations = [
migrations.RunSQL(
"""
CREATE INDEX "userstories_full_text_idx" ON userstories_userstory USING gin(to_tsvector('simple', coalesce(subject, '') || ' ' || coalesce(ref) || ' ' || coalesce(description, '')));
""",
reverse_sql="""DROP INDEX IF EXISTS "userstories_full_text_idx";"""
),
migrations.RunSQL(
"""
CREATE INDEX "tasks_full_text_idx" ON tasks_task USING gin(to_tsvector('simple', coalesce(subject, '') || ' ' || coalesce(ref) || ' ' || coalesce(description, '')));
""",
reverse_sql="""DROP INDEX IF EXISTS "tasks_full_text_idx";"""
),
migrations.RunSQL(
"""
CREATE INDEX "issues_full_text_idx" ON issues_issue USING gin(to_tsvector('simple', coalesce(subject, '') || ' ' || coalesce(ref) || ' ' || coalesce(description, '')));
""",
reverse_sql="""DROP INDEX IF EXISTS "issues_full_text_idx";"""
),
migrations.RunSQL(
"""
CREATE INDEX "wikipages_full_text_idx" ON wiki_wikipage USING gin(to_tsvector('simple', coalesce(slug, '') || ' ' || coalesce(content, '')));
""",
reverse_sql="""DROP INDEX IF EXISTS "wikipages_full_text_idx";"""
),
]

View File

@ -23,12 +23,12 @@ MAX_RESULTS = getattr(settings, "SEARCHES_MAX_RESULTS", 150)
def search_user_stories(project, text):
model_cls = apps.get_model("userstories", "UserStory")
where_clause = ("to_tsvector('simple', coalesce(userstories_userstory.subject, '') || ' ' || "
"coalesce(userstories_userstory.ref) || ' ' || "
"coalesce(userstories_userstory.description, '')) @@ to_tsquery(%s)")
where_clause = ("to_tsvector(coalesce(userstories_userstory.subject) || ' ' || "
"coalesce(userstories_userstory.ref) || ' ' || "
"coalesce(userstories_userstory.description, '')) "
"@@ plainto_tsquery(%s)")
if text:
text += ":*"
return (model_cls.objects.extra(where=[where_clause], params=[text])
.filter(project_id=project.pk)[:MAX_RESULTS])
@ -37,12 +37,11 @@ def search_user_stories(project, text):
def search_tasks(project, text):
model_cls = apps.get_model("tasks", "Task")
where_clause = ("to_tsvector('simple', coalesce(tasks_task.subject, '') || ' ' || "
where_clause = ("to_tsvector(coalesce(tasks_task.subject, '') || ' ' || "
"coalesce(tasks_task.ref) || ' ' || "
"coalesce(tasks_task.description, '')) @@ to_tsquery(%s)")
"coalesce(tasks_task.description, '')) @@ plainto_tsquery(%s)")
if text:
text += ":*"
return (model_cls.objects.extra(where=[where_clause], params=[text])
.filter(project_id=project.pk)[:MAX_RESULTS])
@ -51,12 +50,11 @@ def search_tasks(project, text):
def search_issues(project, text):
model_cls = apps.get_model("issues", "Issue")
where_clause = ("to_tsvector('simple', coalesce(issues_issue.subject) || ' ' || "
where_clause = ("to_tsvector(coalesce(issues_issue.subject) || ' ' || "
"coalesce(issues_issue.ref) || ' ' || "
"coalesce(issues_issue.description)) @@ to_tsquery(%s)")
"coalesce(issues_issue.description, '')) @@ plainto_tsquery(%s)")
if text:
text += ":*"
return (model_cls.objects.extra(where=[where_clause], params=[text])
.filter(project_id=project.pk)[:MAX_RESULTS])
@ -65,11 +63,11 @@ def search_issues(project, text):
def search_wiki_pages(project, text):
model_cls = apps.get_model("wiki", "WikiPage")
where_clause = ("to_tsvector('simple', coalesce(wiki_wikipage.slug) || ' ' || coalesce(wiki_wikipage.content)) "
"@@ to_tsquery(%s)")
where_clause = ("to_tsvector(coalesce(wiki_wikipage.slug) || ' ' || "
"coalesce(wiki_wikipage.content, '')) "
"@@ plainto_tsquery(%s)")
if text:
text += ":*"
return (model_cls.objects.extra(where=[where_clause], params=[text])
.filter(project_id=project.pk)[:MAX_RESULTS])

View File

@ -74,11 +74,11 @@ def searches_initial_data():
m.tsk2 = f.TaskFactory.create(project=m.project1)
m.tsk3 = f.TaskFactory.create(project=m.project1, subject="Back to the future")
m.iss1 = f.IssueFactory.create(project=m.project1, subject="Design and Frontend")
m.iss1 = f.IssueFactory.create(project=m.project1, subject="Backend and Frontend")
m.iss2 = f.IssueFactory.create(project=m.project2)
m.iss3 = f.IssueFactory.create(project=m.project1, subject="Green Frog")
m.iss3 = f.IssueFactory.create(project=m.project1)
m.wiki1 = f.WikiPageFactory.create(project=m.project1, content="Final Frontier")
m.wiki1 = f.WikiPageFactory.create(project=m.project1)
m.wiki2 = f.WikiPageFactory.create(project=m.project1, content="Frontend, future")
m.wiki3 = f.WikiPageFactory.create(project=m.project2)
@ -131,20 +131,6 @@ def test_search_text_query_in_my_project(client, searches_initial_data):
assert len(response.data["wikipages"]) == 0
def test_search_partial_text_query_in_my_project(client, searches_initial_data):
data = searches_initial_data
client.login(data.member1.user)
response = client.get(reverse("search-list"), {"project": data.project1.id, "text": "fron"})
assert response.status_code == 200
assert response.data["count"] == 3
assert len(response.data["userstories"]) == 0
assert len(response.data["tasks"]) == 0
assert len(response.data["issues"]) == 1
assert len(response.data["wikipages"]) == 2
def test_search_text_query_with_an_invalid_project_id(client, searches_initial_data):
data = searches_initial_data