ruby/Ruby-Cogs
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

Update filter

Browse source
This commit is contained in:
Valerie 2025-02-22 21:00:50 -05:00
parent 1520a4599f
commit f8095b4c37
1 changed files with 24 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
filter/filter.py
View file

@ -29,6 +29,8 @@ class FilterStorage:
data.append(value) data.append(value)
self.cursor.execute("REPLACE INTO filter_data (key, value) VALUES (?, ?)", (key, json.dumps(data))) self.cursor.execute("REPLACE INTO filter_data (key, value) VALUES (?, ?)", (key, json.dumps(data)))
self.conn.commit() self.conn.commit()
return True
return False
def remove(self, key, value): def remove(self, key, value):
data = self.get(key) data = self.get(key)
@ -36,6 +38,8 @@ class FilterStorage:
data.remove(value) data.remove(value)
self.cursor.execute("REPLACE INTO filter_data (key, value) VALUES (?, ?)", (key, json.dumps(data))) self.cursor.execute("REPLACE INTO filter_data (key, value) VALUES (?, ?)", (key, json.dumps(data)))
self.conn.commit() self.conn.commit()
return True
return False
def clear(self, key): def clear(self, key):
self.cursor.execute("DELETE FROM filter_data WHERE key = ?", (key,)) self.cursor.execute("DELETE FROM filter_data WHERE key = ?", (key,))
@ -83,6 +87,7 @@ class FilterCog(commands.Cog):
if word_check in message_content: if word_check in message_content:
try: try:
await message.delete() await message.delete()
await message.author.send(f"Your message was deleted because it contained a filtered word: `{word}`")
log_channel_id = await self.config.logChannelId() log_channel_id = await self.config.logChannelId()
if log_channel_id: if log_channel_id:
log_channel = message.guild.get_channel(log_channel_id) log_channel = message.guild.get_channel(log_channel_id)
@ -94,37 +99,46 @@ class FilterCog(commands.Cog):
except discord.HTTPException: except discord.HTTPException:
print("Failed to delete message due to an HTTP error") print("Failed to delete message due to an HTTP error")
@commands.hybrid_command() @commands.group(invoke_without_command=True)
async def filter(self, ctx):
"""Manage the filter system."""
await ctx.send_help(ctx.command)
@filter.command()
async def addword(self, ctx, word: str): async def addword(self, ctx, word: str):
"""Add a word to the filter list.""" """Add a word to the filter list."""
self.storage.add("words", word) if self.storage.add("words", word):
await ctx.send(f"Added `{word}` to the filter list.") await ctx.send(f"Added `{word}` to the filter list.")
else:
await ctx.send(f"`{word}` is already in the filter list.")
@commands.hybrid_command() @filter.command()
async def removeword(self, ctx, word: str): async def removeword(self, ctx, word: str):
"""Remove a word from the filter list.""" """Remove a word from the filter list."""
self.storage.remove("words", word) if self.storage.remove("words", word):
await ctx.send(f"Removed `{word}` from the filter list.") await ctx.send(f"Removed `{word}` from the filter list.")
else:
await ctx.send(f"`{word}` was not found in the filter list.")
@commands.hybrid_command() @filter.command()
async def listwords(self, ctx): async def listwords(self, ctx):
"""List all words in the filter.""" """List all words in the filter."""
words = self.storage.get("words") words = self.storage.get("words")
await ctx.send(f"Filtered words: {', '.join(words) if words else 'None'}") await ctx.send(f"Filtered words: {', '.join(words) if words else 'None'}")
@commands.hybrid_command() @filter.command()
async def clearwords(self, ctx): async def clearwords(self, ctx):
"""Clear all words from the filter list.""" """Clear all words from the filter list."""
self.storage.clear("words") self.storage.clear("words")
await ctx.send("All filtered words have been cleared.") await ctx.send("All filtered words have been cleared.")
@commands.hybrid_command() @filter.command()
async def setlogchannel(self, ctx, channel: discord.TextChannel): async def setlogchannel(self, ctx, channel: discord.TextChannel):
"""Set the logging channel.""" """Set the logging channel."""
await self.config.logChannelId.set(channel.id) await self.config.logChannelId.set(channel.id)
await ctx.send(f"Log channel set to {channel.mention}") await ctx.send(f"Log channel set to {channel.mention}")
@commands.hybrid_command() @filter.command()
async def listautomodrules(self, ctx): async def listautomodrules(self, ctx):
"""List all AutoMod rules in the server.""" """List all AutoMod rules in the server."""
guild = ctx.guild guild = ctx.guild