TotalFreedomBot/commands/moderation.py

import datetime

import discord
from discord.ext import commands

from checks import is_mod_or_has_perms
from functions import read_json, write_json

muted_role_id = 769659653121900546


class Moderation(commands.Cog):
    def __init__(self, bot):
        self.bot = bot
        self.moderator_role_id = 769659653129896023

    @commands.command()
    @commands.has_permissions(kick_members=True)
    async def kick(self, ctx, user: discord.Member, *, reason="No reason specified"):
        """Kicks a user from the guild."""
        await user.kick(reason=f'{reason}**  **by: {ctx.author.name}')
        await ctx.send(
            embed=discord.Embed(embed=f'{user.name} has been kicked by: {ctx.author.name} for reason: {reason}'))
        print(
            f"[{datetime.datetime.utcnow().replace(microsecond=0)} INFO]: [Moderation] Kicked {user.name} from {ctx.guild.name}")

    @commands.command(aliases=['gtfo'])
    @commands.has_permissions(ban_members=True)
    async def ban(self, ctx, user: discord.Member, *, reason="No reason specified"):
        """Bans a user from the guild."""
        await user.ban(reason=f'{reason} || by: {ctx.author.name}', delete_message_days=0)
        await ctx.send(
            embed=discord.Embed(description=f'{user.name} has been banned by: {ctx.author.name} for reason: {reason}'))
        print(
            f"[{datetime.datetime.utcnow().replace(microsecond=0)} INFO]: [Moderation] Banned {user.name} from {ctx.guild.name}")

    @commands.command()
    @commands.has_permissions(ban_members=True)
    async def unban(self, ctx, user: discord.User, *, reason="No reason specified"):
        """Unbans a user from the guild."""
        await ctx.guild.unban(user, reason=f'{reason} || by: {ctx.author.name}')
        await ctx.send(embed=discord.Embed(
            description=f'{user.name} has been unbanned by: {ctx.author.name} for reason: {reason}'))
        print(
            f"[{datetime.datetime.utcnow().replace(microsecond=0)} INFO]: [Moderation] Banned {user.name} from {ctx.guild.name}")

    @commands.command(aliases=['massdelete', 'purge'])
    @commands.has_permissions(manage_messages=True)
    async def prune(self, ctx, msgs):
        """Purge messages from a channel."""
        channel = ctx.channel
        await channel.purge(limit=(int(msgs) + 1))
        await ctx.send(embed=discord.Embed(description=f'{ctx.author.name} deleted {msgs} messages'))
        print(
            f'[{datetime.datetime.utcnow().replace(microsecond=0)} INFO]: [Moderation] {ctx.author.name} purged {msgs} messages in {ctx.guild.name}')

    @commands.command(aliases=['stfu'])
    @commands.has_permissions(manage_messages=True)
    async def mute(self, ctx, member: discord.Member, *, reason=''):
        """Mutes a member of the server."""
        muted_role = ctx.guild.get_role(muted_role_id)
        await member.add_roles(muted_role, reason=f'{reason} || by {ctx.author.name}')
        if reason == '':
            reason = 'No reason specified'
        await ctx.send(embed=discord.Embed(description=f'{member} muted by: {ctx.author.name} for: {reason}'))
        print(
            f'[{datetime.datetime.utcnow().replace(microsecond=0)} INFO]: [Moderation] Muted {member} in {ctx.guild.name}')

    @commands.command()
    @commands.has_permissions(manage_messages=True)
    async def unmute(self, ctx, member: discord.Member, *, reason=''):
        """Unmutes a member of the server."""
        muted_role = ctx.guild.get_role(muted_role_id)
        await member.remove_roles(muted_role, reason=f'{reason} || by {ctx.author.name}')
        await ctx.send(embed=discord.Embed(description=f'{member} unmuted by {ctx.author.name}'))
        print(
            f'[{datetime.datetime.utcnow().replace(microsecond=0)} INFO]: [Moderation] Unmuted {member} in {ctx.guild.name}')

    @commands.command()
    @commands.has_permissions(manage_roles=True)
    async def setreaction(self, ctx, role: discord.Role = None, msg: discord.Message = None, emoji=None):
        if role and msg and emoji:
            await msg.add_reaction(emoji)
            self.bot.reaction_roles.append([role.id, msg.id, emoji])
            data = read_json('config')
            data['reaction_roles'].append([role.id, msg.id, emoji])
            print(data['reaction_roles'])
            write_json('config', data)

    @commands.command()
    @is_mod_or_has_perms(manage_roles=True)
    async def verify(self, ctx, member: discord.Member):
        verify_role = ctx.guild.get_role(self.bot.verification_role)
        await member.add_roles(verify_role)
        await ctx.send(f'```Successfully verified: {member.name}```')


def setup(bot):
    bot.add_cog(Moderation(bot))