small changes

This commit is contained in:
Linus E 2022-05-09 16:57:17 +02:00
parent 3b4743f006
commit d47c869f0b

View File

@ -177,6 +177,8 @@ def set_admin(message):
""" """
user_id = message.from_user.id user_id = message.from_user.id
try:
user = session.query(User).filter_by(telegram_id=user_id).first() user = session.query(User).filter_by(telegram_id=user_id).first()
if not user.admin: if not user.admin:
@ -187,6 +189,10 @@ def set_admin(message):
bot.reply_to(message, "Type the telegram_id and boolean of admin attribute like <telegram_id> <value>") bot.reply_to(message, "Type the telegram_id and boolean of admin attribute like <telegram_id> <value>")
bot.register_next_step_handler(message, set_admin_handler) bot.register_next_step_handler(message, set_admin_handler)
except sqlalchemy.exc.IntegrityError:
session.rollback()
bot.reply_to(message, "Something went wrong.")
def set_admin_handler(message): def set_admin_handler(message):
"""set admin status of user """set admin status of user