a1batross
/
imaginaryfriend
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Fixes to mod_f and mod_d commands working not as expected

This commit is contained in:
REDNBLACK 2017-01-05 23:40:53 +03:00
parent 5ae8e5e3df
commit 83c9975fbb
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/repository/trigram_repository.py
View File

@ -76,7 +76,10 @@ class TrigramRepository(RedisRepository):
for pair in redis.scan_iter(match=search_pattern, count=max_results): for pair in redis.scan_iter(match=search_pattern, count=max_results):
(first, second) = pair.decode(encoding).lstrip(format_pattern).split(self.separator) (first, second) = pair.decode(encoding).lstrip(format_pattern).split(self.separator)
words.add(first if similar_word in first else second) if first.startswith(similar_word):
words.add(first)
if second.startswith(similar_word):
words.add(second)
return list(words)[:10] return list(words)[:10]
@ -86,9 +89,8 @@ class TrigramRepository(RedisRepository):
:param chat_id: ID of chat :param chat_id: ID of chat
:param exact_word: Exact word match :param exact_word: Exact word match
""" """
self.__remove_keys(self.source_name.format(chat_id, exact_word + "\\" + self.separator + '*'))
self.__remove_keys(self.source_name.format(chat_id, exact_word + self.separator + '*')) self.__remove_keys(self.source_name.format(chat_id, '*' + "\\" + self.separator + exact_word))
self.__remove_keys(self.source_name.format(chat_id, '*' + self.separator + exact_word))
# FIXME. Not optimal performance wise # FIXME. Not optimal performance wise
def __remove_keys(self, pattern): def __remove_keys(self, pattern):