hook on connect and disconnect

Author: Katistic

examples/async/WebSocket Client.py

@@ -31,6 +31,12 @@ async def on_notification(self, notification):
     async def on_unhandled_event(self, event, content):
         print("Recieved unhandled event '{}'.".format(event))
 
+    async def on_connect(self):
+        print("Connected to wss pipeline.")
+
+    async def on_disconnect(self):
+        print("Disconnected from wss pipeline.")
+
     async def wait_loop(self):
         await self.login2fa(input("Username: "), input("Password: "), input("2FA Code: "), True)
 

examples/sync/WebSocket Client.py

@@ -30,6 +30,12 @@ def on_notification(self, notification):
     def on_unhandled_event(self, event, content):
         print("Recieved unhandled event '{}'.".format(event))
 
+    def on_connect(self):
+        print("Connected to wss pipeline.")
+
+    def on_disconnect(self):
+        print("Disconnected from wss pipeline.")
+
     def wait_loop(self):
         self.login2fa(input("Username: "), input("Password: "), input("2FA Code: "), True)