Project initHEADmain

1 files changed, 136 insertions, 0 deletions

diff --git a/cogs/ButtBot.py b/cogs/ButtBot.py
new file mode 100644
index 0000000..46373ee
--- /dev/null
+++ b/cogs/ButtBot.py
@@ -0,0 +1,136 @@
+from discord.ext import commands
+import discord
+import importlib
+import utils
+importlib.reload(utils)
+import random
+
+
+class ButtBot(commands.Cog):
+    def __init__(self, bot):
+        self.bot = bot
+        global extension_name
+        extension_name = "[ButtBot] "
+
+    def channel_check(ctx):
+        return ctx.message.channel.id in [458613595060830238]
+
+    @commands.Cog.listener()
+    @commands.check(channel_check)
+    async def on_message(self, ctx, *args):
+        if ctx.author == self.bot.user:
+            return
+
+        butt_settings = await utils.sql('SELECT buttbot_on, buttbot_freq FROM buttbot WHERE guild_id = %s and bot_name = %s;', (ctx.guild.id, self.bot.user.name))
+        if not butt_settings:
+            return
+
+        dice_roll = random.randint(1, butt_settings[0]["buttbot_freq"])
+        word_count = len(ctx.content.split(" "))
+        if dice_roll == 1 and word_count > 3 and butt_settings[0]['buttbot_on'] == 1:
+            lst_words = ctx.content.split(" ")
+            random_key = random.randint(1, len(lst_words)) - 1
+            lst_words[random_key] = "butt"
+            new_string = " ".join(lst_words)
+            await ctx.channel.send(content=new_string)
+
+    @commands.group(
+        aliases=[],
+        application_command_meta=commands.ApplicationCommandMeta(
+            options=[
+                discord.ApplicationCommandOption(
+                    name="on",
+                    description="Turn ButtBot On",
+                    type=discord.ApplicationCommandOptionType.string,
+                    required=True,
+                ),
+                discord.ApplicationCommandOption(
+                    name="off",
+                    description="Turn ButtBot Off",
+                    type=discord.ApplicationCommandOptionType.string,
+                    required=True,
+                ),
+                discord.ApplicationCommandOption(
+                    name="frequency",
+                    description="Change how often ButtBot fires",
+                    type=discord.ApplicationCommandOptionType.integer,
+                    required=True,
+                )
+            ],
+        )
+    )
+    async def buttbot(self, ctx):
+        """Allow Synthy to add more butts to your conversations."""
+        prefix = await self.bot.get_prefix(ctx.message)
+
+        butt_settings = await utils.sql('SELECT buttbot_on, buttbot_freq FROM buttbot WHERE guild_id = %s and bot_name = %s;', (ctx.guild.id, self.bot.user.name))
+
+        if not butt_settings:
+            butt_settings = [{'buttbot_on': 0, 'buttbot_freq': 300}]
+
+        butt_settings[0]['buttbot_on'] = "On" if butt_settings[0]['buttbot_on'] == 1 else "Off"
+
+        emb = await utils.embed(ctx, f"Commands for `{prefix[2]}buttbot`", "ButtBot sometimes reposts a message but replaces a word with `butt`")
+        emb = await utils.field(emb, f"{prefix[2]}buttbot off/on", f"Toggle ButtBot on or off. (Off by default, currently: {butt_settings[0]['buttbot_on']})")
+        emb = await utils.field(emb, f"{prefix[2]}buttbot freq [number]", f"This number determines how often ButtBot will work. A lower number means more butts. (300 by default, currently: {butt_settings[0]['buttbot_freq']})")
+        await ctx.send(embed=emb)
+
+    @buttbot.command(aliases=[], application_command_meta=commands.ApplicationCommandMeta(options=[]))
+    @commands.has_permissions(administrator=True)
+    async def on(self, ctx):
+        """Turn ButtBot On"""
+        await utils.sql('INSERT INTO buttbot (guild_id, buttbot_freq, buttbot_on, bot_name) VALUES (%s, %s, %s, %s) ON CONFLICT (guild_id) DO UPDATE SET buttbot_on = %s', (ctx.guild.id, 300, True, self.bot.user.name, True,))
+        emb = await utils.embed(ctx, "ButtBot", f"ButtBot is now **on** for this server.")
+        await ctx.send(embed=emb)
+
+    @buttbot.command(aliases=[], application_command_meta=commands.ApplicationCommandMeta(options=[]))
+    @commands.has_permissions(administrator=True)
+    async def off(self, ctx):
+        """Turn ButtBot Off"""
+        await utils.sql('INSERT INTO buttbot (guild_id, buttbot_freq, buttbot_on, bot_name) VALUES (%s, %s, %s, %s) ON CONFLICT (guild_id) DO UPDATE SET buttbot_on = %s', (ctx.guild.id, 300, False, self.bot.user.name, False,))
+        emb = await utils.embed(ctx, "ButtBot", f"ButtBot is now **off** for this server.")
+        await ctx.send(embed=emb)
+
+    @buttbot.command(
+        aliases=[],
+        application_command_meta=commands.ApplicationCommandMeta(
+            options=[
+                discord.ApplicationCommandOption(
+                    name="frequency",
+                    description="Change how often ButtBot fires",
+                    type=discord.ApplicationCommandOptionType.integer,
+                    required=True,
+                )
+            ],
+        )
+    )
+    @commands.has_permissions(administrator=True)
+    async def freq(self, ctx, frequency):
+        """Change how often ButtBot fires"""
+        try:
+            buttbot_freq = int(frequency)
+            if buttbot_freq <= 0:
+                emb = await utils.embed(ctx, "ButtBot Frequency", "The frequency must be a number above 0.", colour="red")
+                await ctx.send(embed=emb)
+                return
+        except ValueError:
+            emb = await utils.embed(ctx, "ButtBot Frequency", "Please provide a valid number", colour="red")
+            await ctx.send(embed=emb)
+            return
+
+
+        await utils.sql('INSERT INTO buttbot (guild_id, buttbot_freq, buttbot_on, bot_name) VALUES (%s, %s, %s, %s) ON CONFLICT (guild_id) DO UPDATE SET buttbot_freq = %s;',
+                        (ctx.guild.id, buttbot_freq, True, self.bot.user.name, buttbot_freq,))
+
+        emb = await utils.embed(ctx, "ButtBot Frequency", f"The ButtBot frequency has been set to `{buttbot_freq}`. Remember that the lower this is set, the more butts you'll see.")
+        await ctx.send(embed=emb)
+
+
+def setup(bot):
+    print("INFO: Loading [ButtBot]... ", end="")
+    bot.add_cog(ButtBot(bot))
+    print("Done!")
+
+
+def teardown(bot):
+    print("INFO: Unloading [ButtBot]")