57 lines
2.6 KiB
Python
57 lines
2.6 KiB
Python
import discord
|
|
from discord import app_commands
|
|
import os
|
|
import logging
|
|
import subprocess
|
|
import sys
|
|
|
|
|
|
class Update:
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
self.logger = logging.getLogger('Update')
|
|
self.logger.setLevel(logging.DEBUG)
|
|
handler = logging.FileHandler(filename='log/selena.log', encoding='utf-8', mode='w')
|
|
handler.setFormatter(logging.Formatter('%(asctime)s:%(levelname)s:%(name)s:%(message)s'))
|
|
self.logger.addHandler(handler)
|
|
|
|
async def update_bot(self, interaction: discord.Interaction):
|
|
try:
|
|
await interaction.response.send_message(embed=discord.Embed(description="Updating Selena...", color=discord.Color.green()))
|
|
except discord.errors.InteractionResponded:
|
|
await interaction.followup.send(embed=discord.Embed(description="Updating Selena...", color=discord.Color.green()))
|
|
|
|
# Fetch updates from the specified branch
|
|
branch = 'main' # change this to your branch if necessary
|
|
result = subprocess.run(['git', 'pull', 'origin', branch], capture_output=True, text=True)
|
|
|
|
if result.returncode == 0:
|
|
self.logger.info("Successfully pulled updates from the repository")
|
|
self.logger.info(result.stdout)
|
|
try:
|
|
await interaction.followup.send(embed=discord.Embed(description="Successfully updated Selena. Restarting...", color=discord.Color.green()))
|
|
except discord.errors.InteractionResponded:
|
|
await interaction.followup.send(embed=discord.Embed(description="Successfully updated Selena. Restarting...", color=discord.Color.green()))
|
|
|
|
# Restart the bot (this is just a placeholder, modify according to your setup)
|
|
os.execv(sys.executable, ['python'] + sys.argv)
|
|
else:
|
|
self.logger.error(f"Failed to pull updates: {result.stderr}")
|
|
try:
|
|
await interaction.followup.send(embed=discord.Embed(description="Failed to update Selena. Check logs for details.", color=discord.Color.red()))
|
|
except discord.errors.InteractionResponded:
|
|
await interaction.followup.send(embed=discord.Embed(description="Failed to update Selena. Check logs for details.", color=discord.Color.red()))
|
|
|
|
def setup(self, tree: app_commands.CommandTree):
|
|
@tree.command(name="update", description="Update the bot to the latest version")
|
|
async def update_command(interaction: discord.Interaction):
|
|
await self.update_bot(interaction)
|
|
|
|
if not tree.get_command("update"):
|
|
tree.add_command(update_command)
|
|
|
|
|
|
def setup(bot):
|
|
update = Update(bot)
|
|
update.setup(bot.tree)
|