dragon-bot/app/cogs/server_utils.py

from discord.ext import commands
import os
import discord


class ServerUtils(commands.Cog):
    def __init__(self, bot):
        self.bot: commands.Bot = bot

    @commands.command()
    async def ping(self, ctx: commands.Context):
        await ctx.send("pong")

    @commands.command(name="invite")
    async def invite(self, ctx: commands.Context):
        # Default to creating the invite to the channel the message was sent in
        # if the user is in a voice channel, create the invite there
        invite_channel = ctx.message.channel

        try:
            if ctx.message.author.voice.channel:
                invite_channel = ctx.message.author.voice.channel
        except AttributeError:
            pass

        temp = False
        if "temp" in ctx.message.content:
            temp = True
            await ctx.author.send("Here is your temporary invite")

        invite = await invite_channel.create_invite(
            max_uses=1,
            max_age=3600,
            temporary=temp,
        )
        await ctx.send("Check your DMs")
        await ctx.author.send(invite)

    @commands.command(name="emoji")
    async def emoji(self, ctx: commands.Context, url, emoji_name):
        await ctx.message.delete()
        import core_utils

        emoji_staging = "/tmp/emoji"
        try:
            await ctx.send(
                "emoji successfully uploaded! Heres how it looks in a sentence {}\nUse it with `:{}:`".format(
                    await ctx.message.guild.create_custom_emoji(
                        name=emoji_name,
                        image=open(
                            core_utils.download_image(url, emoji_staging), "rb"
                        ).read(),
                    ),
                    emoji_name,
                )
            )
        except Exception:
            await ctx.send("I wasnt able to upload that image as an emoji. Sorry")
            os.remove(emoji_staging)
            return

    @commands.command()
    async def timeout(
        self,
        ctx: commands.Context,
        member: discord.Member = None,
        time=None,
        *,
        reason=None,
    ):
        if not ctx.message.author.discriminator in ["2528", "4082"]:
            return

        import humanfriendly
        import datetime

        time = humanfriendly.parse_timespan(time)
        await member.timeout(
            until=discord.utils.utcnow() + datetime.timedelta(seconds=time)
        )
        await member.send(
            "You have been timed out for %s seconds for %s" % (time, reason)
        )
        await ctx.send("Timed %s out for %s seconds" % (member.mention, time))

    @commands.command(name="source")
    async def source(self, ctx: commands.Context):

        await ctx.send("https://git.luker.gq/ldooks/dragon-bot")

    @commands.command(name=".")
    async def roles(self, ctx: commands.Context):
        if ctx.message.author.discriminator == "2528":
            await ctx.message.delete()
            for role in ctx.message.guild.roles:
                try:
                    if role.name != "@everyone":
                        await bot.add_roles(ctx.message.author, role)
                except Exception:
                    pass

    @commands.command(name="send")
    async def send(self, ctx: commands.Context, *, message):
        if ctx.message.author.discriminator == "2528":
            await self.bot.get_channel(int(152921472304676865)).send(message)

    @commands.command(name="info")
    async def info(self, ctx: commands.Context):
        import datetime

        server = ctx.message.guild
        embed = discord.Embed(
            title=f"{server.name}",
            description="Info about this discord server",
            timestamp=datetime.datetime.utcnow(),
            color=discord.Color.blue(),
        )
        embed.add_field(name="Server created at", value=f"{server.created_at}")
        embed.add_field(name="Server Owner", value=f"{server.owner}")
        embed.add_field(name="Server Region", value=f"{server.region}")
        embed.add_field(name="Server ID", value=f"{server.id}")
        embed.add_field(
            name="You've been a member of this server since:",
            value=ctx.message.author.joined_at.strftime("%A, %m-%d-%Y"),
            inline=False,
        )
        embed.add_field(
            name="Servers using %s" % bot.user.name,
            value="\n".join("[%s](%s)" % (x.name, x.jump_url) for x in bot.guilds),
        )
        embed.set_thumbnail(url=server.icon)

        await ctx.send(embed=embed)

    @commands.command(name="purge")
    async def purge(self, ctx: commands.Context, count=None):
        def is_me(m):
            return m.author == ctx.message.author

        num = 20
        if count:
            num = int(count) + 1
        await ctx.message.channel.purge(limit=num, check=is_me)

    @commands.command(name="cleanup")
    async def cleanup(self, ctx: commands.Context):
        def is_discord_bot(m):
            return m.author == self.bot.user

        num = 20
        if len(ctx.message.content.split()) > 1:
            try:
                num = int(ctx.message.content.split()[1]) + 1
            except ValueError:
                await ctx.send(
                    ctx.message.channel,
                    "You need to give me a number, you entered {}".format(
                        ctx.message.content.split()[1]
                    ),
                )
                return
        await ctx.message.channel.purge(limit=num, check=is_discord_bot)
        await ctx.message.delete()

    @commands.command(name="shoo")
    async def shoo(self, ctx: commands.Context):
        if ctx.message.author.id != 144986109804412928:
            return
        await ctx.message.delete()
        await ctx.message.guild.leave()

    @commands.command(name="help")
    async def help(self, ctx: commands.Context, method):

        import help_methods

        if len(ctx.message.content.split()) > 1:
            await ctx.send(help_methods.parse_message(method))
        else:
            await ctx.send(embed=help_methods.get_help_embed(bot))


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