Implement running Discord and 4chan gathering simultaneously

sources/dis.py

@@ -0,0 +1,62 @@
+#!/usr/bin/env python
+from operator import attrgetter
+
+import discord
+
+import db
+import util
+
+ATTRMAP = {
+    "msg": "content",
+    "msg_id": "id",
+    "nick": "author.name",
+    "host": "author.discriminator",
+    "ident": "author.nick",
+    "time": "created_at",
+    "channel": "channel.name",
+    "channel_nsfw": "channel.nsfw",
+    "bot": "author.bot",
+    "user_id": "author.id",
+    "channel_id": "channel.id",
+    "net": "author.guild.name",
+    "net_id": "author.guild.id",
+    "guild_member_count": "author.guild.member_count",
+    "channel_category": "channel.category.name",
+    "channel_category_id": "channel.category.id",
+    "channel_category_nsfw": "channel.category.nsfw",
+}
+
+
+class DiscordClient(discord.Client):
+    def __init__(self, *args, **kwargs):
+        self.logger = None
+        self.did_something = False
+        name = self.__class__.__name__
+        self.log = util.get_logger(name)
+
+        super().__init__(*args, **kwargs)
+
+    async def on_ready(self):
+        self.log.info("Discord connection established.")
+
+    def recurse_dict(self, obj):
+        to_return = {}
+        for key, mapped in ATTRMAP.items():
+            try:
+                to_return[key] = attrgetter(mapped)(obj)
+            except AttributeError:
+                continue
+
+        return to_return
+
+    async def on_message(self, message):
+        if not message.content:
+            return
+
+        a = self.recurse_dict(message)
+        a["ts"] = a["time"].isoformat()
+        del a["time"]
+        a["type"] = "msg"
+        a["src"] = "dis"
+
+        db.store_message(a)