fix: websocket callback errors

ansible · Oct 22, 2024 · 61a8f90 · 61a8f90
1 parent 764dcbf
commit 61a8f90
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/awxkit/awxkit/ws.py b/awxkit/awxkit/ws.py
@@ -205,7 +205,7 @@ def unsubscribe(self, wait=True, timeout=10):
         else:
             self._send(json.dumps(dict(groups={}, xrftoken=self.csrftoken)))
 
-    def _on_message(self, message):
+    def _on_message(self, message, *args):
         message = json.loads(message)
         log.debug('received message: {}'.format(message))
         if self._add_received_time:
@@ -230,13 +230,13 @@ def _update_subscription(self, job_id):
         self.subscribe(**subscription)
         self._should_subscribe_to_pending_job = False
 
-    def _on_open(self):
+    def _on_open(self, *args):
         self._ws_connected_flag.set()
 
-    def _on_error(self, error):
-        log.info('Error received: {}'.format(error))
+    def _on_error(self, *args):
+        log.info('Error received: {}'.format(args))
 
-    def _on_close(self):
+    def _on_close(self, *args):
         log.info('Successfully closed ws.')
         self._ws_closed = True