init commit

commands.py

@@ -0,0 +1,129 @@
+import discord, toml
+from discord.ext import commands
+from discord.commands import option
+
+
+with open("config.toml", "r") as f:
+    config = toml.load(f)
+guild_ids=config["bot"]["guild_ids"]
+
+class EssentialCommands(commands.Cog):
+    def __init__(self, bot):
+        self.bot = bot
+
+    @commands.slash_command(guild_ids=guild_ids)
+    async def ping(self, ctx):
+        await ctx.respond('Pong!')
+
+    @commands.slash_command(guild_ids=guild_ids)
+    @option('msg', description="The message to say.")
+    async def say(self, ctx, message):
+        await ctx.send(message)
+
+    @commands.slash_command(guild_ids=guild_ids)
+    async def serverinfo(self, ctx):
+
+        await ctx.defer()
+
+        embed = discord.Embed(title=f"{ctx.guild.name} Info", colour=discord.Colour.blurple())
+
+        embed.add_field(name="Roles", value=len(ctx.guild.roles), inline=False)
+        embed.add_field(name="Members", value=ctx.guild.member_count, inline=False)
+        embed.add_field(name="Owner", value=ctx.guild.owner.mention, inline=False)
+        embed.add_field(name="Text Channels", value=len(ctx.guild.text_channels), inline=False)
+        embed.add_field(name="Voice Channels", value=len(ctx.guild.voice_channels), inline=False)
+        embed.add_field(name="Boost Level", value=ctx.guild.premium_tier, inline=False)
+        embed.add_field(name="Created", value=ctx.guild.created_at, inline=False)
+        embed.add_field(name="ID", value=ctx.guild.id, inline=False)
+        embed.add_field(name="Max File Size", value=f"{str(int(ctx.guild.filesize_limit / (1024 * 1024)))} MB", inline=False)
+        icon = ctx.guild.icon
+        embed.set_thumbnail(url=icon)
+
+        await ctx.followup.send(embed=embed)
+
+    @commands.slash_command(guild_ids=guild_ids)
+    @option('ammount', description="The of messages to clear.", required=True)
+    @commands.has_permissions(manage_messages=True)
+    async def clean(self, ctx, amount: int = 10):
+        await ctx.defer()
+        try:
+            await ctx.channel.purge(limit=amount + 1)
+        except:
+            await ctx.followup.send(f"Could not purge all of the messages.")
+        else:
+            await ctx.followup.send(f"Purged {amount} messages.")
+
+    @commands.slash_command(guild_ids=guild_ids)
+    @commands.has_permissions(manage_channels=True)
+    async def lock(self, ctx, *, reason=None):
+        await ctx.defer()
+        channel = ctx.channel
+        overwrite = channel.overwrites_for(ctx.guild.default_role)
+        overwrite.send_messages = False
+        await channel.set_permissions(ctx.guild.default_role, overwrite=overwrite)
+
+        embed = discord.Embed(title="Locked", description=f"Reason: {reason}")
+        await ctx.followup.send(embed=embed)
+
+    @commands.slash_command(guild_ids=guild_ids)
+    @commands.has_permissions(manage_channels=True)
+    async def unlock(self, ctx, *, reason=None):
+        await ctx.defer()
+        channel = ctx.channel
+        overwrite = channel.overwrites_for(ctx.guild.default_role)
+        overwrite.send_messages = True
+        await channel.set_permissions(ctx.guild.default_role, overwrite=overwrite)
+
+        embed = discord.Embed(title="Unlocked", description=f"Reason: {reason}")
+        await ctx.followup.send(embed=embed)
+
+    @commands.slash_command(guild_ids=guild_ids)
+    @commands.has_permissions(manage_channels=True)
+    async def lockdown(self, ctx, *, reason=None):
+        await ctx.defer()
+        for channel in ctx.guild.channels:
+            if isinstance(channel, discord.TextChannel):
+                overwrite = channel.overwrites_for(ctx.guild.default_role)
+                overwrite.send_messages = False
+                await channel.set_permissions(ctx.guild.default_role, overwrite=overwrite)
+
+        embed = discord.Embed(title="Server Locked Down", description=f"Reason: {reason}")
+        await ctx.followup.send(embed=embed)
+
+    @commands.slash_command(guild_ids=guild_ids)
+    @commands.has_permissions(manage_channels=True)
+    async def unlockdown(self, ctx):
+        await ctx.defer()
+        for channel in ctx.guild.channels:
+            if isinstance(channel, discord.TextChannel):
+                overwrite = channel.overwrites_for(ctx.guild.default_role)
+                overwrite.send_messages = None
+                await channel.set_permissions(ctx.guild.default_role, overwrite=overwrite)
+
+        embed = discord.Embed(title="Server Unlocked", description="All channels have been unlocked.")
+        await ctx.followup.send(embed=embed)
+
+    @commands.slash_command(guild_ids=guild_ids)
+    @option('title', description="The biggest text.", required=True)
+    @option('content', description="The actuall text to put in the embed.", required=True)
+    @option('footer', description="The lower text.", required=False)
+    @option('color', description="The color as a word example: Color: dark_blue", required=False)
+    async def embed(self, ctx, title: str, content: str, color: str = None, footer: str = None):
+        embed = discord.Embed(title=title, description=content)
+
+        if color:
+            try:
+                atr = getattr(discord.Color, color)
+                embed.color = atr()
+            except Exception as e:
+                bot.api.logger("inf", f"Command ivoked a error wrong color: {e}")
+                ctx.send("ERR: Invalid color.")
+                return 1
+        if footer:
+            embed.set_footer(text=footer)
+
+        await ctx.send(embed=embed)
+
+def setup(bot):
+    bot.add_cog(EssentialCommands(bot))
+