Learnhelp/wdiscordbotserver
1
0
Fork 0
mirror of https://gitlab.com/pancakes1234/wdiscordbotserver.git synced 2025-06-16 07:14:21 -06:00
Code Issues Packages Projects Releases Wiki Activity

Update file update.py

Browse Source
This commit is contained in:
ザカリアス・ウィリアム・ポージー 2025-05-27 20:15:57 +09:00
parent 94b107cad4
commit 8e8541bab9
1 changed files with 27 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
cogs/update.py
View File

@ -10,36 +10,50 @@ class GitUpdateCog(commands.Cog):
def __init__(self, bot: commands.Bot):
self.bot = bot
@app_commands.command(name="update", description="Updates the bot code from GitLab and restarts the bot. (Admin Only)")
@app_commands.command(
name="update",
description="Updates the bot code from GitLab and restarts the bot. (Admin Only)"
)
async def update(self, interaction: discord.Interaction):
# Check for administrator permission
if not interaction.user.guild_permissions.administrator:
await interaction.response.send_message("You do not have permission to run this command.", ephemeral=True)
return
await interaction.response.send_message("Initiating update. The bot will restart shortly...")
target_dir = "~/wdiscordbot-internal-server-aws"
repo_url = "https://gitlab.com/pancakes1234/wdiscordbot-internal-server-aws.git"
restart_script = "home/ubuntu/wdiscordbot-internal-server-aws/bot.py"
# Respond with an initial message
await interaction.response.send_message("Initiating update. The bot will restart shortly...", ephemeral=True)
# Define absolute paths and repository URL
target_dir = "/home/ubuntu/wdiscordbot-internal-server-aws"
repo_url = "https://gitlab.com/pancakes1234/wdiscordbot-internal-server-aws.git"
restart_script = "/home/ubuntu/wdiscordbot-internal-server-aws/bot.py"
try:
if os.path.exists(target_dir):
shutil.rmtree(target_dir)
await interaction.edit_original_response(content=f"Removed directory: {target_dir}")
await interaction.followup.send(f"Removed directory: {target_dir}", ephemeral=True)
else:
await interaction.edit_original_response(content=f"Directory {target_dir} does not exist; proceeding with clone...")
await interaction.followup.send(f"Directory {target_dir} does not exist; proceeding with clone...", ephemeral=True)
# Clone the repository
subprocess.run(["git", "clone", repo_url, target_dir], check=True)
await interaction.edit_original_response(content="Repository cloned successfully.")
await interaction.followup.send("Repository cloned successfully.", ephemeral=True)
except Exception as e:
error_msg = f"Update failed: {e}"
print(error_msg)
await interaction.edit_original_response(content=error_msg)
await interaction.followup.send(error_msg, ephemeral=True)
return
try:
await interaction.edit_original_response(content="Bot has updated to the latest commit and is restarting...")
await interaction.followup.send("Bot has updated to the latest commit and is restarting...", ephemeral=True)
# Optionally change working directory if your bot expects it:
os.chdir(target_dir)
# Restart the bot by replacing the current process with the new version
os.execv(sys.executable, [sys.executable, restart_script])
# If os.execv returns, it means it failed
except Exception as e:
await interaction.edit_original_response(content=f"Failed to restart bot: {e}")
error_msg = f"Failed to restart bot: {e}"
print(error_msg)
await interaction.followup.send(error_msg, ephemeral=True)
async def setup(bot: commands.Bot):
await bot.add_cog(GitUpdateCog(bot))