diff --git a/source/bot.py b/source/bot.py index d842277..fe03868 100644 --- a/source/bot.py +++ b/source/bot.py @@ -214,12 +214,12 @@ def set_admin_handler(message): if user is None: bot.reply_to(message, "Error: User with entered telegram id is not registered.") return - + try: - if admin == "True" or admin == "true": + if admin in ("True", "true"): user.admin = True - elif admin == "False" or admin == "false": + elif admin in ("False", "false"): user.admin = False session.commit() @@ -229,10 +229,6 @@ def set_admin_handler(message): session.rollback() bot.reply_to(message, "Something went wrong") - - - - @bot.message_handler(commands=['scoreboard', 'Scoreboard']) def send_scoreboard(message):