Catch telegram bot exceptions

F33RNI · Feb 28, 2023 · 763426b · 763426b
1 parent 029caa8
commit 763426b
Show file tree

Hide file tree

Showing 2 changed files with 98 additions and 65 deletions.
diff --git a/BotHandler.py b/BotHandler.py
@@ -19,6 +19,7 @@
 import logging
 import queue
 import threading
+import time
 
 import telegram
 from telegram import Update
@@ -67,21 +68,26 @@ def bot_start(self):
         Starts bot (blocking)
         :return:
         """
-        try:
-            # Build bot
-            application = ApplicationBuilder().token(self.settings['telegram']['api_key']) \
-                .write_timeout(30).read_timeout(30).build()
-            application.add_handler(CommandHandler(BOT_COMMAND_START, self.bot_command_start))
-            application.add_handler(CommandHandler(BOT_COMMAND_HELP, self.bot_command_help))
-            application.add_handler(CommandHandler(BOT_COMMAND_QUEUE, self.bot_command_queue))
-            application.add_handler(CommandHandler(BOT_COMMAND_GPT, self.bot_command_gpt))
-            application.add_handler(CommandHandler(BOT_COMMAND_DRAW, self.bot_command_draw))
-            application.add_handler(MessageHandler(filters.TEXT & (~filters.COMMAND), self.bot_read_message))
-
-            # Start bot
-            application.run_polling()
-        except Exception as e:
-            logging.error(e)
+        application = None
+        while True:
+            try:
+                # Build bot
+                application = ApplicationBuilder().token(self.settings['telegram']['api_key']) \
+                    .write_timeout(30).read_timeout(30).build()
+                application.add_handler(CommandHandler(BOT_COMMAND_START, self.bot_command_start))
+                application.add_handler(CommandHandler(BOT_COMMAND_HELP, self.bot_command_help))
+                application.add_handler(CommandHandler(BOT_COMMAND_QUEUE, self.bot_command_queue))
+                application.add_handler(CommandHandler(BOT_COMMAND_GPT, self.bot_command_gpt))
+                application.add_handler(CommandHandler(BOT_COMMAND_DRAW, self.bot_command_draw))
+                application.add_handler(MessageHandler(filters.TEXT & (~filters.COMMAND), self.bot_read_message))
+
+                # Start bot
+                application.run_polling()
+            except Exception as e:
+                logging.error('Telegram bot error! ' + str(e))
+            logging.info('Restarting bot polling after 5 seconds...')
+            time.sleep(5)
+            logging.info('Restarting bot polling ...')
 
     def reply_thread_start(self):
         """
@@ -105,29 +111,33 @@ async def create_request(self, request: str, update: Update, context: ContextTyp
         :param request_type:
         :return:
         """
-        user = update.message.from_user
-        chat_id = update.effective_chat.id
-
-        # Check queue length
-        if not self.requests_queue.full():
-            # Add request to queue
-            container = RequestResponseContainer.RequestResponseContainer(chat_id, user.full_name,
-                                                                          update.message.message_id, request=request,
-                                                                          request_type=request_type)
-            self.requests_queue.put(container)
-
-            # Send confirmation message
-            if self.settings['telegram']['show_queue_message']:
-                await context.bot.send_message(chat_id=chat_id, text=str(self.messages['queue_accepted'])
-                                               .format(str(self.requests_queue.qsize()
-                                                           + (1 if self.ai_handler.
-                                                              processing_container is not None else 0)),
-                                                       str(self.settings['telegram']['queue_max']))
-                                               .replace('\\n', '\n'))
-        # Queue overflow
-        else:
-            await context.bot.send_message(chat_id=chat_id,
-                                           text=str(self.messages['queue_overflow']).replace('\\n', '\n'))
+        try:
+            user = update.message.from_user
+            chat_id = update.effective_chat.id
+
+            # Check queue length
+            if not self.requests_queue.full():
+                # Add request to queue
+                container = RequestResponseContainer.RequestResponseContainer(chat_id, user.full_name,
+                                                                              update.message.message_id,
+                                                                              request=request,
+                                                                              request_type=request_type)
+                self.requests_queue.put(container)
+
+                # Send confirmation message
+                if self.settings['telegram']['show_queue_message']:
+                    await context.bot.send_message(chat_id=chat_id, text=str(self.messages['queue_accepted'])
+                                                   .format(str(self.requests_queue.qsize()
+                                                               + (1 if self.ai_handler.
+                                                                  processing_container is not None else 0)),
+                                                           str(self.settings['telegram']['queue_max']))
+                                                   .replace('\\n', '\n'))
+            # Queue overflow
+            else:
+                await context.bot.send_message(chat_id=chat_id,
+                                               text=str(self.messages['queue_overflow']).replace('\\n', '\n'))
+        except Exception as e:
+            logging.error('Error creating request / sending message! ' + str(e))
 
     async def bot_read_message(self, update: Update, context: ContextTypes.DEFAULT_TYPE):
         """
@@ -143,8 +153,11 @@ async def bot_read_message(self, update: Update, context: ContextTypes.DEFAULT_T
             await self.create_request(message, update, context, RequestResponseContainer.REQUEST_TYPE_CHATGPT)
         # No message
         else:
-            await context.bot.send_message(chat_id=chat_id,
-                                           text=str(self.messages['gpt_no_message']).replace('\\n', '\n'))
+            try:
+                await context.bot.send_message(chat_id=chat_id,
+                                               text=str(self.messages['gpt_no_message']).replace('\\n', '\n'))
+            except Exception as e:
+                logging.error('Error sending message! ' + str(e))
 
     async def bot_command_draw(self, update: Update, context: ContextTypes.DEFAULT_TYPE):
         """
@@ -157,19 +170,22 @@ async def bot_command_draw(self, update: Update, context: ContextTypes.DEFAULT_T
         chat_id = update.effective_chat.id
         logging.info('/draw command from user ' + str(user.full_name) + ' request: ' + ' '.join(context.args))
 
-        if len(context.args) > 0:
-            # Combine all arguments to text
-            request = str(' '.join(context.args)).strip()
-            if len(request) > 0:
-                await self.create_request(request, update, context, RequestResponseContainer.REQUEST_TYPE_DALLE)
+        try:
+            if len(context.args) > 0:
+                # Combine all arguments to text
+                request = str(' '.join(context.args)).strip()
+                if len(request) > 0:
+                    await self.create_request(request, update, context, RequestResponseContainer.REQUEST_TYPE_DALLE)
+                # No text
+                else:
+                    await context.bot.send_message(chat_id=chat_id,
+                                                   text=str(self.messages['draw_no_text']).replace('\\n', '\n'))
             # No text
             else:
                 await context.bot.send_message(chat_id=chat_id,
                                                text=str(self.messages['draw_no_text']).replace('\\n', '\n'))
-        # No text
-        else:
-            await context.bot.send_message(chat_id=chat_id,
-                                           text=str(self.messages['draw_no_text']).replace('\\n', '\n'))
+        except Exception as e:
+            logging.error('Error sending message! ' + str(e))
 
     async def bot_command_gpt(self, update: Update, context: ContextTypes.DEFAULT_TYPE):
         """
@@ -182,19 +198,22 @@ async def bot_command_gpt(self, update: Update, context: ContextTypes.DEFAULT_TY
         chat_id = update.effective_chat.id
         logging.info('/gpt command from user ' + str(user.full_name) + ' request: ' + ' '.join(context.args))
 
-        if len(context.args) > 0:
-            # Combine all arguments to text
-            request = str(' '.join(context.args)).strip()
-            if len(request) > 0:
-                await self.create_request(request, update, context, RequestResponseContainer.REQUEST_TYPE_CHATGPT)
+        try:
+            if len(context.args) > 0:
+                # Combine all arguments to text
+                request = str(' '.join(context.args)).strip()
+                if len(request) > 0:
+                    await self.create_request(request, update, context, RequestResponseContainer.REQUEST_TYPE_CHATGPT)
+                # No text
+                else:
+                    await context.bot.send_message(chat_id=chat_id,
+                                                   text=str(self.messages['gpt_no_text']).replace('\\n', '\n'))
             # No text
             else:
                 await context.bot.send_message(chat_id=chat_id,
                                                text=str(self.messages['gpt_no_text']).replace('\\n', '\n'))
-        # No text
-        else:
-            await context.bot.send_message(chat_id=chat_id,
-                                           text=str(self.messages['gpt_no_text']).replace('\\n', '\n'))
+        except Exception as e:
+            logging.error('Error sending message! ' + str(e))
 
     async def bot_command_queue(self, update: Update, context: ContextTypes.DEFAULT_TYPE):
         """
@@ -210,7 +229,11 @@ async def bot_command_queue(self, update: Update, context: ContextTypes.DEFAULT_
 
         # Queue is empty
         if self.requests_queue.empty() and processing_container is None:
-            await context.bot.send_message(chat_id=chat_id, text=str(self.messages['queue_empty']).replace('\\n', '\n'))
+            try:
+                await context.bot.send_message(chat_id=chat_id,
+                                               text=str(self.messages['queue_empty']).replace('\\n', '\n'))
+            except Exception as e:
+                logging.error('Error sending message! ' + str(e))
         else:
             i = 0
             message = ''
@@ -234,8 +257,11 @@ async def bot_command_queue(self, update: Update, context: ContextTypes.DEFAULT_
                 message += ': ' + processing_container.request + '\n\n'
 
             # Send queue stats
-            await context.bot.send_message(chat_id=chat_id, text=str(self.messages['queue_stats'])
-                                           .format(message).replace('\\n', '\n'))
+            try:
+                await context.bot.send_message(chat_id=chat_id, text=str(self.messages['queue_stats'])
+                                               .format(message).replace('\\n', '\n'))
+            except Exception as e:
+                logging.error('Error sending message! ' + str(e))
 
     async def bot_command_help(self, update: Update, context: ContextTypes.DEFAULT_TYPE):
         """
@@ -249,7 +275,11 @@ async def bot_command_help(self, update: Update, context: ContextTypes.DEFAULT_T
         logging.info('/help command from user ' + str(user.full_name))
 
         # Send help message
-        await context.bot.send_message(chat_id=chat_id, text=str(self.messages['help_message']).replace('\\n', '\n'))
+        try:
+            await context.bot.send_message(chat_id=chat_id,
+                                           text=str(self.messages['help_message']).replace('\\n', '\n'))
+        except Exception as e:
+            logging.error('Error sending message! ' + str(e))
 
     async def bot_command_start(self, update: Update, context: ContextTypes.DEFAULT_TYPE):
         """
@@ -263,8 +293,11 @@ async def bot_command_start(self, update: Update, context: ContextTypes.DEFAULT_
         logging.info('/start command from user ' + str(user.full_name))
 
         # Send start message
-        await context.bot.send_message(chat_id=chat_id, text=str(self.messages['start_message'])
-                                       .format(TELEGRAMUS_VERSION).replace('\\n', '\n'))
+        try:
+            await context.bot.send_message(chat_id=chat_id, text=str(self.messages['start_message'])
+                                           .format(TELEGRAMUS_VERSION).replace('\\n', '\n'))
+        except Exception as e:
+            logging.error('Error sending message! ' + str(e))
 
         # Send help message
         await self.bot_command_help(update, context)

diff --git a/main.py b/main.py
@@ -27,7 +27,7 @@
 import Authenticator
 import BotHandler
 
-TELEGRAMUS_VERSION = 'beta_1.7.2'
+TELEGRAMUS_VERSION = 'beta_1.7.3'
 
 # Logging level (INFO for debug, WARN for release)
 LOGGING_LEVEL = logging.INFO