small changes

source/bot.py

@@ -177,6 +177,8 @@ def set_admin(message):
 
     """
     user_id = message.from_user.id
+
+    try:
         user = session.query(User).filter_by(telegram_id=user_id).first()
 
         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.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):
     """set admin status of user