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

Improvements to #7 algorithm

This commit is contained in:
REDNBLACK 2016-11-12 21:53:34 +03:00
parent 701bbd4ae2
commit 889a8f7fde
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/domain/message.py
View File

@ -58,15 +58,18 @@ class Message:
return random.randint(0, 100) < getattr(self.chat, 'random_chance', config['bot']['default_chance'])
def __get_words(self):
text = list(self.text)
symbols = list(self.text)
def prettify(word):
lowercase_word = word.lower().strip()
pretty_word = lowercase_word.strip(config['grammar']['all'])
last_symbol = lowercase_word[-1:]
if last_symbol not in config['grammar']['end_sentence']:
last_symbol = ''
pretty_word = lowercase_word.strip(config['grammar']['all']) + last_symbol
return pretty_word if pretty_word != '' and len(pretty_word) > 2 else lowercase_word
for entity in self.message.entities:
text[entity.offset:entity.length] = ' ' * entity.length
symbols[entity.offset:entity.length] = ' ' * entity.length
return list(filter(None, map(prettify, ''.join(text).split(' '))))
return list(filter(None, map(prettify, ''.join(symbols).split(' '))))