This commit is contained in:
bakatrouble 2019-11-27 22:36:29 +03:00
parent e12f68002b
commit 4e58ee1fc6

View File

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