sparkytron3000/extensions/admin.py

#Adds administrative commands to the bot
import os
import sys
import subprocess
from discord.ext import commands

class Admin(commands.Cog):

    def __init__(self, bot):
        self.bot = bot
        self.admin_ids = [242018983241318410]

    @commands.command(
        description="Kill", 
        help="Kills the bot in event of an emergency. Only special users can do this! Usage: !kill", 
        brief="Kill the bot",
        hidden=True
        )      
    async def kill(self, ctx):
        "Kills the bot"
        if ctx.author.id in self.admin_ids:
            self.bot.logger.info(f"Kill command ran by {ctx.author.id}")
            exit()
        else:
            await ctx.channel.send("You don't have permission to do that.")
            self.bot.logger.info(f"Kill command attempted by {ctx.author.id}")
            
    @commands.command(
        description="Reset", 
        help="Resets the bot in event of an emergency. Only special users can do this! Usage: !reset", 
        brief="Reset the bot",
        hidden=True
        )  
    async def reset(self, ctx):
        if ctx.author.id in self.admin_ids:
            self.bot.logger.info(f"Reset command ran by {ctx.author.id}")
            python = sys.executable
            os.execl(python, python, *sys.argv)
        else:
            await ctx.channel.send("You don't have permission to do that.")
            self.bot.logger.info(f"Reset command attempted by {ctx.author.id}")

    @commands.command(
        description="Update", 
        help="This will update sparkytron to the most recent version on github. Only privileged users can run this command! Usage: !update", 
        brief="Runs git pull",
        hidden=True
        )           
    async def update(self, ctx):
        if ctx.author.id in self.admin_ids:
            self.bot.logger.info(f"Reset command ran by {ctx.author.id}")
            output = subprocess.run(["git","pull"],capture_output=True)
            if output.stderr:
                await ctx.send("Update Attempted")
                await ctx.send(output.stderr.decode('utf-8'))
            else:
                await ctx.send(output.stdout.decode('utf-8'))
        else:
            await ctx.send("You don't have permission to do this.")
            self.bot.logger.info(f"Update command attempted by {ctx.author.id}")

async def setup(bot):
    try:
        bot.logger.info(f"Successfully added Admin cog")
        await bot.add_cog(Admin(bot))
    except:
        bot.logger.exception(f"Failed to add Admin cog")