From 7c688c1c9b494fee205088bf501a91eed038c0be Mon Sep 17 00:00:00 2001 From: retoor Date: Tue, 18 Feb 2025 19:09:51 +0100 Subject: [PATCH] Upgrade. --- src/snekbot/bot.py | 7 +++++-- src/snekbot/rpc.py | 5 +---- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/src/snekbot/bot.py b/src/snekbot/bot.py index 4f04b7e..99935cd 100644 --- a/src/snekbot/bot.py +++ b/src/snekbot/bot.py @@ -36,15 +36,17 @@ class Bot: async def run(self, reconnect=True): while True: + try: await self.run_once() except Exception as ex: print(ex) - + await asyncio.sleep(1) + if not reconnect: break - await asyncio.sleep(1) + def has_joined(self, channel_uid): return channel_uid in self.joined @@ -103,6 +105,7 @@ class Bot: await self.on_message(data.username, data.user_nick, data.channel_uid, data.message) except AttributeError as ex: logger.exception(ex) + raise except Exception as ex: raise ex diff --git a/src/snekbot/rpc.py b/src/snekbot/rpc.py index 9546b9d..c1ca12c 100644 --- a/src/snekbot/rpc.py +++ b/src/snekbot/rpc.py @@ -61,10 +61,7 @@ class RPC: async def method(*args, **kwargs): self.current_call_id = str(uuid.uuid4()) payload = dict(method=name, args=args, kwargs=kwargs, callId=self.current_call_id) - try: - await self.ws.send_json(payload) - except Exception as ex: - logger.exception(ex) + await self.ws.send_json(payload) async def returner(): while True: