Add exclude muted members option

cogs/configuration.py

@@ -89,6 +89,31 @@ async def setup(self, ctx):
  )
  )
 
+ @commands.guild_only()
+ @commands.command(
+ description="Prohibit muted members from creating a ticket",
+ usage="excludemuted",
+ )
+ async def excludemuted(self, ctx):
+ if (await ctx.message.member.guild_permissions()).administrator is False:
+ raise commands.MissingPermissions(["administrator"])
+
+ data = await tools.get_data(self.bot, ctx.guild.id)
+ toset = True
+ if data[15] is True:
+ toset = False
+
+ async with self.bot.pool.acquire() as conn:
+ await conn.execute("UPDATE data SET excludemuted=$1 WHERE guild=$2", toset, ctx.guild.id)
+
+ await ctx.send(
+ Embed(
+ "Block timed out members is now "
+ f"`{'Enabled' if toset is True else 'Disabled'}`.",
+ )
+ )
+
+
  @commands.guild_only()
  @commands.command(
  description="Change the prefix or view the current prefix.",
@@ -463,6 +488,7 @@ async def viewconfig(self, ctx):
  embed.add_field("Advanced Logging", "Enabled" if data[7] is True else "Disabled")
  embed.add_field("Anonymous Messaging", "Enabled" if data[10] is True else "Disabled")
  embed.add_field("Command Only", "Enabled" if data[11] is True else "Disabled")
+ embed.add_field("Exclude Muted", "Enabled" if data[15] is True else "Disabled")
  embed.add_field("Ticket Creation", "Enabled" if toggle is None else f"Disabled ({toggle})")
  embed.add_field("Greeting Message", "*Not set*" if greeting is None else greeting, False)
  embed.add_field("Closing Message", "*Not set*" if closing is None else closing, False)

cogs/direct_message.py

@@ -51,6 +51,16 @@ async def send_mail(self, message, guild):
  ErrorEmbed("That server has blacklisted you from sending a message there.")
  )
  return
+
+ if data[15] is True and member.timed_out_until is not None:
+ embed = ErrorEmbed(
+ "Ticket Not Created",
+ "You can't create a ticket because you are timed out.",
+ timestamp=True,
+ )
+ embed.set_footer(f"{guild.name} | {guild.id}", guild.icon_url)
+ await message.channel.send(embed)
+ return
 
  channel = None
  for text_channel in await guild.text_channels():

migrations/2022-05-01-135059_initialize_db/up.sql

@@ -12,6 +12,7 @@ CREATE TABLE data
  blacklist bigint[] NOT NULL,
  anonymous boolean NOT NULL,
  commandonly boolean NOT NULL,
+ excludemuted boolean NOT NULL,
  PRIMARY KEY (guild)
 );
 

migrations/2024-04-29-192931_add_exclude_muted_members/down.sql

@@ -0,0 +1 @@
+ALTER TABLE data DROP COLUMN excludemuted;

migrations/2024-04-29-192931_add_exclude_muted_members/up.sql

@@ -0,0 +1 @@
+ALTER TABLE data ADD COLUMN excludemuted boolean;

utils/tools.py

@@ -181,7 +181,7 @@ async def get_data(bot, guild):
  return res
 
  return await conn.fetchrow(
- "INSERT INTO data VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13) "
+ "INSERT INTO data VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14) "
  "RETURNING *",
  guild,
  None,
@@ -196,6 +196,7 @@ async def get_data(bot, guild):
  False,
  False,
  None,
+ False,
  )