100 lines
3.5 KiB
Python
100 lines
3.5 KiB
Python
import importlib.util
|
|
import argparse
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
from ..services import config as Config
|
|
from ..lib import color_log
|
|
|
|
def run_command(cmd, description):
|
|
"""Run a command and handle its output."""
|
|
try:
|
|
color_log.Show("INFO", f"Executing: {description}")
|
|
result = subprocess.run(cmd, capture_output=True, text=True)
|
|
if result.stdout:
|
|
print(result.stdout.strip())
|
|
if result.stderr:
|
|
print(result.stderr.strip())
|
|
return result.returncode == 0
|
|
except subprocess.CalledProcessError as e:
|
|
color_log.Show("FAILED", f"Error executing {description}: {str(e)}")
|
|
return False
|
|
|
|
def update_instance(instance_name, action, force_pull=False):
|
|
"""Update a single instance with git pull, module update, and service restart."""
|
|
# 1. Pull latest code
|
|
pull_cmd = ["python", "utility/main.py", "git", "pull", instance_name]
|
|
if force_pull:
|
|
pull_cmd.append("--force")
|
|
if not run_command(pull_cmd, f"Pulling latest code for {instance_name}"):
|
|
color_log.Show("WARNING", f"Skipping module update for {instance_name} due to pull failure")
|
|
return False
|
|
|
|
# 2. Update modules
|
|
module_cmd = ["python", "utility/main.py", "module", action, instance_name]
|
|
if not run_command(module_cmd, f"Updating modules for {instance_name}"):
|
|
color_log.Show("WARNING", f"Module update failed for {instance_name}")
|
|
return False
|
|
|
|
# 3. Restart service
|
|
restart_cmd = ["python", "utility/main.py", "service", "restart", instance_name]
|
|
if not run_command(restart_cmd, f"Restarting service for {instance_name}"):
|
|
color_log.Show("WARNING", f"Service restart failed for {instance_name}")
|
|
return False
|
|
|
|
return True
|
|
|
|
def main():
|
|
# Parse arguments
|
|
parser = argparse.ArgumentParser(description="Update modules for all instances")
|
|
parser.add_argument(
|
|
"action",
|
|
help="Action to perform",
|
|
choices=["uninstall", "install", "upgrade"]
|
|
)
|
|
parser.add_argument(
|
|
"config_path",
|
|
help="Path to the config file"
|
|
)
|
|
parser.add_argument(
|
|
"--force-pull",
|
|
action="store_true",
|
|
help="Force pull with hard reset (discards local changes)"
|
|
)
|
|
args = parser.parse_args()
|
|
|
|
# Load the configuration
|
|
color_log.Show("INFO", f"Loading configuration from {args.config_path}")
|
|
config = Config.Config(config_path=args.config_path)
|
|
|
|
# Get instances
|
|
instances = config.get_instances()
|
|
if not isinstance(instances, list):
|
|
color_log.Show("FAILED", "Error: instances is not a valid list.")
|
|
sys.exit(1)
|
|
|
|
# Process each instance
|
|
success_count = 0
|
|
for instance in instances:
|
|
if "name" not in instance:
|
|
color_log.Show("WARNING", f"Instance missing 'name' field. Skipping.")
|
|
continue
|
|
|
|
instance_name = instance["name"]
|
|
color_log.Show("INFO", f"\nProcessing instance: {instance_name}")
|
|
|
|
if update_instance(instance_name, args.action, args.force_pull):
|
|
success_count += 1
|
|
color_log.Show("OK", f"Successfully updated {instance_name}")
|
|
else:
|
|
color_log.Show("FAILED", f"Failed to update {instance_name}")
|
|
|
|
# Summary
|
|
color_log.Show("INFO", f"\nUpdate Summary:")
|
|
color_log.Show("INFO", f"Total instances: {len(instances)}")
|
|
color_log.Show("INFO", f"Successful updates: {success_count}")
|
|
color_log.Show("INFO", f"Failed updates: {len(instances) - success_count}")
|
|
|
|
if __name__ == "__main__":
|
|
main()
|