Base stuff

Author: devoxin

.gitignore

@@ -102,3 +102,6 @@ venv.bak/
 
 # mypy
 .mypy_cache/
+
+# Custom Files
+config.json

bot.py

@@ -0,0 +1,33 @@
+import json
+import os
+import traceback
+
+import discord
+from discord.ext import commands
+
+with open('config.json') as f:
+ config = json.load(f)
+
+
+class Bot(commands.AutoShardedBot):
+ def __init__(self, **options):
+ super().__init__(**options)
+ self._load_cogs()
+
+ def _load_cogs(self):
+ for ext in os.listdir('extensions'):
+ if ext.endswith('.py'):
+ try:
+ self.load_extension(f'extensions.{ext[:-3]}')
+ except (ImportError, SyntaxError, discord.ClientException):
+ traceback.print_exc()
+
+ async def on_message(self, message):
+ if message.author.bot:
+ return
+
+ await self.process_commands(message)
+
+
+bot = Bot(command_prefix=config.get('prefixes'))
+bot.run(config.get('token'))

extensions/guild.py

@@ -0,0 +1,47 @@
+"""
+Provides log coverage of guild-related events.
+"""
+
+from discord.ext import commands
+from discord.ext.commands import Cog
+
+from utils.decorators import server_configured
+
+
+class GuildLog(commands.Cog):
+ def __init__(self, bot):
+ self.bot = bot
+ # On ready, create a registry of all invites in a guild.
+
+ @Cog.listener()
+ async def on_ready(self):
+ print('Connected to discord!!!')
+
+ @server_configured
+ @Cog.listener()
+ async def on_guild_update(self, before, after):
+ print('hi mom')
+
+ @Cog.listener()
+ async def on_member_join(self, member):
+ ...
+
+ @Cog.listener()
+ async def on_member_remove(self, member):
+ ...
+
+ @Cog.listener()
+ async def on_member_update(self, before, after):
+ ...
+
+ @Cog.listener()
+ async def on_member_ban(self, guild, member):
+ ...
+
+ @Cog.listener()
+ async def on_member_unban(self, guild, member):
+ ...
+
+
+def setup(bot):
+ bot.add_cog(GuildLog(bot))

extensions/messages.py

@@ -0,0 +1,23 @@
+"""
+Provides log coverage of message-related events.
+"""
+
+from discord.ext import commands
+from discord.ext.commands import Cog
+
+
+class MessageLog(Cog):
+ def __init__(self, bot):
+ self.bot = bot
+
+ @Cog.listener()
+ async def on_message_edit(self, before, after):
+ ...
+
+ @Cog.listener()
+ async def on_message_delete(self, message):
+ ...
+
+
+def setup(bot):
+ bot.add_cog(MessageLog(bot))

requirements.txt

@@ -0,0 +1,2 @@
+discord.py
+pymongo

utils/db.py

@@ -0,0 +1,7 @@
+from pymongo import MongoClient
+
+client = MongoClient('mongodb://localhost:27017/')
+db = client['modlog']
+
+config = db['config']
+invites = db['invitetracker']

utils/decorators.py

@@ -0,0 +1,34 @@
+from functools import wraps
+
+from discord import Guild
+
+from utils.db import config
+
+
+async def no_op(*args, **kwargs):
+ pass
+
+
+def server_configured(func):
+ @wraps(func)
+ async def wrapper(*args, **kwargs):
+ e = next((o for o in args if isinstance(o, Guild) or hasattr(o, 'guild')), None)
+
+ if not e:
+ print(f'[SUSPICIOUS] No entities passed to {func.__name__} are, or contain a `Guild`!')
+ return await no_op()
+
+ if hasattr(e, 'guild'):
+ e = e.guild
+
+ guild_id = str(e.id)
+ g_conf = config.find_one({'_id': guild_id})
+
+ if g_conf and g_conf.get('channel') is not None:
+ log_channel_id = int(g_conf['channel'])
+ log_channel = next(c for c in e.text_channels if c.id == log_channel_id)
+ return await func(*args, **kwargs, log_channel=log_channel)
+
+ return await no_op() # Guild has no config, or existing log channel.
+
+ return wrapper