feat(update): Implement external script-based update system

- Created update_script.py for reliable external updates
- Modified update command to use external script approach
- Bot now shuts down, launches update script, and restarts automatically
- Added proper repository handling and dependency updates
- Improved update UI with better status messages and buttons
- Fixed issues with in-process updates that could cause corruption

This approach is much more reliable as it:
- Avoids updating files while they're in use
- Properly handles git operations without conflicts
- Ensures clean restart after updates
- Provides better error handling and user feedback
This commit is contained in:
Cursor User
2025-06-18 17:06:30 +02:00
parent 156d79ff32
commit 7b1d54a866
4 changed files with 269 additions and 77 deletions

121
update_script.py Normal file
View File

@@ -0,0 +1,121 @@
#!/usr/bin/env python3
"""
Ultroid Update Script
This script handles updating the bot while it's not running.
"""
import os
import sys
import subprocess
import time
from pathlib import Path
def run_command(cmd, shell=True):
"""Run a command and return success status."""
try:
result = subprocess.run(cmd, shell=shell, capture_output=True, text=True)
print(f"Command: {cmd}")
print(f"Output: {result.stdout}")
if result.stderr:
print(f"Error: {result.stderr}")
return result.returncode == 0
except Exception as e:
print(f"Error running command '{cmd}': {e}")
return False
def main():
"""Main update function."""
print("🔄 Starting Ultroid update process...")
# Get script directory
script_dir = Path(__file__).parent.absolute()
os.chdir(script_dir)
print(f"📁 Working directory: {script_dir}")
# Check if we're in a git repository
if not (script_dir / ".git").exists():
print("❌ Not a git repository. Cannot update.")
return False
# Get the repository URL from command line args or database
repo_url = sys.argv[1] if len(sys.argv) > 1 else None
# Fetch and pull updates
print("📥 Fetching updates from repository...")
if repo_url:
print(f"🔗 Using repository: {repo_url}")
# Set up remote if needed
if not run_command("git remote get-url origin"):
run_command(f"git remote add origin {repo_url}")
else:
run_command(f"git remote set-url origin {repo_url}")
# Fetch latest changes
if not run_command("git fetch origin"):
print("❌ Failed to fetch updates")
return False
# Get current branch
result = subprocess.run("git branch --show-current", shell=True, capture_output=True, text=True)
current_branch = result.stdout.strip() or "main"
print(f"🌿 Current branch: {current_branch}")
# Pull updates
print("⬇️ Pulling updates...")
if not run_command(f"git pull origin {current_branch}"):
print("❌ Failed to pull updates")
return False
# Update dependencies
print("📦 Installing/updating dependencies...")
if not run_command("pip3 install -r requirements.txt --upgrade"):
print("⚠️ Warning: Failed to update some dependencies")
# Try alternative pip command
run_command("pip3 install -r requirements.txt --break-system-packages --upgrade")
print("✅ Update completed successfully!")
return True
def restart_bot():
"""Restart the bot after update."""
print("🔄 Restarting Ultroid...")
# Check if we have a virtual environment
venv_python = None
if os.path.exists("venv/bin/python"):
venv_python = "venv/bin/python"
elif os.path.exists("venv/Scripts/python.exe"):
venv_python = "venv/Scripts/python.exe"
# Determine how to start the bot
if len(sys.argv) > 1 and sys.argv[-1] == "main.py":
# Started with main.py
if venv_python:
os.execv(venv_python, [venv_python, "main.py"])
else:
os.execv(sys.executable, [sys.executable, "main.py"])
else:
# Started as module
if venv_python:
os.execv(venv_python, [venv_python, "-m", "pyUltroid"])
else:
os.execv(sys.executable, [sys.executable, "-m", "pyUltroid"])
if __name__ == "__main__":
print("🚀 Ultroid Update Script")
print("=" * 40)
# Wait a moment for the bot to fully shutdown
time.sleep(2)
# Perform update
if main():
print("=" * 40)
restart_bot()
else:
print("❌ Update failed. Please check the errors above.")
sys.exit(1)