87 lines
4.6 KiB
Python
87 lines
4.6 KiB
Python
import discord
|
|
import sqlite3
|
|
from discord.ext import commands
|
|
|
|
DBFILENAME = "bot.db"
|
|
|
|
class Management(commands.Cog):
|
|
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
|
|
#Commands
|
|
@commands.command(aliases=["purge", "clear"], brief="Cleans messages from channel", description="Clears either 1 or a user set amount of messages from the channel the command was called within.")
|
|
@commands.has_permissions(manage_messages=True)
|
|
async def clean(self, ctx, *, amount=1):
|
|
await ctx.channel.purge(limit=int(amount) + 1)
|
|
|
|
@commands.command(brief="Kicks user", description="Removes a specified user from the server. They can re-join if they have an invite link.")
|
|
@commands.has_permissions(kick_members=True)#@commands.has_any_role("Owner", "Moderator")
|
|
async def kick(self, ctx, member : discord.Member, *, reason=None):
|
|
await member.kick(reason=reason)
|
|
|
|
@commands.command(brief="Ban user", description="Removes a specified user from the server. They will be unable to re-join until they have been unbanned.")
|
|
@commands.has_permissions(ban_members=True)
|
|
async def ban(self, ctx, member : discord.Member, *, reason=None):
|
|
await member.ban(reason=reason)
|
|
|
|
@commands.command(brief="Unban user", description="Removes a specified user from the servers list of banned users. Allowing them to re-join the server with an invite link.")
|
|
@commands.has_permissions(ban_members=True)
|
|
async def unban(self, ctx, *, member):
|
|
banned_users = await ctx.guild.bans()
|
|
member_name, member_discriminator = member.split("#")
|
|
for ban_entry in banned_users:
|
|
user = ban_entry.user
|
|
if (user.name, user.discriminator) == (member_name, member_discriminator):
|
|
await ctx.guild.unban(user)
|
|
|
|
@commands.command(aliases=["a"], brief="Echo messsage", description="The bot will echo the text following the command also deleting the message used to call the commmand.")
|
|
@commands.has_permissions(manage_messages = True)
|
|
async def announcement(self, ctx, *, message=""):
|
|
await ctx.message.channel.purge(limit=1)
|
|
await ctx.send(f"{message}")
|
|
|
|
@commands.command(brief="Warn user", description="Sends a warning message with reasoning to the specified user.")
|
|
@commands.has_permissions(manage_messages = True)
|
|
async def warn(self, ctx, member : discord.Member, *, reason=None):
|
|
await ctx.message.channel.purge(limit=1)
|
|
#await member.send(f"**WARNING** from {ctx.message.author.mention}:\n{reason}")
|
|
embed = discord.Embed(title=f"**WARNING** from", description=f"{ctx.message.author.mention}")
|
|
embed.add_field(name="Reason:", value=(f"{reason}"))
|
|
embed.colour = (0xff0000)
|
|
await member.send(content=None, embed=embed)
|
|
|
|
@commands.command(brief="User messages", description="Check specified users profanity rating.")
|
|
@commands.has_permissions(manage_messages=True)
|
|
async def messages(self, ctx, member: discord.Member):
|
|
db = sqlite3.connect(DBFILENAME)#Open the database
|
|
cursor = db.cursor()
|
|
cursor.execute("SELECT total FROM users WHERE user=?", (member.id,))#Select mentioned user to see if the value exists
|
|
totalResult = cursor.fetchone()
|
|
totalResult = int(totalResult[0])
|
|
cursor.execute("SELECT profanity FROM users WHERE user=?", (member.id,))#Select mentioned user to see if the value exists
|
|
profanityResult = cursor.fetchone()
|
|
profanityResult = int(profanityResult[0])
|
|
profanityPercent = round((profanityResult / totalResult * 100), 2)
|
|
cursor.execute("SELECT message FROM messages WHERE bool=? AND user=?", (1, member.id))
|
|
messageResult = cursor.fetchall()
|
|
db.close()
|
|
if not messageResult: #if messageResult is empty
|
|
allMessages = "No messages to show"
|
|
else:
|
|
messageResult = messageResult[-5:]#Only select the last 5 messages which contained profanity
|
|
allMessages = ""
|
|
for i in range(0, len(messageResult)):
|
|
allMessages += str(messageResult[i])
|
|
allMessages = allMessages.replace("(", "")
|
|
allMessages = allMessages.replace(")", "")
|
|
allMessages = allMessages.replace(",", " ")
|
|
embed = discord.Embed(title="Messages", description="")
|
|
embed.add_field(name="Total", value=f"{totalResult}")
|
|
embed.add_field(name="Profanity", value=f"{profanityResult}")
|
|
embed.add_field(name="Profanity %", value=f"{profanityPercent}")
|
|
embed.add_field(name="Recent rude messages", value=f"{allMessages}")
|
|
await ctx.send(content=None, embed=embed)
|
|
|
|
def setup(bot):
|
|
bot.add_cog(Management(bot))
|