X-Git-Url: https://git.openstreetmap.org./osqa.git/blobdiff_plain/9f2913c2565411c1bb8de79a9e7d17e441c9b409..cae5bfd94d187b8258411467ad6231f24b84e4f0:/forum_modules/pgfulltext/handlers.py diff --git a/forum_modules/pgfulltext/handlers.py b/forum_modules/pgfulltext/handlers.py index 196790b..18a23e3 100644 --- a/forum_modules/pgfulltext/handlers.py +++ b/forum_modules/pgfulltext/handlers.py @@ -3,25 +3,27 @@ from django.db.models import Q from forum.models.question import Question, QuestionManager from forum.modules.decorators import decorate +repl_re = re.compile(r"^'|[^\'\-_\s\w]|'$", re.UNICODE) +sing_quote_re = re.compile(r"\'+") + @decorate(QuestionManager.search, needs_origin=False) def question_search(self, keywords): - repl_re = re.compile(r"[^\'\-_\s\w]", re.UNICODE) - tsquery = " | ".join([k for k in repl_re.sub('', keywords).split(' ') if k]) + tsquery = " | ".join([k for k in repl_re.sub('', sing_quote_re.sub("'", keywords.strip())).split(' ') if k]) ilike = keywords + u"%%" return self.extra( - tables = ['forum_rootnode_doc'], - select={ - 'ranking': """ + tables = ['forum_rootnode_doc'], + select={ + 'ranking': """ rank_exact_matches(ts_rank_cd('{0.1, 0.2, 0.8, 1.0}'::float4[], "forum_rootnode_doc"."document", to_tsquery('english', %s), 32)) """, - }, - where=[""" + }, + where=[""" "forum_rootnode_doc"."node_id" = "forum_node"."id" AND ("forum_rootnode_doc"."document" @@ to_tsquery('english', %s) OR "forum_node"."title" ILIKE %s) """], - params=[tsquery, ilike], - select_params=[tsquery], - order_by=['-ranking'] - ) + params=[tsquery, ilike], + select_params=[tsquery], + order_by=['-ranking'] + )