1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
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]")
|
