bakatrouble/telegram_bots
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix mtg

Browse Source
This commit is contained in:
bakatrouble 2019-11-27 22:41:31 +03:00
parent 4e58ee1fc6
commit b821e96084
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
bots/modules/_mtg.py
View File

@ -49,11 +49,12 @@ class MTG(object):
if second_word is not None: if second_word is not None:
return first_word, second_word return first_word, second_word
elif first_word is not None: elif first_word is not None:
if ('.' + '+' + first_word in self.MC) and len(self.MC['.' + '+' + first_word]) != 0: word = '.' + '+' + first_word
second_word = random.choice(self.MC['.' + '+' + first_word]) word_cap = '.' + '+' + first_word.capitalize()
elif ('.' + '+' + first_word.capitalize() in self.MC) and \ if word in self.MC and len(self.MC[word]):
len(self.MC['.' + '+' + first_word.capitalize()]) != 0: second_word = random.choice(self.MC[word])
second_word = random.choice(self.MC['.' + '+' + first_word.capitalize()]) elif word_cap in self.MC and len(self.MC[word_cap]):
second_word = random.choice(self.MC[word_cap])
else: else:
return self.choose_start_words() return self.choose_start_words()
return first_word, second_word return first_word, second_word
@ -67,7 +68,7 @@ class MTG(object):
while not (len(gen_words) > size and w2 == '.'): while not (len(gen_words) > size and w2 == '.'):
gen_words.append(w2.capitalize() if (w1 == '.') else w2) gen_words.append(w2.capitalize() if (w1 == '.') else w2)
w1, w2 = w2, random.choice(self.MC[w1 + '+' + w2]) w1, w2 = w2, random.choice(self.MC[w1 + '+' + w2])
gen_words.append(w2) gen_words.append(w2)
text = ' '.join(gen_words) text = ' '.join(gen_words)
text = text.replace(' ,', ',').replace(' .', '.') text = text.replace(' ,', ',').replace(' .', '.')
return text return text