aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorGravatar Gareth Coles <[email protected]>2018-05-30 18:06:01 +0100
committerGravatar Gareth Coles <[email protected]>2018-05-30 18:06:01 +0100
commit7a667113a185274876703f79ff35c0101a5ed0e4 (patch)
tree2786b6019be1e5578bc6be1ea800429290b2b408
parentRMQ cog (diff)
Address review
-rw-r--r--bot/cogs/rmq.py15
1 files changed, 12 insertions, 3 deletions
diff --git a/bot/cogs/rmq.py b/bot/cogs/rmq.py
index 0140f0476..13cc7e8c2 100644
--- a/bot/cogs/rmq.py
+++ b/bot/cogs/rmq.py
@@ -1,10 +1,10 @@
+import json
import logging
import aio_pika
from discord.ext.commands import AutoShardedBot
-from bot.constants import RabbitMQ
-
+from bot.constants import RabbitMQ, Channels
log = logging.getLogger(__name__)
@@ -26,7 +26,7 @@ class RMQ:
host=RabbitMQ.host, port=RabbitMQ.port, login=RabbitMQ.username, password=RabbitMQ.password
)
- log.debug("Connected to RabbitMQ")
+ log.info("Connected to RabbitMQ")
self.channel = await self.rmq.channel()
self.queue = await self.channel.declare_queue("bot_events", durable=True)
@@ -49,6 +49,15 @@ class RMQ:
log.debug(f"Message: {message}")
log.debug(f"Data: {data}")
+ try:
+ data = json.loads(data)
+ await self.send_test(f"JSON: {data}")
+ except Exception:
+ await self.send_test(f"Non-JSON: {data}")
+
+ async def send_test(self, data):
+ await self.bot.get_channel(Channels.devtest).send(data)
+
def setup(bot):
bot.add_cog(RMQ(bot))