bots: minor cleanup for trivia_quiz bot.

This commit is contained in:
Rhea Parekh 2018-06-11 19:02:23 +05:30 committed by showell
parent 0ea49c96ed
commit ba98220d44

View file

@ -38,7 +38,7 @@ class TriviaQuizHandler:
return
try:
quiz_payload = get_quiz_from_id(quiz_id, bot_handler)
except KeyError:
except (KeyError, TypeError):
bot_response = 'Invalid quiz id'
bot_handler.send_reply(message, bot_response)
return
@ -132,9 +132,7 @@ def get_quiz_from_payload(payload):
def generate_quiz_id(storage) -> str:
try:
quiz_num = storage.get('quiz_id')
except KeyError:
quiz_num = 0
except TypeError:
except (KeyError, TypeError):
quiz_num = 0
quiz_num += 1
quiz_num = quiz_num % (1000)