From 08cd4e0b8e6473dcf8dc18915b00aafc86f13215 Mon Sep 17 00:00:00 2001 From: Dmitrii Morozov Date: Wed, 10 Apr 2024 20:52:53 +0200 Subject: Fix notifications plus minor improvements --- tgbot.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) (limited to 'tgbot.py') diff --git a/tgbot.py b/tgbot.py index 5422cb6..539a1c5 100755 --- a/tgbot.py +++ b/tgbot.py @@ -19,21 +19,20 @@ class FortnitePresenceObserver(PresenceObserver): # Map name -> last seen not playing timestamp seconds statuses = {} - async def update(self, display_name: str, playing: bool) -> None: - print('FortnitePresenceObserver: {} playing = {}'.format(display_name, playing)) + async def update(self, display_name: str, playing: bool, party_size: int) -> None: if playing: if not display_name in self.statuses: - await self.__notifyFriendPlaying(display_name) + await self.__notifyFriendPlaying(display_name, party_size) self.statuses[display_name] = time.time() else: diff = time.time() - self.statuses[display_name] if diff > 60 * 60: # 60 minutes - self.__notifyFriendPlaying(display_name) + self.__notifyFriendPlaying(display_name, party_size) else: self.statuses[display_name] = time.time() - async def __notifyFriendPlaying(self, display_name: str): - await send_message_to_all(formatFriendOnline(display_name)) + 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"]) userRepository = UserRepository('db.sqlite') -- cgit v1.2.3