Beispiel #1
0
    async def on_command_error(self, ctx: Context,
                               error: Exception) -> NoReturn:
        """If an unexpected error occurs, it displays an... error message?

        Attributes:
        -----------
        - `error` - error information

        """
        s = await Settings(ctx.guild.id)
        lang = await s.get_field("locale", CONFIG["default_locale"])
        STRINGS = Strings(lang)
        COMMANDS = Commands(lang)

        if isinstance(error, commands.CommandNotFound):
            return
        await ctx.message.add_reaction(CONFIG["no_emoji"])

        if isinstance(error, commands.MissingRequiredArgument):
            prefix = await s.get_field("prefix", CONFIG["default_prefix"])

            if ctx.command.cog.name != "Jishaku":
                embed = Utils.error_embed(STRINGS["etc"]["usage"].format(
                    COMMANDS[ctx.command.cog.name]["commands"][
                        ctx.command.name]["usage"].format(prefix)))
        elif isinstance(error, commands.MissingPermissions):
            embed = Utils.error_embed(STRINGS["error"]["missing_perms"])

        elif isinstance(error, commands.BotMissingPermissions):
            embed = Utils.error_embed(
                STRINGS["error"]["missing_bot_perms"].format(" ".join(
                    "+ " + STRINGS["etc"]["permissions"][f"{perm}"]
                    for perm in error.missing_perms)))

        elif isinstance(error, commands.CommandOnCooldown):
            embed = Utils.error_embed(STRINGS["error"]["cooldown"].format(
                error.retry_after))

        elif isinstance(error, commands.errors.NSFWChannelRequired):
            embed = discord.Embed(
                title=STRINGS["error"]["nsfwerrortitle"],
                description=STRINGS["error"]["nsfwnotcorrectspot"],
                color=0xFF0000,
            )
            embed.add_field(
                name=STRINGS["error"]["nsfwlogerror"],
                value=STRINGS["error"]["nsfwtraceback"].format(str(error)),
                inline=False,
            )

        else:
            embed = discord.Embed(color=0xDD0000)
            embed.title = STRINGS["error"]["on_error_title"]
            embed.description = STRINGS["error"]["on_error_text"].format(
                str(error))
            Logger.warn(str(error))

        msg = await ctx.send(embed=embed)
        await asyncio.sleep(20)
        await msg.delete()
Beispiel #2
0
    async def unban(self, ctx: Context, *, member: Member) -> NoReturn:
        """Unbans the user.

        Attributes:
        -----------
        - `member` - user tag. Example: `name#1234`

        """
        s = await Settings(ctx.guild.id)
        lang = await s.get_field("locale", CONFIG["default_locale"])
        STRINGS = Strings(lang)

        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)
                await ctx.message.add_reaction(CONFIG["yes_emoji"])
                return

        await ctx.message.add_reaction(CONFIG["no_emoji"])
        embed = Utils.error_embed(STRINGS["error"]["user_not_found"])
        await ctx.send(embed=embed)
Beispiel #3
0
    async def mute(
        self, ctx: Context, member: discord.Member, *, reason: str = "N/A"
    ) -> NoReturn:
        s = await Settings(ctx.guild.id)
        lang = await s.get_field("locale", CONFIG["default_locale"])
        STRINGS = Strings(lang)
        mute_role_id = await s.get_field("mute_role_id")

        if (
            mute_role_id is None
            or discord.utils.get(ctx.guild.roles, id=mute_role_id) is None
        ):
            embed = Utils.done_embed(STRINGS["moderation"]["on_mute_role_create"])
            await ctx.send(embed=embed)
            mute_role = await ctx.guild.create_role(name="OpenMod - Muted")

            await s.set_field("mute_role_id", mute_role.id)
            mute_role_id = await s.get_field("mute_role_id")

        else:
            mute_role = discord.utils.get(ctx.guild.roles, id=mute_role_id)

            for user_role in member.roles:
                if user_role == mute_role:
                    embed = Utils.error_embed(STRINGS["error"]["already_muted"])
                    await ctx.send(embed=embed)
                    return

        for channel in ctx.guild.text_channels:
            await channel.set_permissions(
                mute_role, read_messages=True, send_messages=False, speak=False
            )

        await member.add_roles(mute_role)
        await ctx.message.add_reaction(CONFIG["yes_emoji"])
Beispiel #4
0
    async def setname(self, ctx: Context, member: Member, *,
                      name: str) -> NoReturn:
        s = await Settings(ctx.guild.id)
        lang = await s.get_field("locale", CONFIG["default_locale"])
        STRINGS = Strings(lang)

        if len(name) > 32:
            embed = Utils.error_embed(STRINGS["error"]["too_long_name"])
            await ctx.send(embed=embed)
        elif (ctx.message.author.guild_permissions.manage_nicknames
              or member == ctx.message.author):
            await member.edit(nick=name)
            await ctx.message.add_reaction(CONFIG["yes_emoji"])
        else:
            embed = Utils.error_embed(STRINGS["error"]["missing_perms"])
            await ctx.send(embed=embed)
Beispiel #5
0
    async def kick(self,
                   ctx: Context,
                   member: Member,
                   *,
                   reason: str = "N/A") -> NoReturn:
        """Kicks the user.

        Attributes:
        -----------
        - `member` - user
        - `reason` - kick reason

        """
        s = await Settings(ctx.guild.id)
        lang = await s.get_field("locale", CONFIG["default_locale"])
        STRINGS = Strings(lang)

        select_components = [[
            Button(style=ButtonStyle.green, label="✓"),
            Button(style=ButtonStyle.red, label="X"),
        ]]
        done_components = [[
            Button(style=ButtonStyle.grey, label="·", disabled=True),
        ]]

        embedconfirm = discord.Embed(
            title="Kick Command",
            description="```Do you want to kick this member?```",
        )
        await ctx.send(embed=embedconfirm, components=select_components)
        response = await self.bot.wait_for(
            "button_click", check=lambda message: message.author == ctx.author)
        if response.component.label == "✓":
            await response.respond(
                type=7,
                embed=discord.Embed(
                    title="Action Completed",
                    description=f"Kicked {member} for {reason}",
                    color=0xDD2E44,
                ),
                components=done_components,
            )
            if not member.bot:
                embed = Utils.error_embed(
                    STRINGS["moderation"]["dm_kick"].format(ctx.guild, reason))
                await member.send(embed=embed)
            await asyncio.sleep(5)
            await member.kick()
            await ctx.message.add_reaction(CONFIG["yes_emoji"])
        else:
            await response.respond(
                type=7,
                embed=discord.Embed(
                    title="Action Aborted",
                    description="The action was aborted by clicking the no button",
                    color=0xDD2E44,
                ),
                components=done_components,
            )
            return
Beispiel #6
0
    async def ban(
        self, ctx: Context, member: discord.Member, *, reason: str = "N/A"
    ) -> NoReturn:
        """Bans the user.

        Attributes:
        -----------
        - `member` - user
        - `reason` - ban reason

        """
        s = await Settings(ctx.guild.id)
        lang = await s.get_field("locale", CONFIG["default_locale"])
        STRINGS = Strings(lang)

        try:
            embed = Utils.error_embed(
                STRINGS["moderation"]["dm_ban"].format(ctx.guild.name, reason)
            )
            await member.send(embed=embed)
            await asyncio.sleep(5)
            await member.ban(reason=reason)

        except discord.Forbidden:
            await ctx.message.add_reaction(CONFIG["no_emoji"])
            embed = Utils.error_embed(STRINGS["error"]["ban_fail"])
            msg = await ctx.send(embed=embed)
            await asyncio.sleep(5)
            await msg.delete()

        else:
            try:
                embed = Utils.error_embed(
                    STRINGS["moderation"]["dm_ban"].format(ctx.guild.name, reason)
                )
                await member.send(embed=embed)
            except:
                pass

            await ctx.message.add_reaction(CONFIG["yes_emoji"])
Beispiel #7
0
    async def _reload(self, ctx: Context, *, module: str) -> NoReturn:
        """Loads a module (cog). If the module is not found
            or an error is found in its code, it will throw an error.

        Attributes:
        -----------
        - `module` - the module to load

        """
        try:
            self.bot.reload_extension(f"cogs.{module}")
        except Exception as e:
            await ctx.message.add_reaction(CONFIG["no_emoji"])
            embed = Utils.error_embed("`{}`: {}".format(type(e).__name__, e))
            await ctx.send(embed=embed)
        else:
            await ctx.message.add_reaction(CONFIG["yes_emoji"])
Beispiel #8
0
    async def multiban(self,
                       ctx: Context,
                       members: Greedy[Member],
                       *,
                       reason: str = "N/A") -> NoReturn:
        """Bans multiple users.

        Attributes:
        -----------
        - `member` - user
        - `reason` - ban reason

        """
        s = await Settings(ctx.guild.id)
        lang = await s.get_field("locale", CONFIG["default_locale"])
        STRINGS = Strings(lang)
        not_banned_members = []

        for member in members:
            try:
                await member.ban(reason=reason)
                await ctx.send("Members banned")

            except discord.Forbidden:
                not_banned_members.append(member.mention)

            else:
                try:
                    embed = Utils.error_embed(
                        STRINGS["moderation"]["dm_ban"].format(
                            ctx.guild.name, reason))
                    await member.send(embed=embed)
                except:
                    pass

        if not not_banned_members:
            await ctx.message.add_reaction(CONFIG["yes_emoji"])
        else:
            await ctx.message.add_reaction(CONFIG["warn_emoji"])
            msg = await ctx.send(
                Utils.warn_embed(
                    STRINGS["moderation"]["on_not_full_multiban"].format(
                        ", ".join(not_banned_members))))
            await asyncio.sleep(30)
            await msg.delete()
Beispiel #9
0
    async def kick(
        self, ctx: Context, member: discord.Member, *, reason: str = "N/A"
    ) -> NoReturn:
        """Kicks the user.

        Attributes:
        -----------
        - `member` - user
        - `reason` - kick reason

        """
        s = await Settings(ctx.guild.id)
        lang = await s.get_field("locale", CONFIG["default_locale"])
        STRINGS = Strings(lang)

        embed = Utils.error_embed(
            STRINGS["moderation"]["dm_kick"].format(ctx.guild, reason)
        )
        await member.send(embed=embed)
        await asyncio.sleep(5)
        await member.kick()
        await ctx.message.add_reaction(CONFIG["yes_emoji"])
Beispiel #10
0
    async def help(self, ctx: Context, command: str = None) -> NoReturn:
        """Shows help for a specific command, or displays a complete list of commands.

        Attributes:
        -----------
        - `command` - the command to display help for.
            If `command` is empty, displays a complete list of commands.
            If the command does not exist, writes that the command was not found.

        """
        s = await Settings(ctx.guild.id)
        lang = await s.get_field("locale", CONFIG["default_locale"])
        prefix = await s.get_field("prefix", CONFIG["default_prefix"])
        STRINGS = Strings(lang)
        COMMANDS = Commands(lang)

        if command is None:
            embed = discord.Embed(
                title=STRINGS["general"]["commands_list"],
                description=STRINGS["general"]["help_list_description"].format(
                    prefix),
                color=0xEF940B,
            )
            for i in COMMANDS:
                title = COMMANDS[i]["title"]

                description = ", ".join(f"`{j}`"
                                        for j in COMMANDS[i]["commands"])

                if self.bot.get_cog(i) is not None:
                    embed.add_field(name=title,
                                    value=description,
                                    inline=False)
            embed.set_footer(text=self.bot.user.name,
                             icon_url=self.bot.user.avatar_url)
            await ctx.send(embed=embed)

        elif command != "":
            for i in COMMANDS:
                for j in COMMANDS[i]["commands"]:
                    if command == j:
                        embed = discord.Embed(
                            title=STRINGS["general"]["helpsystemtitle"].format(
                                f"`{prefix}{j}`"),
                            color=0xEF940B,
                        )

                        embed.add_field(
                            name=STRINGS["general"]["description"],
                            value=COMMANDS[i]["commands"][j]["description"],
                            inline=False,
                        )

                        embed.add_field(
                            name=STRINGS["general"]["usage"],
                            value=COMMANDS[i]["commands"][j]["usage"].format(
                                prefix),
                            inline=False,
                        )

                        if len(COMMANDS[i]["commands"][j]["aliases"]) > 0:
                            aliases = ", ".join(f"`{alias}`"
                                                for alias in COMMANDS[i]
                                                ["commands"][j]["aliases"])

                            embed.add_field(
                                name=STRINGS["general"]["aliases"],
                                value=aliases,
                                inline=False,
                            )

                        embed.set_footer(text=self.bot.user.name,
                                         icon_url=self.bot.user.avatar_url)

                        await ctx.send(embed=embed)
                        return
            else:
                await ctx.send(embed=Utils.error_embed(STRINGS["error"]
                                                       ["command_not_found"]))
Beispiel #11
0
    async def unban(self, ctx, *, member) -> NoReturn:
        """Unbans the user.

        Attributes:
        -----------
        - `member` - user tag. Example: `name#1234`

        """
        s = await Settings(ctx.guild.id)
        lang = await s.get_field("locale", CONFIG["default_locale"])
        STRINGS = Strings(lang)

        select_components = [[
            Button(style=ButtonStyle.green, label="✓"),
            Button(style=ButtonStyle.red, label="X"),
        ]]
        done_components = [[
            Button(style=ButtonStyle.grey, label="·", disabled=True),
        ]]

        embedconfirm = discord.Embed(
            title="Unban Command",
            description="```Do you want to unban this member?```",
        )
        await ctx.send(embed=embedconfirm, components=select_components)
        response = await self.bot.wait_for(
            "button_click", check=lambda message: message.author == ctx.author)

        if "#" in ctx.message.content and response.component.label == "✓":
            banned_users = await ctx.guild.bans()
            for ban_entry in banned_users:
                member_name, member_discriminator = member.split("#")
                user = ban_entry.user
                if (user.name, user.discriminator) == (
                        member_name,
                        member_discriminator,
                ):
                    await ctx.guild.unban(user)
                    await response.respond(
                        type=7,
                        embed=discord.Embed(
                            title="Action confirmed",
                            description=f"Unbanned {user}",
                            color=0xFF8000,
                        ),
                        components=done_components,
                    )

            return
        elif response.component.label == "✓":
            member = await self.client.fetch_user(int(member))
            await ctx.guild.unban(member)
            await response.respond(
                type=7,
                embed=discord.Embed(
                    title="Action confirmed",
                    description=f"Unbanned {member}",
                    color=0xFF8000,
                ),
                components=done_components,
            )
        else:
            await response.respond(
                type=7,
                embed=discord.Embed(
                    title="Action Aborted",
                    description="The action was aborted by clicking the no button",
                    color=0xDD2E44,
                ),
                components=done_components,
            )

        await ctx.message.add_reaction(CONFIG["no_emoji"])
        embed = Utils.error_embed(STRINGS["error"]["user_not_found"])
        await ctx.send(embed=embed)