on_notification ws event

Author: Katistic

vrcpy/wss.py

@@ -19,7 +19,8 @@ def __ws_message(self, ws, message):
         switch = {
             "friend-location": self.__ws_friend_location,
             "friend-active": self.__ws_friend_active,
-            "friend-offline": self.__ws_friend_offline
+            "friend-offline": self.__ws_friend_offline,
+            "notification": self.__ws_notification
         }
 
         if message["type"] in switch:
@@ -70,6 +71,9 @@ def on_friend_offline(self, friend):
     def on_friend_location(self, friend):
         pass
 
+    def on_notification(self, notification):
+        pass
+
     # WS handles
 
     def __ws_friend_active(self, content):
@@ -84,6 +88,9 @@ def __ws_friend_location(self, content):
     def __ws_friend_offline(self, content):
         self.on_friend_offline(self.fetch_user_by_id(content["userId"]))
 
+    def __ws_notification(self, content):
+        self.on_notification(None)
+
     # Internal Client overwrites
 
     def login(self, username, password):
@@ -120,6 +127,9 @@ async def on_friend_offline(self, friend):
     async def on_friend_location(self, friend):
         pass
 
+    async def on_notification(self, notification):
+        pass
+
     # WS handles
 
     async def __ws_friend_active(self, content):
@@ -134,6 +144,9 @@ async def __ws_friend_location(self, content):
     async def __ws_friend_offline(self, content):
         await self.on_friend_offline(await self.fetch_user_by_id(content["userId"]))
 
+    async def __ws_notification(self, content):
+        await self.on_notification(None)
+
     # Internal Client overwrites
 
     async def login(self, username, password):