This commit is contained in:
bakatrouble 2019-11-27 22:41:31 +03:00
parent 4e58ee1fc6
commit b821e96084

View File

@ -49,11 +49,12 @@ class MTG(object):
if second_word is not None:
return first_word, second_word
elif first_word is not None:
if ('.' + '+' + first_word in self.MC) and len(self.MC['.' + '+' + first_word]) != 0:
second_word = random.choice(self.MC['.' + '+' + first_word])
elif ('.' + '+' + first_word.capitalize() in self.MC) and \
len(self.MC['.' + '+' + first_word.capitalize()]) != 0:
second_word = random.choice(self.MC['.' + '+' + first_word.capitalize()])
word = '.' + '+' + first_word
word_cap = '.' + '+' + first_word.capitalize()
if word in self.MC and len(self.MC[word]):
second_word = random.choice(self.MC[word])
elif word_cap in self.MC and len(self.MC[word_cap]):
second_word = random.choice(self.MC[word_cap])
else:
return self.choose_start_words()
return first_word, second_word