From 670cec5335db5aa989879afa334b5311f9fe1617 Mon Sep 17 00:00:00 2001 From: Specoolazius Date: Sat, 2 Apr 2022 21:00:06 +0200 Subject: [PATCH] admin extension with update and restart command --- bot/extensions/admin.py | 37 +++++++++++++++++++++++++++++++++++++ 1 file changed, 37 insertions(+) diff --git a/bot/extensions/admin.py b/bot/extensions/admin.py index e69de29..7596665 100644 --- a/bot/extensions/admin.py +++ b/bot/extensions/admin.py @@ -0,0 +1,37 @@ +import asyncio +import os + +import discord +from discord.commands import SlashCommandGroup + +from libs import Client + + +class Admin(discord.Cog): + """< discord.Cog > + + Some developer tools for managing the bot. + """ + + def __init__(self, bot: Client): + self.bot = bot + + __dev_group = SlashCommandGroup(name='dev', description='Developer settings') + + @__dev_group.command(name='update') + async def __update_bot(self, ctx: discord.ApplicationContext) -> None: + await ctx.respond('starting update...') + + if 0 == await self.bot.execute_shell('update.sh'): + await ctx.respond('Updated bot from https://github.com/Specoolazius/mcserver-discordbot\n' + 'You may need to restart the bot') + + else: + await ctx.respond(f'Failed to update bot. Check {os.path.join(self.bot.config.log_path, "discord.log")} ' + f'for more detailed information') + + @__dev_group.command(name='restart') + async def __restart_service(self, ctx: discord.ApplicationContext) -> None: + await ctx.respond('attempting restart...') + await asyncio.create_subprocess_shell(f'sudo systemctl restart') +