Fix notifications plus minor improvements

1 files changed, 10 insertions, 5 deletions

diff --git a/FortniteEvents.py b/FortniteEvents.py
index 5be03f2..b5ae91c 100644
--- a/FortniteEvents.py
+++ b/FortniteEvents.py
@@ -9,14 +9,19 @@ class IncomingFriendRequest:
             await incoming_request.accept()
 
 class PresenceObserver:
-    async def update(self, display_name: str, playing: bool) -> None:
+    async def update(self, display_name: str, playing: bool, party_size: int) -> None:
         pass
 
 class FriendPresence:
-    async def on_event(before, after: fortnitepy.Presence, observers):
+    async def on_event(before, after: fortnitepy.Presence, observer: PresenceObserver):
         if before is not None and after is not None:
             if before.playing != after.playing:
                 print('FriendPresence changed for user {}, before {}, after {}'.format(after.friend.display_name, before.playing, after.playing))
-                for observer in observers:
-                    if isinstance(observer, PresenceObserver):
-                        await observer.update(after.friend.display_name, after.playing)
+                party_size: int = 0
+                if after.has_properties:
+                    party: fortnitepy.PresenceParty = after.party
+                    party_size = int(party.playercount)
+                await observer.update(
+                    after.friend.display_name, 
+                    after.playing, 
+                    party_size)