updated zabbix import
This commit is contained in:
parent
04aef7d999
commit
95503e8f36
231
zabbix_import.sh
231
zabbix_import.sh
|
@ -1,31 +1,229 @@
|
||||||
|
#!/bin/bash
|
||||||
|
|
||||||
|
# --- Paths to the files created by the Nagios export script ---
|
||||||
|
# These are passed as environment variables to the embedded Python script.
|
||||||
|
export ZABBIX_HOSTGROUPS_FILE="zabbix_hostgroups.txt"
|
||||||
|
export ZABBIX_HOSTS_FILE="zabbix_hosts.txt"
|
||||||
|
export ZABBIX_COMMANDS_FILE="zabbix_commands.txt"
|
||||||
|
export ZABBIX_SERVICES_RAW_FILE="zabbix_services_raw.txt"
|
||||||
|
|
||||||
|
echo "--- Zabbix Automatic Import Script (Bash wrapper for Python) ---"
|
||||||
|
echo " Attempting to ensure all Python dependencies are met."
|
||||||
|
echo ""
|
||||||
|
|
||||||
|
# --- OS Detection for package manager ---
|
||||||
|
DETECTED_OS=""
|
||||||
|
if [ -f /etc/os-release ]; then
|
||||||
|
. /etc/os-release
|
||||||
|
if [[ "$ID" == "debian" || "$ID_LIKE" == "debian" ]]; then
|
||||||
|
DETECTED_OS="debian"
|
||||||
|
elif [[ "$ID" == "ubuntu" ]]; then
|
||||||
|
DETECTED_OS="ubuntu"
|
||||||
|
elif [[ "$ID" == "centos" || "$ID" == "rhel" || "$ID_LIKE" == "fedora" || "$ID" == "fedora" ]]; then
|
||||||
|
DETECTED_OS="redhat"
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
|
||||||
|
install_package() {
|
||||||
|
local package_name="$1"
|
||||||
|
local install_cmd=""
|
||||||
|
|
||||||
|
case "$DETECTED_OS" in
|
||||||
|
"debian"|"ubuntu")
|
||||||
|
install_cmd="sudo apt update && sudo apt install -y $package_name"
|
||||||
|
;;
|
||||||
|
"redhat")
|
||||||
|
if command -v dnf &> /dev/null; then
|
||||||
|
install_cmd="sudo dnf install -y $package_name"
|
||||||
|
else
|
||||||
|
install_cmd="sudo yum install -y $package_name"
|
||||||
|
fi
|
||||||
|
;;
|
||||||
|
*)
|
||||||
|
echo "Error: Unsupported operating system for automatic package installation." >&2
|
||||||
|
echo "Please install '$package_name' manually." >&2
|
||||||
|
return 1
|
||||||
|
;;
|
||||||
|
esac
|
||||||
|
|
||||||
|
echo "Attempting to install '$package_name' using: $install_cmd"
|
||||||
|
if eval "$install_cmd"; then
|
||||||
|
echo "'$package_name' installed successfully."
|
||||||
|
return 0
|
||||||
|
else
|
||||||
|
echo "Error: Installation of '$package_name' failed. Please try installing it manually." >&2
|
||||||
|
return 1
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
# --- Check and install Python3 if missing ---
|
||||||
|
echo "Checking for 'python3'..."
|
||||||
|
if ! command -v python3 &> /dev/null; then
|
||||||
|
echo "---------------------------------------------------------"
|
||||||
|
echo "WARNING: 'python3' is not found on your system."
|
||||||
|
echo "This script requires Python 3 to run."
|
||||||
|
echo ""
|
||||||
|
read -p "Do you want to attempt to install 'python3' now? (y/N): " -n 1 -r REPLY_PYTHON
|
||||||
|
echo
|
||||||
|
if [[ $REPLY_PYTHON =~ ^[Yy]$ ]]; then
|
||||||
|
if install_package "python3"; then
|
||||||
|
echo "'python3' installed. Verifying..."
|
||||||
|
if ! command -v python3 &> /dev/null; then
|
||||||
|
echo "Error: Python 3 not found even after installation. Please check your system PATH." >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
else
|
||||||
|
echo "Python 3 installation failed. Exiting." >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
else
|
||||||
|
echo "Python 3 installation skipped. Exiting." >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
else
|
||||||
|
echo "'python3' found."
|
||||||
|
fi
|
||||||
|
|
||||||
|
# --- Determine the correct pip command (pip3 preferred) ---
|
||||||
|
PIP_CMD=""
|
||||||
|
if command -v pip3 &> /dev/null; then
|
||||||
|
PIP_CMD="pip3"
|
||||||
|
elif command -v pip &> /dev/null; then
|
||||||
|
PIP_CMD="pip"
|
||||||
|
fi
|
||||||
|
|
||||||
|
# --- Check and install pip (or python3-pip) if missing ---
|
||||||
|
echo "Checking for 'pip' (Python package installer)..."
|
||||||
|
if [[ -z "$PIP_CMD" ]]; then
|
||||||
|
echo "---------------------------------------------------------"
|
||||||
|
echo "WARNING: Neither 'pip3' nor 'pip' command found."
|
||||||
|
echo "Python package installer (pip) is required to install 'pyzabbix'."
|
||||||
|
echo ""
|
||||||
|
read -p "Do you want to attempt to install 'python3-pip' now? (y/N): " -n 1 -r REPLY_PIP
|
||||||
|
echo
|
||||||
|
if [[ $REPLY_PIP =~ ^[Yy]$ ]]; then
|
||||||
|
if install_package "python3-pip"; then
|
||||||
|
echo "'python3-pip' installed. Verifying..."
|
||||||
|
if command -v pip3 &> /dev/null; then
|
||||||
|
PIP_CMD="pip3"
|
||||||
|
elif command -v pip &> /dev/null; then
|
||||||
|
PIP_CMD="pip"
|
||||||
|
fi
|
||||||
|
if [[ -z "$PIP_CMD" ]]; then
|
||||||
|
echo "Error: pip not found even after installation. Please check your system PATH or pip setup." >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
else
|
||||||
|
echo "Pip installation failed. Exiting." >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
else
|
||||||
|
echo "Pip installation skipped. Exiting." >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
else
|
||||||
|
echo "'$PIP_CMD' found."
|
||||||
|
fi
|
||||||
|
|
||||||
|
|
||||||
|
# --- Function to check if pyzabbix is installed ---
|
||||||
|
check_pyzabbix() {
|
||||||
|
python3 -c "import pyzabbix" &> /dev/null
|
||||||
|
}
|
||||||
|
|
||||||
|
# --- Check and potentially install pyzabbix ---
|
||||||
|
echo "Checking for 'pyzabbix' library..."
|
||||||
|
if ! check_pyzabbix; then
|
||||||
|
echo "---------------------------------------------------------"
|
||||||
|
echo "WARNING: The 'pyzabbix' Python library is not found."
|
||||||
|
echo "This script requires 'pyzabbix' to interact with the Zabbix API."
|
||||||
|
echo ""
|
||||||
|
read -p "Do you want to attempt to install 'pyzabbix' now using '$PIP_CMD'? (y/N): " -n 1 -r REPLY_PYZABBIX
|
||||||
|
echo
|
||||||
|
if [[ $REPLY_PYZABBIX =~ ^[Yy]$ ]]
|
||||||
|
then
|
||||||
|
echo "Attempting to install 'pyzabbix'..."
|
||||||
|
if "$PIP_CMD" install pyzabbix; then
|
||||||
|
echo "'pyzabbix' installed successfully."
|
||||||
|
else
|
||||||
|
echo "Installation failed. This often requires elevated privileges (sudo)."
|
||||||
|
read -p "Do you want to try installing 'pyzabbix' with 'sudo $PIP_CMD'? (y/N): " -n 1 -r REPLY_SUDO
|
||||||
|
echo
|
||||||
|
if [[ $REPLY_SUDO =~ ^[Yy]$ ]]
|
||||||
|
then
|
||||||
|
if sudo "$PIP_CMD" install pyzabbix; then
|
||||||
|
echo "'pyzabbix' installed successfully with sudo."
|
||||||
|
else
|
||||||
|
echo "Error: 'pyzabbix' installation failed even with sudo." >&2
|
||||||
|
echo "Please install it manually using: sudo $PIP_CMD install pyzabbix" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
else
|
||||||
|
echo "Automatic installation skipped. Exiting." >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
|
||||||
|
# Verify installation after attempt
|
||||||
|
if ! check_pyzabbix; then
|
||||||
|
echo "Error: 'pyzabbix' is still not found after installation attempt. Exiting." >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
else
|
||||||
|
echo "Automatic installation skipped. Exiting." >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
else
|
||||||
|
echo "'pyzabbix' library found. Proceeding..."
|
||||||
|
fi
|
||||||
|
|
||||||
|
echo ""
|
||||||
|
|
||||||
|
# --- Check if export files exist ---
|
||||||
|
REQUIRED_FILES=("$ZABBIX_HOSTGROUPS_FILE" "$ZABBIX_HOSTS_FILE" "$ZABBIX_COMMANDS_FILE" "$ZABBIX_SERVICES_RAW_FILE")
|
||||||
|
MISSING_FILES=()
|
||||||
|
for file in "${REQUIRED_FILES[@]}"; do
|
||||||
|
if [[ ! -f "$file" ]]; then
|
||||||
|
MISSING_FILES+=("$file")
|
||||||
|
fi
|
||||||
|
done
|
||||||
|
|
||||||
|
if [[ ${#MISSING_FILES[@]} -gt 0 ]]; then
|
||||||
|
echo "Error: The following required export files are missing:" >&2
|
||||||
|
for missing_file in "${MISSING_FILES[@]}"; do
|
||||||
|
echo " - $missing_file" >&2
|
||||||
|
done
|
||||||
|
echo "Please ensure you have run the Nagios export script first to create these files." >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
echo "All core dependencies seem to be in order. Launching Zabbix import."
|
||||||
|
# Run the embedded Python script
|
||||||
|
python3 - <<EOF_PYTHON
|
||||||
import os
|
import os
|
||||||
import sys
|
import sys
|
||||||
import json
|
import json
|
||||||
from pyzabbix import ZabbixAPI, ZabbixAPIException
|
from pyzabbix import ZabbixAPI, ZabbixAPIException
|
||||||
import getpass # For secure password input
|
import getpass
|
||||||
|
|
||||||
# --- Paths to the files created by the Nagios export script ---
|
# Paths to the files created by the Nagios export script
|
||||||
ZABBIX_HOSTGROUPS_FILE = 'zabbix_hostgroups.txt'
|
# These are read from environment variables set by the bash script.
|
||||||
ZABBIX_HOSTS_FILE = 'zabbix_hosts.txt'
|
ZABBIX_HOSTGROUPS_FILE = os.environ.get('ZABBIX_HOSTGROUPS_FILE', 'zabbix_hostgroups.txt')
|
||||||
ZABBIX_COMMANDS_FILE = 'zabbix_commands.txt' # Now contains JSON
|
ZABBIX_HOSTS_FILE = os.environ.get('ZABBIX_HOSTS_FILE', 'zabbix_hosts.txt')
|
||||||
ZABBIX_SERVICES_RAW_FILE = 'zabbix_services_raw.txt'
|
ZABBIX_COMMANDS_FILE = os.environ.get('ZABBIX_COMMANDS_FILE', 'zabbix_commands.txt')
|
||||||
|
ZABBIX_SERVICES_RAW_FILE = os.environ.get('ZABBIX_SERVICES_RAW_FILE', 'zabbix_services_raw.txt')
|
||||||
|
|
||||||
# --- Main script ---
|
# --- Main script ---
|
||||||
def main():
|
def main():
|
||||||
print("--- Zabbix Automatic Import Script ---")
|
|
||||||
|
|
||||||
# Ask user for Zabbix API Configuration
|
|
||||||
print("\nPlease provide your Zabbix API connection details:")
|
print("\nPlease provide your Zabbix API connection details:")
|
||||||
zabbix_server = input("Zabbix Server URL (e.g., http://your_zabbix_server/api_jsonrpc.php): ").strip()
|
zabbix_server = input("Zabbix Server URL (e.g., http://your_zabbix_server/api_jsonrpc.php): ").strip()
|
||||||
zabbix_user = input("Zabbix API Login Name: ").strip()
|
zabbix_user = input("Zabbix API Login Name: ").strip()
|
||||||
zabbix_password = getpass.getpass("Zabbix API Password: ").strip()
|
zabbix_password = getpass.getpass("Zabbix API Password: ").strip()
|
||||||
|
|
||||||
# Validate inputs
|
|
||||||
if not zabbix_server or not zabbix_user or not zabbix_password:
|
if not zabbix_server or not zabbix_user or not zabbix_password:
|
||||||
print("Error: All Zabbix connection details must be provided.", file=sys.stderr)
|
print("Error: All Zabbix connection details must be provided.", file=sys.stderr)
|
||||||
sys.exit(1)
|
sys.exit(1)
|
||||||
|
|
||||||
# Connect to Zabbix API
|
|
||||||
try:
|
try:
|
||||||
zapi = ZabbixAPI(zabbix_server)
|
zapi = ZabbixAPI(zabbix_server)
|
||||||
zapi.login(zabbix_user, zabbix_password)
|
zapi.login(zabbix_user, zabbix_password)
|
||||||
|
@ -37,7 +235,7 @@ def main():
|
||||||
|
|
||||||
# 1. Import Hostgroups
|
# 1. Import Hostgroups
|
||||||
print("\n--- Importing Hostgroups ---")
|
print("\n--- Importing Hostgroups ---")
|
||||||
created_hostgroups = {} # To store mapping of Nagios group name to Zabbix group ID
|
created_hostgroups = {}
|
||||||
if not os.path.exists(ZABBIX_HOSTGROUPS_FILE):
|
if not os.path.exists(ZABBIX_HOSTGROUPS_FILE):
|
||||||
print(f"Warning: Hostgroups file not found: {ZABBIX_HOSTGROUPS_FILE}. Skipping hostgroup import.")
|
print(f"Warning: Hostgroups file not found: {ZABBIX_HOSTGROUPS_FILE}. Skipping hostgroup import.")
|
||||||
else:
|
else:
|
||||||
|
@ -249,4 +447,11 @@ def process_host_entry(zapi, host_lines, created_hostgroups):
|
||||||
print(f" Error creating host '{host_name}': {e}", file=sys.stderr)
|
print(f" Error creating host '{host_name}': {e}", file=sys.stderr)
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
main()
|
main()
|
||||||
|
EOF_PYTHON
|
||||||
|
|
||||||
|
echo ""
|
||||||
|
echo "--- Zabbix Import Process Complete ---"
|
||||||
|
echo "Please review the output above for any errors or warnings."
|
||||||
|
echo "Remember to manually configure Zabbix Items, Triggers, and Templates"
|
||||||
|
echo "based on the exported Nagios Commands and Services."
|
Loading…
Reference in New Issue