Fix notifications plus minor improvements

1 files changed, 5 insertions, 6 deletions

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')