From d47c869f0ba1ecc64abf397d67be1b9f6f10b142 Mon Sep 17 00:00:00 2001
From: Linus E <75929322+Rripped@users.noreply.github.com>
Date: Mon, 9 May 2022 16:57:17 +0200
Subject: [PATCH] small changes

---
 source/bot.py | 20 +++++++++++++-------
 1 file changed, 13 insertions(+), 7 deletions(-)

diff --git a/source/bot.py b/source/bot.py
index 737c284..d842277 100644
--- a/source/bot.py
+++ b/source/bot.py
@@ -177,15 +177,21 @@ def set_admin(message):
 
     """
     user_id = message.from_user.id
-    user = session.query(User).filter_by(telegram_id=user_id).first()
 
-    if not user.admin:
-        bot.reply_to(message, "Error: Admin rights are required to change admin rights of users.")
-        return
+    try:
+        user = session.query(User).filter_by(telegram_id=user_id).first()
 
-    if user.admin:
-        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)
+        if not user.admin:
+            bot.reply_to(message, "Error: Admin rights are required to change admin rights of users.")
+            return
+
+        if user.admin:
+            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):