added requested changes

olijeffers0n · Jun 7, 2023 · d8067bd · d8067bd
1 parent 6b3509e
commit d8067bd
Showing 1 changed file with 10 additions and 7 deletions.
diff --git a/rustplus/api/remote/rustws.py b/rustplus/api/remote/rustws.py
@@ -82,15 +82,18 @@ async def connect(
                     self.connection = await connect(address, close_timeout=0, ping_interval=None)
                     self.connected_time = time.time()
 
-                    if asyncio.iscoroutinefunction(self.on_success):
-                        await self.on_success()
-                    else:
-                        try:
-                            self.on_success()
-                        except TypeError:
-                            pass
+                    try:
+                        if self.on_success is not None:
+                            if asyncio.iscoroutinefunction(self.on_success):
+                                await self.on_success()
+                            else:
+                                self.on_success()
+
+                    except Exception as e:
+                        self.logger.warning(e)
 
                     break
+
                 except Exception as exception:
                     print_error = True