[fix] query highlight fix

This commit is contained in:
asciimoo 2013-11-10 21:47:42 +01:00
parent cbb397939d
commit b9ebf18f59
1 changed files with 5 additions and 3 deletions

View File

@ -112,11 +112,13 @@ def highlight_content(content, query):
query_regex = '({0})'.format(re.escape(query))
content = re.sub(query_regex, '<b>\\1</b>', content, flags=re.I)
else:
regex_parts = []
for chunk in query.split():
if len(chunk) == 1:
query_regex = '(\W+{0}\W+)'.format(re.escape(chunk))
regex_parts.append('\W+{0}\W+'.format(re.escape(chunk)))
else:
query_regex = '({0})'.format(re.escape(chunk))
regex_parts.append('{0}'.format(re.escape(chunk)))
query_regex = '({0})'.format('|'.join(regex_parts))
content = re.sub(query_regex, '<b>\\1</b>', content, flags=re.I)
return content