Formatting of online message + exception handler for telebot

1 files changed, 10 insertions, 2 deletions

diff --git a/tgbot.py b/tgbot.py
index 8c4fa03..a414682 100755
--- a/tgbot.py
+++ b/tgbot.py
@@ -1,6 +1,6 @@
 #!/usr/bin/python3
 
-import os, time, telebot, asyncio, telebot.async_telebot, nest_asyncio
+import os, time, telebot, asyncio, telebot.async_telebot, nest_asyncio, sys, logging
 from FortniteStatusNotifier import *
 from Formatter import *
 from FortniteClient import *
@@ -34,7 +34,15 @@ class FortnitePresenceObserver(PresenceObserver):
     async def __notifyFriendPlaying(self, display_name: str, party_size: int):
         await send_message_to_all(formatFriendOnline(display_name, party_size))
 
-bot = telebot.async_telebot.AsyncTeleBot(os.environ["TELEBOT_BOT_TOKEN"])
+class ExceptionHandler(telebot.ExceptionHandler):
+    def handle(self, exception):
+        logging.error('Exception happened: {}'.format(str(exception)))
+        sys.exit('Exiting with telebot exception')
+        return True
+
+bot = telebot.async_telebot.AsyncTeleBot(
+    token=os.environ["TELEBOT_BOT_TOKEN"],
+    exception_handler=ExceptionHandler())
 userRepository = UserRepository('db.sqlite')
 fortniteStatusWrapper = FortniteStatusNotifier(FortniteStatusObserver())
 fortniteClient = FortniteClient(FortnitePresenceObserver())