From 45c6a88acee5507215c51cc3529acb285790ebfd Mon Sep 17 00:00:00 2001 From: Dmitrii Morozov Date: Fri, 12 Apr 2024 13:56:20 +0200 Subject: Formatting of online message + exception handler for telebot --- tgbot.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) (limited to 'tgbot.py') 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()) -- cgit v1.2.3