change: added simplier solution
This commit is contained in:
@@ -230,53 +230,17 @@ class ZabbixAgentTool:
|
||||
logger.info(f"Upgrading Zabbix agent on {self.distro_family}-based system")
|
||||
|
||||
if self.distro_family == 'debian':
|
||||
# Update package lists
|
||||
# Simple apt upgrade
|
||||
self._run_command("sudo apt update")
|
||||
|
||||
# Check what zabbix packages are installed first
|
||||
result = self._run_command("dpkg -l | grep -E 'zabbix-agent'", check=False, log_output=True)
|
||||
if result.returncode != 0:
|
||||
logger.warning("No Zabbix agent packages found installed")
|
||||
return
|
||||
|
||||
# Get list of installed zabbix agent packages
|
||||
installed_packages = []
|
||||
if result.stdout:
|
||||
for line in result.stdout.strip().split('\n'):
|
||||
if 'zabbix-agent' in line:
|
||||
# Extract package name from dpkg output
|
||||
parts = line.split()
|
||||
if len(parts) >= 2:
|
||||
package_name = parts[1]
|
||||
installed_packages.append(package_name)
|
||||
|
||||
if not installed_packages:
|
||||
logger.warning("No Zabbix agent packages found to upgrade")
|
||||
return
|
||||
|
||||
# Upgrade each installed package individually
|
||||
for package in installed_packages:
|
||||
try:
|
||||
logger.info(f"Upgrading {package}")
|
||||
# Use DEBIAN_FRONTEND=noninteractive to avoid prompts
|
||||
self._run_command(f"sudo DEBIAN_FRONTEND=noninteractive apt upgrade -y {package}")
|
||||
except Exception as e:
|
||||
logger.warning(f"Could not upgrade {package}: {e}")
|
||||
self._run_command("sudo apt upgrade -y")
|
||||
|
||||
elif self.distro_family == 'rhel':
|
||||
# For RHEL-based systems
|
||||
# Simple yum/dnf upgrade - try both
|
||||
try:
|
||||
# Try yum first
|
||||
result = self._run_command("yum list installed | grep zabbix-agent", check=False)
|
||||
if result.returncode == 0:
|
||||
self._run_command("sudo yum update -y zabbix-agent*")
|
||||
else:
|
||||
# Try dnf
|
||||
result = self._run_command("dnf list installed | grep zabbix-agent", check=False)
|
||||
if result.returncode == 0:
|
||||
self._run_command("sudo dnf update -y zabbix-agent*")
|
||||
else:
|
||||
logger.warning("No Zabbix agent packages found to upgrade")
|
||||
self._run_command("sudo yum update -y")
|
||||
except:
|
||||
try:
|
||||
self._run_command("sudo dnf update -y")
|
||||
except Exception as e:
|
||||
logger.warning(f"Could not upgrade packages: {e}")
|
||||
else:
|
||||
|
||||
108
agent-backup-tool/simple_agent_tool.py
Executable file
108
agent-backup-tool/simple_agent_tool.py
Executable file
@@ -0,0 +1,108 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
Simple Zabbix Agent Backup/Upgrade Tool - PoC
|
||||
"""
|
||||
|
||||
import os
|
||||
import sys
|
||||
import argparse
|
||||
import subprocess
|
||||
import shutil
|
||||
from datetime import datetime
|
||||
from pathlib import Path
|
||||
|
||||
ZABBIX_CONFIG_DIR = "/etc/zabbix"
|
||||
SCRIPT_DIR = Path(__file__).parent.absolute()
|
||||
BACKUP_DIR = SCRIPT_DIR / "backups"
|
||||
|
||||
def run_command(cmd):
|
||||
"""Run command and return result"""
|
||||
print(f"Running: {cmd}")
|
||||
result = subprocess.run(cmd, shell=True, capture_output=True, text=True)
|
||||
if result.returncode != 0:
|
||||
print(f"ERROR: {result.stderr.strip()}")
|
||||
sys.exit(1)
|
||||
return result
|
||||
|
||||
def backup_configs():
|
||||
"""Backup zabbix configs"""
|
||||
BACKUP_DIR.mkdir(exist_ok=True)
|
||||
timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
|
||||
backup_dir = BACKUP_DIR / f"backup_{timestamp}"
|
||||
backup_dir.mkdir()
|
||||
|
||||
config_files = list(Path(ZABBIX_CONFIG_DIR).glob("zabbix_agent*.conf"))
|
||||
if not config_files:
|
||||
print("No config files found")
|
||||
return None
|
||||
|
||||
for config_file in config_files:
|
||||
shutil.copy2(config_file, backup_dir / config_file.name)
|
||||
print(f"Backed up: {config_file.name}")
|
||||
|
||||
print(f"Backup saved to: {backup_dir}")
|
||||
return str(backup_dir)
|
||||
|
||||
def restore_configs(backup_path):
|
||||
"""Restore configs from backup"""
|
||||
backup_dir = Path(backup_path)
|
||||
if not backup_dir.exists():
|
||||
print(f"Backup not found: {backup_path}")
|
||||
sys.exit(1)
|
||||
|
||||
config_files = list(backup_dir.glob("zabbix_agent*.conf"))
|
||||
for config_file in config_files:
|
||||
target = Path(ZABBIX_CONFIG_DIR) / config_file.name
|
||||
shutil.copy2(config_file, target)
|
||||
print(f"Restored: {config_file.name}")
|
||||
|
||||
# Restart service
|
||||
services = ['zabbix-agent2', 'zabbix-agent']
|
||||
for service in services:
|
||||
try:
|
||||
run_command(f"sudo systemctl restart {service}")
|
||||
print(f"Restarted: {service}")
|
||||
break
|
||||
except:
|
||||
continue
|
||||
|
||||
def upgrade_system():
|
||||
"""Simple system upgrade"""
|
||||
if os.path.exists('/etc/debian_version'):
|
||||
run_command("sudo apt update")
|
||||
run_command("sudo apt upgrade -y")
|
||||
else:
|
||||
try:
|
||||
run_command("sudo yum update -y")
|
||||
except:
|
||||
run_command("sudo dnf update -y")
|
||||
|
||||
def main():
|
||||
parser = argparse.ArgumentParser(description="Simple Zabbix Agent Tool")
|
||||
parser.add_argument('action', choices=['backup', 'restore', 'upgrade'])
|
||||
parser.add_argument('--backup-path', help='Backup path for restore')
|
||||
args = parser.parse_args()
|
||||
|
||||
if args.action == 'backup':
|
||||
backup_path = backup_configs()
|
||||
if backup_path:
|
||||
print(f"SUCCESS: Backup created at {backup_path}")
|
||||
|
||||
elif args.action == 'restore':
|
||||
if not args.backup_path:
|
||||
print("ERROR: --backup-path required")
|
||||
sys.exit(1)
|
||||
restore_configs(args.backup_path)
|
||||
print("SUCCESS: Restore completed")
|
||||
|
||||
elif args.action == 'upgrade':
|
||||
print("Creating backup before upgrade...")
|
||||
backup_path = backup_configs()
|
||||
print("Upgrading system...")
|
||||
upgrade_system()
|
||||
print("Restoring configs...")
|
||||
restore_configs(backup_path)
|
||||
print(f"SUCCESS: Upgrade completed. Backup at {backup_path}")
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Reference in New Issue
Block a user