mirror of
https://github.com/TheNetworkGuy/netbox-zabbix-sync.git
synced 2025-07-14 01:41:25 -06:00
Merge branch 'main' of https://github.com/TheNetworkGuy/netbox-zabbix-sync
This commit is contained in:
commit
e8a733cbd0
22
README.md
22
README.md
@ -255,6 +255,28 @@ You can set the proxy for a device using the 'proxy' key in config context.
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
```
|
```
|
||||||
|
It is now posible to specify proxy groups with the introduction of Proxy groups in Zabbix 7. Specifying a group in the config context on older Zabbix releases will have no impact and the script will ignore the statement.
|
||||||
|
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
"zabbix": {
|
||||||
|
"proxy_group": "yourawesomeproxygroup.local"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
The script will prefer groups when specifying both a proxy and group. This is done with the assumption that groups are more resiliant and HA ready, making it a more logical choice to use for proxy linkage. This also makes migrating from a proxy to proxy group easier since the group take priority over the invidivual proxy.
|
||||||
|
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
"zabbix": {
|
||||||
|
"proxy": "yourawesomeproxy.local",
|
||||||
|
"proxy_group": "yourawesomeproxygroup.local"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
```
|
||||||
|
In the example above the host will use the group on Zabbix 7. On Zabbix 6 and below the host will use the proxy. Zabbix 7 will use the proxy value when ommiting the proxy_group value.
|
||||||
|
|
||||||
Because of the possible amount of destruction when setting up Netbox but forgetting the proxy command, the sync works a bit different. By default everything is synced except in a situation where the Zabbix host has a proxy configured but nothing is configured in Netbox. To force deletion and a full sync, set the `full_proxy_sync` variable in the config file.
|
Because of the possible amount of destruction when setting up Netbox but forgetting the proxy command, the sync works a bit different. By default everything is synced except in a situation where the Zabbix host has a proxy configured but nothing is configured in Netbox. To force deletion and a full sync, set the `full_proxy_sync` variable in the config file.
|
||||||
|
|
||||||
### Set interface parameters within Netbox
|
### Set interface parameters within Netbox
|
||||||
|
753
modules/device.py
Normal file
753
modules/device.py
Normal file
@ -0,0 +1,753 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
# pylint: disable=invalid-name, logging-not-lazy, too-many-locals, logging-fstring-interpolation, too-many-lines
|
||||||
|
"""
|
||||||
|
Device specific handeling for Netbox to Zabbix
|
||||||
|
"""
|
||||||
|
from os import sys
|
||||||
|
from logging import getLogger
|
||||||
|
from zabbix_utils import APIRequestError
|
||||||
|
from modules.exceptions import (SyncInventoryError, TemplateError, SyncExternalError,
|
||||||
|
InterfaceConfigError, JournalError)
|
||||||
|
from modules.interface import ZabbixInterface
|
||||||
|
from modules.tools import build_path
|
||||||
|
try:
|
||||||
|
from config import (
|
||||||
|
template_cf, device_cf,
|
||||||
|
traverse_site_groups,
|
||||||
|
traverse_regions,
|
||||||
|
inventory_sync,
|
||||||
|
inventory_automatic,
|
||||||
|
inventory_map
|
||||||
|
)
|
||||||
|
except ModuleNotFoundError:
|
||||||
|
print("Configuration file config.py not found in main directory."
|
||||||
|
"Please create the file or rename the config.py.example file to config.py.")
|
||||||
|
sys.exit(0)
|
||||||
|
|
||||||
|
class NetworkDevice():
|
||||||
|
# pylint: disable=too-many-instance-attributes, too-many-arguments
|
||||||
|
"""
|
||||||
|
Represents Network device.
|
||||||
|
INPUT: (Netbox device class, ZabbixAPI class, journal flag, NB journal class)
|
||||||
|
"""
|
||||||
|
|
||||||
|
def __init__(self, nb, zabbix, nb_journal_class, nb_version, journal=None, logger=None):
|
||||||
|
self.nb = nb
|
||||||
|
self.id = nb.id
|
||||||
|
self.name = nb.name
|
||||||
|
self.visible_name = None
|
||||||
|
self.status = nb.status.label
|
||||||
|
self.zabbix = zabbix
|
||||||
|
self.zabbix_id = None
|
||||||
|
self.group_id = None
|
||||||
|
self.nb_api_version = nb_version
|
||||||
|
self.zbx_template_names = []
|
||||||
|
self.zbx_templates = []
|
||||||
|
self.hostgroup = None
|
||||||
|
self.tenant = nb.tenant
|
||||||
|
self.config_context = nb.config_context
|
||||||
|
self.zbxproxy = None
|
||||||
|
self.zabbix_state = 0
|
||||||
|
self.journal = journal
|
||||||
|
self.nb_journals = nb_journal_class
|
||||||
|
self.inventory_mode = -1
|
||||||
|
self.inventory = {}
|
||||||
|
self.logger = logger if logger else getLogger(__name__)
|
||||||
|
self._setBasics()
|
||||||
|
|
||||||
|
def _setBasics(self):
|
||||||
|
"""
|
||||||
|
Sets basic information like IP address.
|
||||||
|
"""
|
||||||
|
# Return error if device does not have primary IP.
|
||||||
|
if self.nb.primary_ip:
|
||||||
|
self.cidr = self.nb.primary_ip.address
|
||||||
|
self.ip = self.cidr.split("/")[0]
|
||||||
|
else:
|
||||||
|
e = f"Device {self.name}: no primary IP."
|
||||||
|
self.logger.info(e)
|
||||||
|
raise SyncInventoryError(e)
|
||||||
|
|
||||||
|
# Check if device has custom field for ZBX ID
|
||||||
|
if device_cf in self.nb.custom_fields:
|
||||||
|
self.zabbix_id = self.nb.custom_fields[device_cf]
|
||||||
|
else:
|
||||||
|
e = f"Custom field {device_cf} not found for {self.name}."
|
||||||
|
self.logger.warning(e)
|
||||||
|
raise SyncInventoryError(e)
|
||||||
|
|
||||||
|
# Validate hostname format.
|
||||||
|
odd_character_list = ["ä", "ö", "ü", "Ä", "Ö", "Ü", "ß"]
|
||||||
|
self.use_visible_name = False
|
||||||
|
if any(letter in self.name for letter in odd_character_list):
|
||||||
|
self.name = f"NETBOX_ID{self.id}"
|
||||||
|
self.visible_name = self.nb.name
|
||||||
|
self.use_visible_name = True
|
||||||
|
self.logger.info(f"Device {self.visible_name} contains special characters. "
|
||||||
|
f"Using {self.name} as name for the Netbox object "
|
||||||
|
f"and using {self.visible_name} as visible name in Zabbix.")
|
||||||
|
else:
|
||||||
|
pass
|
||||||
|
|
||||||
|
def set_hostgroup(self, hg_format, nb_site_groups, nb_regions):
|
||||||
|
"""Set the hostgroup for this device"""
|
||||||
|
# Get all variables from the NB data
|
||||||
|
dev_location = str(self.nb.location) if self.nb.location else None
|
||||||
|
# Check the Netbox version. Use backwards compatibility for versions 2 and 3.
|
||||||
|
if self.nb_api_version.startswith(("2", "3")):
|
||||||
|
dev_role = self.nb.device_role.name
|
||||||
|
else:
|
||||||
|
dev_role = self.nb.role.name
|
||||||
|
manufacturer = self.nb.device_type.manufacturer.name
|
||||||
|
region = str(self.nb.site.region) if self.nb.site.region else None
|
||||||
|
site = self.nb.site.name
|
||||||
|
site_group = str(self.nb.site.group) if self.nb.site.group else None
|
||||||
|
tenant = str(self.tenant) if self.tenant else None
|
||||||
|
tenant_group = str(self.tenant.group) if tenant else None
|
||||||
|
# Set mapper for string -> variable
|
||||||
|
hostgroup_vars = {"dev_location": dev_location, "dev_role": dev_role,
|
||||||
|
"manufacturer": manufacturer, "region": region,
|
||||||
|
"site": site, "site_group": site_group,
|
||||||
|
"tenant": tenant, "tenant_group": tenant_group}
|
||||||
|
# Generate list based off string input format
|
||||||
|
hg_items = hg_format.split("/")
|
||||||
|
hostgroup = ""
|
||||||
|
# Go through all hostgroup items
|
||||||
|
for item in hg_items:
|
||||||
|
# Check if the variable (such as Tenant) is empty.
|
||||||
|
if not hostgroup_vars[item]:
|
||||||
|
continue
|
||||||
|
# Check if the item is a custom field name
|
||||||
|
if item not in hostgroup_vars:
|
||||||
|
cf_value = self.nb.custom_fields[item] if item in self.nb.custom_fields else None
|
||||||
|
if cf_value:
|
||||||
|
# If there is a cf match, add the value of this cf to the hostgroup
|
||||||
|
hostgroup += cf_value + "/"
|
||||||
|
# Should there not be a match, this means that
|
||||||
|
# the variable is invalid. Skip regardless.
|
||||||
|
continue
|
||||||
|
# Add value of predefined variable to hostgroup format
|
||||||
|
if item == "site_group" and nb_site_groups and traverse_site_groups:
|
||||||
|
group_path = build_path(site_group, nb_site_groups)
|
||||||
|
hostgroup += "/".join(group_path) + "/"
|
||||||
|
elif item == "region" and nb_regions and traverse_regions:
|
||||||
|
region_path = build_path(region, nb_regions)
|
||||||
|
hostgroup += "/".join(region_path) + "/"
|
||||||
|
else:
|
||||||
|
hostgroup += hostgroup_vars[item] + "/"
|
||||||
|
# If the final hostgroup variable is empty
|
||||||
|
if not hostgroup:
|
||||||
|
e = (f"{self.name} has no reliable hostgroup. This is"
|
||||||
|
"most likely due to the use of custom fields that are empty.")
|
||||||
|
self.logger.error(e)
|
||||||
|
raise SyncInventoryError(e)
|
||||||
|
# Remove final inserted "/" and set hostgroup to class var
|
||||||
|
self.hostgroup = hostgroup.rstrip("/")
|
||||||
|
|
||||||
|
def set_template(self, prefer_config_context, overrule_custom):
|
||||||
|
""" Set Template """
|
||||||
|
self.zbx_template_names = None
|
||||||
|
# Gather templates ONLY from the device specific context
|
||||||
|
if prefer_config_context:
|
||||||
|
try:
|
||||||
|
self.zbx_template_names = self.get_templates_context()
|
||||||
|
except TemplateError as e:
|
||||||
|
self.logger.warning(e)
|
||||||
|
return True
|
||||||
|
# Gather templates from the custom field but overrule
|
||||||
|
# them should there be any device specific templates
|
||||||
|
if overrule_custom:
|
||||||
|
try:
|
||||||
|
self.zbx_template_names = self.get_templates_context()
|
||||||
|
except TemplateError:
|
||||||
|
pass
|
||||||
|
if not self.zbx_template_names:
|
||||||
|
self.zbx_template_names = self.get_templates_cf()
|
||||||
|
return True
|
||||||
|
# Gather templates ONLY from the custom field
|
||||||
|
self.zbx_template_names = self.get_templates_cf()
|
||||||
|
return True
|
||||||
|
|
||||||
|
def get_templates_cf(self):
|
||||||
|
""" Get template from custom field """
|
||||||
|
# Get Zabbix templates from the device type
|
||||||
|
device_type_cfs = self.nb.device_type.custom_fields
|
||||||
|
# Check if the ZBX Template CF is present
|
||||||
|
if template_cf in device_type_cfs:
|
||||||
|
# Set value to template
|
||||||
|
return [device_type_cfs[template_cf]]
|
||||||
|
# Custom field not found, return error
|
||||||
|
e = (f"Custom field {template_cf} not "
|
||||||
|
f"found for {self.nb.device_type.manufacturer.name}"
|
||||||
|
f" - {self.nb.device_type.display}.")
|
||||||
|
raise TemplateError(e)
|
||||||
|
|
||||||
|
def get_templates_context(self):
|
||||||
|
""" Get Zabbix templates from the device context """
|
||||||
|
if "zabbix" not in self.config_context:
|
||||||
|
e = ("Key 'zabbix' not found in config "
|
||||||
|
f"context for template host {self.name}")
|
||||||
|
raise TemplateError(e)
|
||||||
|
if "templates" not in self.config_context["zabbix"]:
|
||||||
|
e = ("Key 'templates' not found in config "
|
||||||
|
f"context 'zabbix' for template host {self.name}")
|
||||||
|
raise TemplateError(e)
|
||||||
|
return self.config_context["zabbix"]["templates"]
|
||||||
|
|
||||||
|
def set_inventory(self, nbdevice):
|
||||||
|
""" Set host inventory """
|
||||||
|
self.inventory_mode = -1
|
||||||
|
self.inventory = {}
|
||||||
|
if inventory_sync:
|
||||||
|
# Set inventory mode to automatic or manual
|
||||||
|
self.inventory_mode = 1 if inventory_automatic else 0
|
||||||
|
|
||||||
|
# Let's build an inventory dict for each property in the inventory_map
|
||||||
|
for nb_inv_field, zbx_inv_field in inventory_map.items():
|
||||||
|
field_list = nb_inv_field.split("/") # convert str to list based on delimiter
|
||||||
|
# start at the base of the dict...
|
||||||
|
value = nbdevice
|
||||||
|
# ... and step through the dict till we find the needed value
|
||||||
|
for item in field_list:
|
||||||
|
value = value[item] if value else None
|
||||||
|
# Check if the result is usable and expected
|
||||||
|
# We want to apply any int or float 0 values,
|
||||||
|
# even if python thinks those are empty.
|
||||||
|
if ((value and isinstance(value, int | float | str )) or
|
||||||
|
(isinstance(value, int | float) and int(value) ==0)):
|
||||||
|
self.inventory[zbx_inv_field] = str(value)
|
||||||
|
elif not value:
|
||||||
|
# empty value should just be an empty string for API compatibility
|
||||||
|
self.logger.debug(f"Inventory lookup for '{nb_inv_field}'"
|
||||||
|
" returned an empty value")
|
||||||
|
self.inventory[zbx_inv_field] = ""
|
||||||
|
else:
|
||||||
|
# Value is not a string or numeral, probably not what the user expected.
|
||||||
|
self.logger.error(f"Inventory lookup for '{nb_inv_field}' returned"
|
||||||
|
" an unexpected type: it will be skipped.")
|
||||||
|
return True
|
||||||
|
|
||||||
|
def isCluster(self):
|
||||||
|
"""
|
||||||
|
Checks if device is part of cluster.
|
||||||
|
"""
|
||||||
|
return bool(self.nb.virtual_chassis)
|
||||||
|
|
||||||
|
def getClusterMaster(self):
|
||||||
|
"""
|
||||||
|
Returns chassis master ID.
|
||||||
|
"""
|
||||||
|
if not self.isCluster():
|
||||||
|
e = (f"Unable to proces {self.name} for cluster calculation: "
|
||||||
|
f"not part of a cluster.")
|
||||||
|
self.logger.warning(e)
|
||||||
|
raise SyncInventoryError(e)
|
||||||
|
if not self.nb.virtual_chassis.master:
|
||||||
|
e = (f"{self.name} is part of a Netbox virtual chassis which does "
|
||||||
|
"not have a master configured. Skipping for this reason.")
|
||||||
|
self.logger.error(e)
|
||||||
|
raise SyncInventoryError(e)
|
||||||
|
return self.nb.virtual_chassis.master.id
|
||||||
|
|
||||||
|
def promoteMasterDevice(self):
|
||||||
|
"""
|
||||||
|
If device is Primary in cluster,
|
||||||
|
promote device name to the cluster name.
|
||||||
|
Returns True if succesfull, returns False if device is secondary.
|
||||||
|
"""
|
||||||
|
masterid = self.getClusterMaster()
|
||||||
|
if masterid == self.id:
|
||||||
|
self.logger.debug(f"Device {self.name} is primary cluster member. "
|
||||||
|
f"Modifying hostname from {self.name} to " +
|
||||||
|
f"{self.nb.virtual_chassis.name}.")
|
||||||
|
self.name = self.nb.virtual_chassis.name
|
||||||
|
return True
|
||||||
|
self.logger.debug(f"Device {self.name} is non-primary cluster member.")
|
||||||
|
return False
|
||||||
|
|
||||||
|
def zbxTemplatePrepper(self, templates):
|
||||||
|
"""
|
||||||
|
Returns Zabbix template IDs
|
||||||
|
INPUT: list of templates from Zabbix
|
||||||
|
OUTPUT: True
|
||||||
|
"""
|
||||||
|
# Check if there are templates defined
|
||||||
|
if not self.zbx_template_names:
|
||||||
|
e = f"No templates found for device {self.name}"
|
||||||
|
self.logger.info(e)
|
||||||
|
raise SyncInventoryError()
|
||||||
|
# Set variable to empty list
|
||||||
|
self.zbx_templates = []
|
||||||
|
# Go through all templates definded in Netbox
|
||||||
|
for nb_template in self.zbx_template_names:
|
||||||
|
template_match = False
|
||||||
|
# Go through all templates found in Zabbix
|
||||||
|
for zbx_template in templates:
|
||||||
|
# If the template names match
|
||||||
|
if zbx_template['name'] == nb_template:
|
||||||
|
# Set match variable to true, add template details
|
||||||
|
# to class variable and return debug log
|
||||||
|
template_match = True
|
||||||
|
self.zbx_templates.append({"templateid": zbx_template['templateid'],
|
||||||
|
"name": zbx_template['name']})
|
||||||
|
e = (f"Found template {zbx_template['name']}"
|
||||||
|
f" for host {self.name}.")
|
||||||
|
self.logger.debug(e)
|
||||||
|
# Return error should the template not be found in Zabbix
|
||||||
|
if not template_match:
|
||||||
|
e = (f"Unable to find template {nb_template} "
|
||||||
|
f"for host {self.name} in Zabbix. Skipping host...")
|
||||||
|
self.logger.warning(e)
|
||||||
|
raise SyncInventoryError(e)
|
||||||
|
|
||||||
|
def getZabbixGroup(self, groups):
|
||||||
|
"""
|
||||||
|
Returns Zabbix group ID
|
||||||
|
INPUT: list of hostgroups
|
||||||
|
OUTPUT: True / False
|
||||||
|
"""
|
||||||
|
# Go through all groups
|
||||||
|
for group in groups:
|
||||||
|
if group['name'] == self.hostgroup:
|
||||||
|
self.group_id = group['groupid']
|
||||||
|
e = f"Found group {group['name']} for host {self.name}."
|
||||||
|
self.logger.debug(e)
|
||||||
|
return True
|
||||||
|
return False
|
||||||
|
|
||||||
|
def cleanup(self):
|
||||||
|
"""
|
||||||
|
Removes device from external resources.
|
||||||
|
Resets custom fields in Netbox.
|
||||||
|
"""
|
||||||
|
if self.zabbix_id:
|
||||||
|
try:
|
||||||
|
self.zabbix.host.delete(self.zabbix_id)
|
||||||
|
self.nb.custom_fields[device_cf] = None
|
||||||
|
self.nb.save()
|
||||||
|
e = f"Deleted host {self.name} from Zabbix."
|
||||||
|
self.logger.info(e)
|
||||||
|
self.create_journal_entry("warning", "Deleted host from Zabbix")
|
||||||
|
except APIRequestError as e:
|
||||||
|
e = f"Zabbix returned the following error: {str(e)}."
|
||||||
|
self.logger.error(e)
|
||||||
|
raise SyncExternalError(e) from e
|
||||||
|
|
||||||
|
def _zabbixHostnameExists(self):
|
||||||
|
"""
|
||||||
|
Checks if hostname exists in Zabbix.
|
||||||
|
"""
|
||||||
|
# Validate the hostname or visible name field
|
||||||
|
if not self.use_visible_name:
|
||||||
|
zbx_filter = {'host': self.name}
|
||||||
|
else:
|
||||||
|
zbx_filter = {'name': self.visible_name}
|
||||||
|
host = self.zabbix.host.get(filter=zbx_filter, output=[])
|
||||||
|
return bool(host)
|
||||||
|
|
||||||
|
def setInterfaceDetails(self):
|
||||||
|
"""
|
||||||
|
Checks interface parameters from Netbox and
|
||||||
|
creates a model for the interface to be used in Zabbix.
|
||||||
|
"""
|
||||||
|
try:
|
||||||
|
# Initiate interface class
|
||||||
|
interface = ZabbixInterface(self.nb.config_context, self.ip)
|
||||||
|
# Check if Netbox has device context.
|
||||||
|
# If not fall back to old config.
|
||||||
|
if interface.get_context():
|
||||||
|
# If device is SNMP type, add aditional information.
|
||||||
|
if interface.interface["type"] == 2:
|
||||||
|
interface.set_snmp()
|
||||||
|
else:
|
||||||
|
interface.set_default()
|
||||||
|
return [interface.interface]
|
||||||
|
except InterfaceConfigError as e:
|
||||||
|
e = f"{self.name}: {e}"
|
||||||
|
self.logger.warning(e)
|
||||||
|
raise SyncInventoryError(e) from e
|
||||||
|
|
||||||
|
def setProxy(self, proxy_list):
|
||||||
|
"""
|
||||||
|
Sets proxy or proxy group if this
|
||||||
|
value has been defined in config context
|
||||||
|
|
||||||
|
input: List of all proxies and proxy gorups in standardized format
|
||||||
|
"""
|
||||||
|
# check if the key Zabbix is defined in the config context
|
||||||
|
if not "zabbix" in self.nb.config_context:
|
||||||
|
return False
|
||||||
|
# Proxy group takes priority over a proxy due
|
||||||
|
# to it being HA and therefore being more reliable
|
||||||
|
# Includes proxy group fix since Zabbix <= 6 should ignore this
|
||||||
|
proxy_types = ["proxy"]
|
||||||
|
if str(self.zabbix.version).startswith('7'):
|
||||||
|
# Only insert groups in front of list for Zabbix7
|
||||||
|
proxy_types.insert(0, "proxy_group")
|
||||||
|
for proxy_type in proxy_types:
|
||||||
|
# Check if the key exists in Netbox CC
|
||||||
|
if proxy_type in self.nb.config_context["zabbix"]:
|
||||||
|
proxy_name = self.nb.config_context["zabbix"][proxy_type]
|
||||||
|
# go through all proxies
|
||||||
|
for proxy in proxy_list:
|
||||||
|
# If the proxy does not match the type, ignore and continue
|
||||||
|
if not proxy["type"] == proxy_type:
|
||||||
|
continue
|
||||||
|
# If the proxy name matches
|
||||||
|
if proxy["name"] == proxy_name:
|
||||||
|
self.zbxproxy = proxy
|
||||||
|
return True
|
||||||
|
else:
|
||||||
|
self.logger.warning(f"Device {self.name}: unable to find proxy {proxy_name}")
|
||||||
|
break
|
||||||
|
return False
|
||||||
|
|
||||||
|
def createInZabbix(self, groups, templates, proxies,
|
||||||
|
description="Host added by Netbox sync script."):
|
||||||
|
"""
|
||||||
|
Creates Zabbix host object with parameters from Netbox object.
|
||||||
|
"""
|
||||||
|
# Check if hostname is already present in Zabbix
|
||||||
|
if not self._zabbixHostnameExists():
|
||||||
|
# Get group and template ID's for host
|
||||||
|
if not self.getZabbixGroup(groups):
|
||||||
|
e = (f"Unable to find group '{self.hostgroup}' "
|
||||||
|
f"for host {self.name} in Zabbix.")
|
||||||
|
self.logger.warning(e)
|
||||||
|
raise SyncInventoryError(e)
|
||||||
|
self.zbxTemplatePrepper(templates)
|
||||||
|
templateids = []
|
||||||
|
for template in self.zbx_templates:
|
||||||
|
templateids.append({'templateid': template['templateid']})
|
||||||
|
# Set interface, group and template configuration
|
||||||
|
interfaces = self.setInterfaceDetails()
|
||||||
|
groups = [{"groupid": self.group_id}]
|
||||||
|
# Set Zabbix proxy if defined
|
||||||
|
self.setProxy(proxies)
|
||||||
|
# Set basic data for host creation
|
||||||
|
create_data = {"host": self.name,
|
||||||
|
"name": self.visible_name,
|
||||||
|
"status": self.zabbix_state,
|
||||||
|
"interfaces": interfaces,
|
||||||
|
"groups": groups,
|
||||||
|
"templates": templateids,
|
||||||
|
"description": description,
|
||||||
|
"inventory_mode": self.inventory_mode,
|
||||||
|
"inventory": self.inventory
|
||||||
|
}
|
||||||
|
# If a Zabbix proxy or Zabbix Proxy group has been defined
|
||||||
|
if self.zbxproxy:
|
||||||
|
# If a lower version than 7 is used, we can assume that
|
||||||
|
# the proxy is a normal proxy and not a proxy group
|
||||||
|
if not str(self.zabbix.version).startswith('7'):
|
||||||
|
create_data["proxy_hostid"] = self.zbxproxy["id"]
|
||||||
|
else:
|
||||||
|
# Configure either a proxy or proxy group
|
||||||
|
create_data[self.zbxproxy["idtype"]] = self.zbxproxy["id"]
|
||||||
|
create_data["monitored_by"] = self.zbxproxy["monitored_by"]
|
||||||
|
# Add host to Zabbix
|
||||||
|
try:
|
||||||
|
host = self.zabbix.host.create(**create_data)
|
||||||
|
self.zabbix_id = host["hostids"][0]
|
||||||
|
except APIRequestError as e:
|
||||||
|
e = f"Couldn't add {self.name}, Zabbix returned {str(e)}."
|
||||||
|
self.logger.error(e)
|
||||||
|
raise SyncExternalError(e) from e
|
||||||
|
# Set Netbox custom field to hostID value.
|
||||||
|
self.nb.custom_fields[device_cf] = int(self.zabbix_id)
|
||||||
|
self.nb.save()
|
||||||
|
msg = f"Created host {self.name} in Zabbix."
|
||||||
|
self.logger.info(msg)
|
||||||
|
self.create_journal_entry("success", msg)
|
||||||
|
else:
|
||||||
|
e = f"Unable to add {self.name} to Zabbix: host already present."
|
||||||
|
self.logger.warning(e)
|
||||||
|
|
||||||
|
def createZabbixHostgroup(self):
|
||||||
|
"""
|
||||||
|
Creates Zabbix host group based on hostgroup format.
|
||||||
|
"""
|
||||||
|
try:
|
||||||
|
groupid = self.zabbix.hostgroup.create(name=self.hostgroup)
|
||||||
|
e = f"Added hostgroup '{self.hostgroup}'."
|
||||||
|
self.logger.info(e)
|
||||||
|
data = {'groupid': groupid["groupids"][0], 'name': self.hostgroup}
|
||||||
|
return data
|
||||||
|
except APIRequestError as e:
|
||||||
|
e = f"Couldn't add hostgroup, Zabbix returned {str(e)}."
|
||||||
|
self.logger.error(e)
|
||||||
|
raise SyncExternalError(e) from e
|
||||||
|
|
||||||
|
def updateZabbixHost(self, **kwargs):
|
||||||
|
"""
|
||||||
|
Updates Zabbix host with given parameters.
|
||||||
|
INPUT: Key word arguments for Zabbix host object.
|
||||||
|
"""
|
||||||
|
try:
|
||||||
|
self.zabbix.host.update(hostid=self.zabbix_id, **kwargs)
|
||||||
|
except APIRequestError as e:
|
||||||
|
e = f"Zabbix returned the following error: {str(e)}."
|
||||||
|
self.logger.error(e)
|
||||||
|
raise SyncExternalError(e) from None
|
||||||
|
self.logger.info(f"Updated host {self.name} with data {kwargs}.")
|
||||||
|
self.create_journal_entry("info", "Updated host in Zabbix with latest NB data.")
|
||||||
|
|
||||||
|
def ConsistencyCheck(self, groups, templates, proxies, proxy_power, create_hostgroups):
|
||||||
|
# pylint: disable=too-many-branches, too-many-statements
|
||||||
|
"""
|
||||||
|
Checks if Zabbix object is still valid with Netbox parameters.
|
||||||
|
"""
|
||||||
|
# Check if the hostgroup exists.
|
||||||
|
# If not, create the hostgroup and try finding the group again
|
||||||
|
if not self.getZabbixGroup(groups):
|
||||||
|
if create_hostgroups:
|
||||||
|
new_group = self.createZabbixHostgroup()
|
||||||
|
groups.append(new_group)
|
||||||
|
self.getZabbixGroup(groups)
|
||||||
|
else:
|
||||||
|
e = (f"Device {self.name}: different hostgroup is required but "
|
||||||
|
"unable to create hostgroup without generation permission.")
|
||||||
|
self.logger.warning(e)
|
||||||
|
raise SyncInventoryError(e)
|
||||||
|
self.zbxTemplatePrepper(templates)
|
||||||
|
self.setProxy(proxies)
|
||||||
|
host = self.zabbix.host.get(filter={'hostid': self.zabbix_id},
|
||||||
|
selectInterfaces=['type', 'ip',
|
||||||
|
'port', 'details',
|
||||||
|
'interfaceid'],
|
||||||
|
selectGroups=["groupid"],
|
||||||
|
selectParentTemplates=["templateid"],
|
||||||
|
selectInventory=list(inventory_map.values()))
|
||||||
|
if len(host) > 1:
|
||||||
|
e = (f"Got {len(host)} results for Zabbix hosts "
|
||||||
|
f"with ID {self.zabbix_id} - hostname {self.name}.")
|
||||||
|
self.logger.error(e)
|
||||||
|
raise SyncInventoryError(e)
|
||||||
|
if len(host) == 0:
|
||||||
|
e = (f"No Zabbix host found for {self.name}. "
|
||||||
|
f"This is likely the result of a deleted Zabbix host "
|
||||||
|
f"without zeroing the ID field in Netbox.")
|
||||||
|
self.logger.error(e)
|
||||||
|
raise SyncInventoryError(e)
|
||||||
|
host = host[0]
|
||||||
|
if host["host"] == self.name:
|
||||||
|
self.logger.debug(f"Device {self.name}: hostname in-sync.")
|
||||||
|
else:
|
||||||
|
self.logger.warning(f"Device {self.name}: hostname OUT of sync. "
|
||||||
|
f"Received value: {host['host']}")
|
||||||
|
self.updateZabbixHost(host=self.name)
|
||||||
|
# Execute check depending on wether the name is special or not
|
||||||
|
if self.use_visible_name:
|
||||||
|
if host["name"] == self.visible_name:
|
||||||
|
self.logger.debug(f"Device {self.name}: visible name in-sync.")
|
||||||
|
else:
|
||||||
|
self.logger.warning(f"Device {self.name}: visible name OUT of sync."
|
||||||
|
f" Received value: {host['name']}")
|
||||||
|
self.updateZabbixHost(name=self.visible_name)
|
||||||
|
|
||||||
|
# Check if the templates are in-sync
|
||||||
|
if not self.zbx_template_comparer(host["parentTemplates"]):
|
||||||
|
self.logger.warning(f"Device {self.name}: template(s) OUT of sync.")
|
||||||
|
# Update Zabbix with NB templates and clear any old / lost templates
|
||||||
|
self.updateZabbixHost(templates_clear=host["parentTemplates"],
|
||||||
|
templates=self.zbx_templates)
|
||||||
|
else:
|
||||||
|
self.logger.debug(f"Device {self.name}: template(s) in-sync.")
|
||||||
|
|
||||||
|
for group in host["groups"]:
|
||||||
|
if group["groupid"] == self.group_id:
|
||||||
|
self.logger.debug(f"Device {self.name}: hostgroup in-sync.")
|
||||||
|
break
|
||||||
|
else:
|
||||||
|
self.logger.warning(f"Device {self.name}: hostgroup OUT of sync.")
|
||||||
|
self.updateZabbixHost(groups={'groupid': self.group_id})
|
||||||
|
|
||||||
|
if int(host["status"]) == self.zabbix_state:
|
||||||
|
self.logger.debug(f"Device {self.name}: status in-sync.")
|
||||||
|
else:
|
||||||
|
self.logger.warning(f"Device {self.name}: status OUT of sync.")
|
||||||
|
self.updateZabbixHost(status=str(self.zabbix_state))
|
||||||
|
# Check if a proxy has been defined
|
||||||
|
if self.zbxproxy:
|
||||||
|
# Check if proxy or proxy group is defined
|
||||||
|
if (self.zbxproxy["idtype"] in host and
|
||||||
|
host[self.zbxproxy["idtype"]] == self.zbxproxy["id"]):
|
||||||
|
self.logger.debug(f"Device {self.name}: proxy in-sync.")
|
||||||
|
# Backwards compatibility for Zabbix <= 6
|
||||||
|
elif "proxy_hostid" in host and host["proxy_hostid"] == self.zbxproxy["id"]:
|
||||||
|
self.logger.debug(f"Device {self.name}: proxy in-sync.")
|
||||||
|
# Proxy does not match, update Zabbix
|
||||||
|
else:
|
||||||
|
self.logger.warning(f"Device {self.name}: proxy OUT of sync.")
|
||||||
|
# Zabbix <= 6 patch
|
||||||
|
if not str(self.zabbix.version).startswith('7'):
|
||||||
|
self.updateZabbixHost(proxy_hostid=self.zbxproxy['id'])
|
||||||
|
# Zabbix 7+
|
||||||
|
else:
|
||||||
|
# Prepare data structure for updating either proxy or group
|
||||||
|
update_data = {self.zbxproxy["idtype"]: self.zbxproxy["id"],
|
||||||
|
"monitored_by": self.zbxproxy['monitored_by']}
|
||||||
|
self.updateZabbixHost(**update_data)
|
||||||
|
else:
|
||||||
|
# No proxy is defined in Netbox
|
||||||
|
proxy_set = False
|
||||||
|
# Check if a proxy is defined. Uses the proxy_hostid key for backwards compatibility
|
||||||
|
for key in ("proxy_hostid", "proxyid", "proxy_groupid"):
|
||||||
|
if key in host:
|
||||||
|
if bool(int(host[key])):
|
||||||
|
proxy_set = True
|
||||||
|
if proxy_power and proxy_set:
|
||||||
|
# Zabbix <= 6 fix
|
||||||
|
self.logger.warning(f"Device {self.name}: no proxy is configured in Netbox "
|
||||||
|
"but is configured in Zabbix. Removing proxy config in Zabbix")
|
||||||
|
if "proxy_hostid" in host and bool(host["proxy_hostid"]):
|
||||||
|
self.updateZabbixHost(proxy_hostid=0)
|
||||||
|
# Zabbix 7 proxy
|
||||||
|
elif "proxyid" in host and bool(host["proxyid"]):
|
||||||
|
self.updateZabbixHost(proxyid=0, monitored_by=0)
|
||||||
|
# Zabbix 7 proxy group
|
||||||
|
elif "proxy_groupid" in host and bool(host["proxy_groupid"]):
|
||||||
|
self.updateZabbixHost(proxy_groupid=0, monitored_by=0)
|
||||||
|
# Checks if a proxy has been defined in Zabbix and if proxy_power config has been set
|
||||||
|
if proxy_set and not proxy_power:
|
||||||
|
# Display error message
|
||||||
|
self.logger.error(f"Device {self.name} is configured "
|
||||||
|
f"with proxy in Zabbix but not in Netbox. The"
|
||||||
|
" -p flag was ommited: no "
|
||||||
|
"changes have been made.")
|
||||||
|
if not proxy_set:
|
||||||
|
self.logger.debug(f"Device {self.name}: proxy in-sync.")
|
||||||
|
# Check host inventory
|
||||||
|
if inventory_sync:
|
||||||
|
# check inventory mode first, as we need it set to parse
|
||||||
|
# actual inventory values
|
||||||
|
if str(host['inventory_mode']) == str(self.inventory_mode):
|
||||||
|
self.logger.debug(f"Device {self.name}: inventory_mode in-sync.")
|
||||||
|
else:
|
||||||
|
self.logger.warning(f"Device {self.name}: inventory_mode OUT of sync.")
|
||||||
|
self.updateZabbixHost(inventory_mode=str(self.inventory_mode))
|
||||||
|
# Now we can check if inventory is in-sync.
|
||||||
|
if host['inventory'] == self.inventory:
|
||||||
|
self.logger.debug(f"Device {self.name}: inventory in-sync.")
|
||||||
|
else:
|
||||||
|
self.logger.warning(f"Device {self.name}: inventory OUT of sync.")
|
||||||
|
self.updateZabbixHost(inventory=self.inventory)
|
||||||
|
|
||||||
|
# If only 1 interface has been found
|
||||||
|
# pylint: disable=too-many-nested-blocks
|
||||||
|
if len(host['interfaces']) == 1:
|
||||||
|
updates = {}
|
||||||
|
# Go through each key / item and check if it matches Zabbix
|
||||||
|
for key, item in self.setInterfaceDetails()[0].items():
|
||||||
|
# Check if Netbox value is found in Zabbix
|
||||||
|
if key in host["interfaces"][0]:
|
||||||
|
# If SNMP is used, go through nested dict
|
||||||
|
# to compare SNMP parameters
|
||||||
|
if isinstance(item,dict) and key == "details":
|
||||||
|
for k, i in item.items():
|
||||||
|
if k in host["interfaces"][0][key]:
|
||||||
|
# Set update if values don't match
|
||||||
|
if host["interfaces"][0][key][k] != str(i):
|
||||||
|
# If dict has not been created, add it
|
||||||
|
if key not in updates:
|
||||||
|
updates[key] = {}
|
||||||
|
updates[key][k] = str(i)
|
||||||
|
# If SNMP version has been changed
|
||||||
|
# break loop and force full SNMP update
|
||||||
|
if k == "version":
|
||||||
|
break
|
||||||
|
# Force full SNMP config update
|
||||||
|
# when version has changed.
|
||||||
|
if key in updates:
|
||||||
|
if "version" in updates[key]:
|
||||||
|
for k, i in item.items():
|
||||||
|
updates[key][k] = str(i)
|
||||||
|
continue
|
||||||
|
# Set update if values don't match
|
||||||
|
if host["interfaces"][0][key] != str(item):
|
||||||
|
updates[key] = item
|
||||||
|
if updates:
|
||||||
|
# If interface updates have been found: push to Zabbix
|
||||||
|
self.logger.warning(f"Device {self.name}: Interface OUT of sync.")
|
||||||
|
if "type" in updates:
|
||||||
|
# Changing interface type not supported. Raise exception.
|
||||||
|
e = (f"Device {self.name}: changing interface type to "
|
||||||
|
f"{str(updates['type'])} is not supported.")
|
||||||
|
self.logger.error(e)
|
||||||
|
raise InterfaceConfigError(e)
|
||||||
|
# Set interfaceID for Zabbix config
|
||||||
|
updates["interfaceid"] = host["interfaces"][0]['interfaceid']
|
||||||
|
try:
|
||||||
|
# API call to Zabbix
|
||||||
|
self.zabbix.hostinterface.update(updates)
|
||||||
|
e = f"Solved {self.name} interface conflict."
|
||||||
|
self.logger.info(e)
|
||||||
|
self.create_journal_entry("info", e)
|
||||||
|
except APIRequestError as e:
|
||||||
|
e = f"Zabbix returned the following error: {str(e)}."
|
||||||
|
self.logger.error(e)
|
||||||
|
raise SyncExternalError(e) from e
|
||||||
|
else:
|
||||||
|
# If no updates are found, Zabbix interface is in-sync
|
||||||
|
e = f"Device {self.name}: interface in-sync."
|
||||||
|
self.logger.debug(e)
|
||||||
|
else:
|
||||||
|
e = (f"Device {self.name} has unsupported interface configuration."
|
||||||
|
f" Host has total of {len(host['interfaces'])} interfaces. "
|
||||||
|
"Manual interfention required.")
|
||||||
|
self.logger.error(e)
|
||||||
|
raise SyncInventoryError(e)
|
||||||
|
|
||||||
|
def create_journal_entry(self, severity, message):
|
||||||
|
"""
|
||||||
|
Send a new Journal entry to Netbox. Usefull for viewing actions
|
||||||
|
in Netbox without having to look in Zabbix or the script log output
|
||||||
|
"""
|
||||||
|
if self.journal:
|
||||||
|
# Check if the severity is valid
|
||||||
|
if severity not in ["info", "success", "warning", "danger"]:
|
||||||
|
self.logger.warning(f"Value {severity} not valid for NB journal entries.")
|
||||||
|
return False
|
||||||
|
journal = {"assigned_object_type": "dcim.device",
|
||||||
|
"assigned_object_id": self.id,
|
||||||
|
"kind": severity,
|
||||||
|
"comments": message
|
||||||
|
}
|
||||||
|
try:
|
||||||
|
self.nb_journals.create(journal)
|
||||||
|
self.logger.debug(f"Created journal entry in NB for host {self.name}")
|
||||||
|
return True
|
||||||
|
except JournalError(e) as e:
|
||||||
|
self.logger.warning("Unable to create journal entry for "
|
||||||
|
f"{self.name}: NB returned {e}")
|
||||||
|
return False
|
||||||
|
return False
|
||||||
|
|
||||||
|
def zbx_template_comparer(self, tmpls_from_zabbix):
|
||||||
|
"""
|
||||||
|
Compares the Netbox and Zabbix templates with each other.
|
||||||
|
Should there be a mismatch then the function will return false
|
||||||
|
|
||||||
|
INPUT: list of NB and ZBX templates
|
||||||
|
OUTPUT: Boolean True/False
|
||||||
|
"""
|
||||||
|
succesfull_templates = []
|
||||||
|
# Go through each Netbox template
|
||||||
|
for nb_tmpl in self.zbx_templates:
|
||||||
|
# Go through each Zabbix template
|
||||||
|
for pos, zbx_tmpl in enumerate(tmpls_from_zabbix):
|
||||||
|
# Check if template IDs match
|
||||||
|
if nb_tmpl["templateid"] == zbx_tmpl["templateid"]:
|
||||||
|
# Templates match. Remove this template from the Zabbix templates
|
||||||
|
# and add this NB template to the list of successfull templates
|
||||||
|
tmpls_from_zabbix.pop(pos)
|
||||||
|
succesfull_templates.append(nb_tmpl)
|
||||||
|
self.logger.debug(f"Device {self.name}: template "
|
||||||
|
f"{nb_tmpl['name']} is present in Zabbix.")
|
||||||
|
break
|
||||||
|
if len(succesfull_templates) == len(self.zbx_templates) and len(tmpls_from_zabbix) == 0:
|
||||||
|
# All of the Netbox templates have been confirmed as successfull
|
||||||
|
# and the ZBX template list is empty. This means that
|
||||||
|
# all of the templates match.
|
||||||
|
return True
|
||||||
|
return False
|
33
modules/exceptions.py
Normal file
33
modules/exceptions.py
Normal file
@ -0,0 +1,33 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
"""
|
||||||
|
All custom exceptions used for Exception generation
|
||||||
|
"""
|
||||||
|
class SyncError(Exception):
|
||||||
|
""" Class SyncError """
|
||||||
|
|
||||||
|
class JournalError(Exception):
|
||||||
|
""" Class SyncError """
|
||||||
|
|
||||||
|
class SyncExternalError(SyncError):
|
||||||
|
""" Class SyncExternalError """
|
||||||
|
|
||||||
|
class SyncInventoryError(SyncError):
|
||||||
|
""" Class SyncInventoryError """
|
||||||
|
|
||||||
|
class SyncDuplicateError(SyncError):
|
||||||
|
""" Class SyncDuplicateError """
|
||||||
|
|
||||||
|
class EnvironmentVarError(SyncError):
|
||||||
|
""" Class EnvironmentVarError """
|
||||||
|
|
||||||
|
class InterfaceConfigError(SyncError):
|
||||||
|
""" Class InterfaceConfigError """
|
||||||
|
|
||||||
|
class ProxyConfigError(SyncError):
|
||||||
|
""" Class ProxyConfigError """
|
||||||
|
|
||||||
|
class HostgroupError(SyncError):
|
||||||
|
""" Class HostgroupError """
|
||||||
|
|
||||||
|
class TemplateError(SyncError):
|
||||||
|
""" Class TemplateError """
|
82
modules/interface.py
Normal file
82
modules/interface.py
Normal file
@ -0,0 +1,82 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
"""
|
||||||
|
All of the Zabbix interface related configuration
|
||||||
|
"""
|
||||||
|
from modules.exceptions import InterfaceConfigError
|
||||||
|
|
||||||
|
class ZabbixInterface():
|
||||||
|
"""Class that represents a Zabbix interface."""
|
||||||
|
|
||||||
|
def __init__(self, context, ip):
|
||||||
|
self.context = context
|
||||||
|
self.ip = ip
|
||||||
|
self.skelet = {"main": "1", "useip": "1", "dns": "", "ip": self.ip}
|
||||||
|
self.interface = self.skelet
|
||||||
|
|
||||||
|
def get_context(self):
|
||||||
|
""" check if Netbox custom context has been defined. """
|
||||||
|
if "zabbix" in self.context:
|
||||||
|
zabbix = self.context["zabbix"]
|
||||||
|
if("interface_type" in zabbix and "interface_port" in zabbix):
|
||||||
|
self.interface["type"] = zabbix["interface_type"]
|
||||||
|
self.interface["port"] = zabbix["interface_port"]
|
||||||
|
return True
|
||||||
|
return False
|
||||||
|
return False
|
||||||
|
|
||||||
|
def set_snmp(self):
|
||||||
|
""" Check if interface is type SNMP """
|
||||||
|
# pylint: disable=too-many-branches
|
||||||
|
if self.interface["type"] == 2:
|
||||||
|
# Checks if SNMP settings are defined in Netbox
|
||||||
|
if "snmp" in self.context["zabbix"]:
|
||||||
|
snmp = self.context["zabbix"]["snmp"]
|
||||||
|
self.interface["details"] = {}
|
||||||
|
# Checks if bulk config has been defined
|
||||||
|
if "bulk" in snmp:
|
||||||
|
self.interface["details"]["bulk"] = str(snmp.pop("bulk"))
|
||||||
|
else:
|
||||||
|
# Fallback to bulk enabled if not specified
|
||||||
|
self.interface["details"]["bulk"] = "1"
|
||||||
|
# SNMP Version config is required in Netbox config context
|
||||||
|
if snmp.get("version"):
|
||||||
|
self.interface["details"]["version"] = str(snmp.pop("version"))
|
||||||
|
else:
|
||||||
|
e = "SNMP version option is not defined."
|
||||||
|
raise InterfaceConfigError(e)
|
||||||
|
# If version 1 or 2 is used, get community string
|
||||||
|
if self.interface["details"]["version"] in ['1','2']:
|
||||||
|
if "community" in snmp:
|
||||||
|
# Set SNMP community to confix context value
|
||||||
|
community = snmp["community"]
|
||||||
|
else:
|
||||||
|
# Set SNMP community to default
|
||||||
|
community = "{$SNMP_COMMUNITY}"
|
||||||
|
self.interface["details"]["community"] = str(community)
|
||||||
|
# If version 3 has been used, get all
|
||||||
|
# SNMPv3 Netbox related configs
|
||||||
|
elif self.interface["details"]["version"] == '3':
|
||||||
|
items = ["securityname", "securitylevel", "authpassphrase",
|
||||||
|
"privpassphrase", "authprotocol", "privprotocol",
|
||||||
|
"contextname"]
|
||||||
|
for key, item in snmp.items():
|
||||||
|
if key in items:
|
||||||
|
self.interface["details"][key] = str(item)
|
||||||
|
else:
|
||||||
|
e = "Unsupported SNMP version."
|
||||||
|
raise InterfaceConfigError(e)
|
||||||
|
else:
|
||||||
|
e = "Interface type SNMP but no parameters provided."
|
||||||
|
raise InterfaceConfigError(e)
|
||||||
|
else:
|
||||||
|
e = "Interface type is not SNMP, unable to set SNMP details"
|
||||||
|
raise InterfaceConfigError(e)
|
||||||
|
|
||||||
|
def set_default(self):
|
||||||
|
""" Set default config to SNMPv2, port 161 and community macro. """
|
||||||
|
self.interface = self.skelet
|
||||||
|
self.interface["type"] = "2"
|
||||||
|
self.interface["port"] = "161"
|
||||||
|
self.interface["details"] = {"version": "2",
|
||||||
|
"community": "{$SNMP_COMMUNITY}",
|
||||||
|
"bulk": "1"}
|
44
modules/tools.py
Normal file
44
modules/tools.py
Normal file
@ -0,0 +1,44 @@
|
|||||||
|
"""A collection of tools used by several classes"""
|
||||||
|
def convert_recordset(recordset):
|
||||||
|
""" Converts netbox RedcordSet to list of dicts. """
|
||||||
|
recordlist = []
|
||||||
|
for record in recordset:
|
||||||
|
recordlist.append(record.__dict__)
|
||||||
|
return recordlist
|
||||||
|
|
||||||
|
def build_path(endpoint, list_of_dicts):
|
||||||
|
"""
|
||||||
|
Builds a path list of related parent/child items.
|
||||||
|
This can be used to generate a joinable list to
|
||||||
|
be used in hostgroups.
|
||||||
|
"""
|
||||||
|
item_path = []
|
||||||
|
itemlist = [i for i in list_of_dicts if i['name'] == endpoint]
|
||||||
|
item = itemlist[0] if len(itemlist) == 1 else None
|
||||||
|
item_path.append(item['name'])
|
||||||
|
while item['_depth'] > 0:
|
||||||
|
itemlist = [i for i in list_of_dicts if i['name'] == str(item['parent'])]
|
||||||
|
item = itemlist[0] if len(itemlist) == 1 else None
|
||||||
|
item_path.append(item['name'])
|
||||||
|
item_path.reverse()
|
||||||
|
return item_path
|
||||||
|
|
||||||
|
def proxy_prepper(proxy_list, proxy_group_list):
|
||||||
|
"""
|
||||||
|
Function that takes 2 lists and converts them using a
|
||||||
|
standardized format for further processing.
|
||||||
|
"""
|
||||||
|
output = []
|
||||||
|
for proxy in proxy_list:
|
||||||
|
proxy["type"] = "proxy"
|
||||||
|
proxy["id"] = proxy["proxyid"]
|
||||||
|
proxy["idtype"] = "proxyid"
|
||||||
|
proxy["monitored_by"] = 1
|
||||||
|
output.append(proxy)
|
||||||
|
for group in proxy_group_list:
|
||||||
|
group["type"] = "proxy_group"
|
||||||
|
group["id"] = group["proxy_groupid"]
|
||||||
|
group["idtype"] = "proxy_groupid"
|
||||||
|
group["monitored_by"] = 2
|
||||||
|
output.append(group)
|
||||||
|
return output
|
@ -1,35 +1,30 @@
|
|||||||
#!/usr/bin/env python3
|
#!/usr/bin/env python3
|
||||||
# pylint: disable=invalid-name, logging-not-lazy, too-many-locals, logging-fstring-interpolation, too-many-lines
|
# pylint: disable=invalid-name, logging-not-lazy, too-many-locals, logging-fstring-interpolation
|
||||||
|
|
||||||
|
|
||||||
"""Netbox to Zabbix sync script."""
|
"""Netbox to Zabbix sync script."""
|
||||||
import logging
|
import logging
|
||||||
import argparse
|
import argparse
|
||||||
from os import environ, path, sys
|
from os import environ, path, sys
|
||||||
from packaging import version
|
|
||||||
from pynetbox import api
|
from pynetbox import api
|
||||||
from pyzabbix import ZabbixAPI, ZabbixAPIException
|
from zabbix_utils import ZabbixAPI, APIRequestError, ProcessingError
|
||||||
|
from modules.device import NetworkDevice
|
||||||
|
from modules.tools import convert_recordset, proxy_prepper
|
||||||
|
from modules.exceptions import EnvironmentVarError, HostgroupError, SyncError
|
||||||
try:
|
try:
|
||||||
from config import (
|
from config import (
|
||||||
templates_config_context,
|
templates_config_context,
|
||||||
templates_config_context_overrule,
|
templates_config_context_overrule,
|
||||||
clustering, create_hostgroups,
|
clustering, create_hostgroups,
|
||||||
create_journal, full_proxy_sync,
|
create_journal, full_proxy_sync,
|
||||||
template_cf, device_cf,
|
|
||||||
zabbix_device_removal,
|
zabbix_device_removal,
|
||||||
zabbix_device_disable,
|
zabbix_device_disable,
|
||||||
hostgroup_format,
|
hostgroup_format,
|
||||||
traverse_site_groups,
|
|
||||||
traverse_regions,
|
|
||||||
inventory_sync,
|
|
||||||
inventory_automatic,
|
|
||||||
inventory_map,
|
|
||||||
nb_device_filter
|
nb_device_filter
|
||||||
)
|
)
|
||||||
except ModuleNotFoundError:
|
except ModuleNotFoundError:
|
||||||
print("Configuration file config.py not found in main directory."
|
print("Configuration file config.py not found in main directory."
|
||||||
"Please create the file or rename the config.py.example file to config.py.")
|
"Please create the file or rename the config.py.example file to config.py.")
|
||||||
sys.exit(0)
|
sys.exit(1)
|
||||||
|
|
||||||
# Set logging
|
# Set logging
|
||||||
log_format = logging.Formatter('%(asctime)s - %(name)s - '
|
log_format = logging.Formatter('%(asctime)s - %(name)s - '
|
||||||
@ -48,31 +43,6 @@ logger.addHandler(lgout)
|
|||||||
logger.addHandler(lgfile)
|
logger.addHandler(lgfile)
|
||||||
logger.setLevel(logging.WARNING)
|
logger.setLevel(logging.WARNING)
|
||||||
|
|
||||||
|
|
||||||
def convert_recordset(recordset):
|
|
||||||
""" Converts netbox RedcordSet to list of dicts. """
|
|
||||||
recordlist = []
|
|
||||||
for record in recordset:
|
|
||||||
recordlist.append(record.__dict__)
|
|
||||||
return recordlist
|
|
||||||
|
|
||||||
def build_path(endpoint, list_of_dicts):
|
|
||||||
"""
|
|
||||||
Builds a path list of related parent/child items.
|
|
||||||
This can be used to generate a joinable list to
|
|
||||||
be used in hostgroups.
|
|
||||||
"""
|
|
||||||
item_path = []
|
|
||||||
itemlist = [i for i in list_of_dicts if i['name'] == endpoint]
|
|
||||||
item = itemlist[0] if len(itemlist) == 1 else None
|
|
||||||
item_path.append(item['name'])
|
|
||||||
while item['_depth'] > 0:
|
|
||||||
itemlist = [i for i in list_of_dicts if i['name'] == str(item['parent'])]
|
|
||||||
item = itemlist[0] if len(itemlist) == 1 else None
|
|
||||||
item_path.append(item['name'])
|
|
||||||
item_path.reverse()
|
|
||||||
return item_path
|
|
||||||
|
|
||||||
def main(arguments):
|
def main(arguments):
|
||||||
"""Run the sync process."""
|
"""Run the sync process."""
|
||||||
# pylint: disable=too-many-branches, too-many-statements
|
# pylint: disable=too-many-branches, too-many-statements
|
||||||
@ -120,19 +90,17 @@ def main(arguments):
|
|||||||
raise HostgroupError(e)
|
raise HostgroupError(e)
|
||||||
# Set Zabbix API
|
# Set Zabbix API
|
||||||
try:
|
try:
|
||||||
zabbix = ZabbixAPI(zabbix_host)
|
if not zabbix_token:
|
||||||
if "ZABBIX_TOKEN" in env_vars:
|
zabbix = ZabbixAPI(zabbix_host, user=zabbix_user, password=zabbix_pass)
|
||||||
zabbix.login(api_token=zabbix_token)
|
|
||||||
else:
|
else:
|
||||||
m=("Logging in with Zabbix user and password,"
|
zabbix = ZabbixAPI(zabbix_host, token=zabbix_token)
|
||||||
" consider using an API token instead.")
|
zabbix.check_auth()
|
||||||
logger.warning(m)
|
except (APIRequestError, ProcessingError) as e:
|
||||||
zabbix.login(zabbix_user, zabbix_pass)
|
e = f"Zabbix returned the following error: {str(e)}"
|
||||||
except ZabbixAPIException as e:
|
|
||||||
e = f"Zabbix returned the following error: {str(e)}."
|
|
||||||
logger.error(e)
|
logger.error(e)
|
||||||
|
sys.exit(1)
|
||||||
# Set API parameter mapping based on API version
|
# Set API parameter mapping based on API version
|
||||||
if version.parse(zabbix.api_version()) < version.parse("7.0.0"):
|
if not str(zabbix.version).startswith('7'):
|
||||||
proxy_name = "host"
|
proxy_name = "host"
|
||||||
else:
|
else:
|
||||||
proxy_name = "name"
|
proxy_name = "name"
|
||||||
@ -144,25 +112,33 @@ def main(arguments):
|
|||||||
zabbix_groups = zabbix.hostgroup.get(output=['groupid', 'name'])
|
zabbix_groups = zabbix.hostgroup.get(output=['groupid', 'name'])
|
||||||
zabbix_templates = zabbix.template.get(output=['templateid', 'name'])
|
zabbix_templates = zabbix.template.get(output=['templateid', 'name'])
|
||||||
zabbix_proxies = zabbix.proxy.get(output=['proxyid', proxy_name])
|
zabbix_proxies = zabbix.proxy.get(output=['proxyid', proxy_name])
|
||||||
# Get Netbox API version
|
# Set empty list for proxy processing Zabbix <= 6
|
||||||
nb_version = netbox.version
|
zabbix_proxygroups = []
|
||||||
# Sanitize data
|
if str(zabbix.version).startswith('7'):
|
||||||
|
zabbix_proxygroups = zabbix.proxygroup.get(output=["proxy_groupid", "name"])
|
||||||
|
# Sanitize proxy data
|
||||||
if proxy_name == "host":
|
if proxy_name == "host":
|
||||||
for proxy in zabbix_proxies:
|
for proxy in zabbix_proxies:
|
||||||
proxy['name'] = proxy.pop('host')
|
proxy['name'] = proxy.pop('host')
|
||||||
|
# Prepare list of all proxy and proxy_groups
|
||||||
|
zabbix_proxy_list = proxy_prepper(zabbix_proxies, zabbix_proxygroups)
|
||||||
|
|
||||||
|
# Get Netbox API version
|
||||||
|
nb_version = netbox.version
|
||||||
|
|
||||||
# Go through all Netbox devices
|
# Go through all Netbox devices
|
||||||
for nb_device in netbox_devices:
|
for nb_device in netbox_devices:
|
||||||
try:
|
try:
|
||||||
|
# Set device instance set data such as hostgroup and template information.
|
||||||
device = NetworkDevice(nb_device, zabbix, netbox_journals, nb_version,
|
device = NetworkDevice(nb_device, zabbix, netbox_journals, nb_version,
|
||||||
create_journal)
|
create_journal, logger)
|
||||||
device.set_hostgroup(hostgroup_format,netbox_site_groups,netbox_regions)
|
device.set_hostgroup(hostgroup_format,netbox_site_groups,netbox_regions)
|
||||||
device.set_template(templates_config_context, templates_config_context_overrule)
|
device.set_template(templates_config_context, templates_config_context_overrule)
|
||||||
device.set_inventory(nb_device)
|
device.set_inventory(nb_device)
|
||||||
# Checks if device is part of cluster.
|
# Checks if device is part of cluster.
|
||||||
# Requires clustering variable
|
# Requires clustering variable
|
||||||
if device.isCluster() and clustering:
|
if device.isCluster() and clustering:
|
||||||
# Check if device is master or slave
|
# Check if device is primary or secondary
|
||||||
if device.promoteMasterDevice():
|
if device.promoteMasterDevice():
|
||||||
e = (f"Device {device.name} is "
|
e = (f"Device {device.name} is "
|
||||||
f"part of cluster and primary.")
|
f"part of cluster and primary.")
|
||||||
@ -187,11 +163,16 @@ def main(arguments):
|
|||||||
logger.info(f"Skipping host {device.name} since its "
|
logger.info(f"Skipping host {device.name} since its "
|
||||||
f"not in the active state.")
|
f"not in the active state.")
|
||||||
continue
|
continue
|
||||||
|
# Check if the device is in the disabled state
|
||||||
if device.status in zabbix_device_disable:
|
if device.status in zabbix_device_disable:
|
||||||
device.zabbix_state = 1
|
device.zabbix_state = 1
|
||||||
else:
|
# Check if device is already in Zabbix
|
||||||
device.zabbix_state = 0
|
if device.zabbix_id:
|
||||||
# Add hostgroup is variable is True
|
device.ConsistencyCheck(zabbix_groups, zabbix_templates,
|
||||||
|
zabbix_proxy_list, full_proxy_sync,
|
||||||
|
create_hostgroups)
|
||||||
|
continue
|
||||||
|
# Add hostgroup is config is set
|
||||||
# and Hostgroup is not present in Zabbix
|
# and Hostgroup is not present in Zabbix
|
||||||
if create_hostgroups:
|
if create_hostgroups:
|
||||||
for group in zabbix_groups:
|
for group in zabbix_groups:
|
||||||
@ -202,807 +183,13 @@ def main(arguments):
|
|||||||
# Create new hostgroup
|
# Create new hostgroup
|
||||||
hostgroup = device.createZabbixHostgroup()
|
hostgroup = device.createZabbixHostgroup()
|
||||||
zabbix_groups.append(hostgroup)
|
zabbix_groups.append(hostgroup)
|
||||||
# Device is already present in Zabbix
|
|
||||||
if device.zabbix_id:
|
|
||||||
device.ConsistencyCheck(zabbix_groups, zabbix_templates,
|
|
||||||
zabbix_proxies, full_proxy_sync)
|
|
||||||
# Add device to Zabbix
|
# Add device to Zabbix
|
||||||
else:
|
|
||||||
device.createInZabbix(zabbix_groups, zabbix_templates,
|
device.createInZabbix(zabbix_groups, zabbix_templates,
|
||||||
zabbix_proxies)
|
zabbix_proxy_list)
|
||||||
except SyncError:
|
except SyncError:
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
|
||||||
class SyncError(Exception):
|
|
||||||
""" Class SyncError """
|
|
||||||
|
|
||||||
class JournalError(Exception):
|
|
||||||
""" Class SyncError """
|
|
||||||
|
|
||||||
class SyncExternalError(SyncError):
|
|
||||||
""" Class SyncExternalError """
|
|
||||||
|
|
||||||
class SyncInventoryError(SyncError):
|
|
||||||
""" Class SyncInventoryError """
|
|
||||||
|
|
||||||
class SyncDuplicateError(SyncError):
|
|
||||||
""" Class SyncDuplicateError """
|
|
||||||
|
|
||||||
class EnvironmentVarError(SyncError):
|
|
||||||
""" Class EnvironmentVarError """
|
|
||||||
|
|
||||||
class InterfaceConfigError(SyncError):
|
|
||||||
""" Class InterfaceConfigError """
|
|
||||||
|
|
||||||
class ProxyConfigError(SyncError):
|
|
||||||
""" Class ProxyConfigError """
|
|
||||||
|
|
||||||
class HostgroupError(SyncError):
|
|
||||||
""" Class HostgroupError """
|
|
||||||
|
|
||||||
class TemplateError(SyncError):
|
|
||||||
""" Class TemplateError """
|
|
||||||
|
|
||||||
class NetworkDevice():
|
|
||||||
# pylint: disable=too-many-instance-attributes, too-many-arguments
|
|
||||||
"""
|
|
||||||
Represents Network device.
|
|
||||||
INPUT: (Netbox device class, ZabbixAPI class, journal flag, NB journal class)
|
|
||||||
"""
|
|
||||||
|
|
||||||
def __init__(self, nb, zabbix, nb_journal_class, nb_version, journal=None):
|
|
||||||
self.nb = nb
|
|
||||||
self.id = nb.id
|
|
||||||
self.name = nb.name
|
|
||||||
self.visible_name = None
|
|
||||||
self.status = nb.status.label
|
|
||||||
self.zabbix = zabbix
|
|
||||||
self.zabbix_id = None
|
|
||||||
self.group_id = None
|
|
||||||
self.nb_api_version = nb_version
|
|
||||||
self.zbx_template_names = []
|
|
||||||
self.zbx_templates = []
|
|
||||||
self.hostgroup = None
|
|
||||||
self.tenant = nb.tenant
|
|
||||||
self.config_context = nb.config_context
|
|
||||||
self.zbxproxy = "0"
|
|
||||||
self.zabbix_state = 0
|
|
||||||
self.journal = journal
|
|
||||||
self.nb_journals = nb_journal_class
|
|
||||||
self.inventory_mode = -1
|
|
||||||
self.inventory = {}
|
|
||||||
self._setBasics()
|
|
||||||
|
|
||||||
def _setBasics(self):
|
|
||||||
"""
|
|
||||||
Sets basic information like IP address.
|
|
||||||
"""
|
|
||||||
# Return error if device does not have primary IP.
|
|
||||||
if self.nb.primary_ip:
|
|
||||||
self.cidr = self.nb.primary_ip.address
|
|
||||||
self.ip = self.cidr.split("/")[0]
|
|
||||||
else:
|
|
||||||
e = f"Device {self.name}: no primary IP."
|
|
||||||
logger.info(e)
|
|
||||||
raise SyncInventoryError(e)
|
|
||||||
|
|
||||||
# Check if device has custom field for ZBX ID
|
|
||||||
if device_cf in self.nb.custom_fields:
|
|
||||||
self.zabbix_id = self.nb.custom_fields[device_cf]
|
|
||||||
else:
|
|
||||||
e = f"Custom field {device_cf} not found for {self.name}."
|
|
||||||
logger.warning(e)
|
|
||||||
raise SyncInventoryError(e)
|
|
||||||
|
|
||||||
# Validate hostname format.
|
|
||||||
odd_character_list = ["ä", "ö", "ü", "Ä", "Ö", "Ü", "ß"]
|
|
||||||
self.use_visible_name = False
|
|
||||||
if any(letter in self.name for letter in odd_character_list):
|
|
||||||
self.name = f"NETBOX_ID{self.id}"
|
|
||||||
self.visible_name = self.nb.name
|
|
||||||
self.use_visible_name = True
|
|
||||||
logger.info(f"Device {self.visible_name} contains special characters. "
|
|
||||||
f"Using {self.name} as name for the Netbox object "
|
|
||||||
f"and using {self.visible_name} as visible name in Zabbix.")
|
|
||||||
else:
|
|
||||||
pass
|
|
||||||
|
|
||||||
def set_hostgroup(self, hg_format, nb_site_groups, nb_regions):
|
|
||||||
"""Set the hostgroup for this device"""
|
|
||||||
# Get all variables from the NB data
|
|
||||||
dev_location = str(self.nb.location) if self.nb.location else None
|
|
||||||
# Check the Netbox version. Use backwards compatibility for versions 2 and 3.
|
|
||||||
if self.nb_api_version.startswith(("2", "3")):
|
|
||||||
dev_role = self.nb.device_role.name
|
|
||||||
else:
|
|
||||||
dev_role = self.nb.role.name
|
|
||||||
manufacturer = self.nb.device_type.manufacturer.name
|
|
||||||
region = str(self.nb.site.region) if self.nb.site.region else None
|
|
||||||
site = self.nb.site.name
|
|
||||||
site_group = str(self.nb.site.group) if self.nb.site.group else None
|
|
||||||
tenant = str(self.tenant) if self.tenant else None
|
|
||||||
tenant_group = str(self.tenant.group) if tenant else None
|
|
||||||
# Set mapper for string -> variable
|
|
||||||
hostgroup_vars = {"dev_location": dev_location, "dev_role": dev_role,
|
|
||||||
"manufacturer": manufacturer, "region": region,
|
|
||||||
"site": site, "site_group": site_group,
|
|
||||||
"tenant": tenant, "tenant_group": tenant_group}
|
|
||||||
# Generate list based off string input format
|
|
||||||
hg_items = hg_format.split("/")
|
|
||||||
hostgroup = ""
|
|
||||||
# Go through all hostgroup items
|
|
||||||
for item in hg_items:
|
|
||||||
# Check if the variable (such as Tenant) is empty.
|
|
||||||
if not hostgroup_vars[item]:
|
|
||||||
continue
|
|
||||||
# Check if the item is a custom field name
|
|
||||||
if item not in hostgroup_vars:
|
|
||||||
cf_value = self.nb.custom_fields[item] if item in self.nb.custom_fields else None
|
|
||||||
if cf_value:
|
|
||||||
# If there is a cf match, add the value of this cf to the hostgroup
|
|
||||||
hostgroup += cf_value + "/"
|
|
||||||
# Should there not be a match, this means that
|
|
||||||
# the variable is invalid. Skip regardless.
|
|
||||||
continue
|
|
||||||
# Add value of predefined variable to hostgroup format
|
|
||||||
if item == "site_group" and nb_site_groups and traverse_site_groups:
|
|
||||||
group_path = build_path(site_group, nb_site_groups)
|
|
||||||
hostgroup += "/".join(group_path) + "/"
|
|
||||||
elif item == "region" and nb_regions and traverse_regions:
|
|
||||||
region_path = build_path(region, nb_regions)
|
|
||||||
hostgroup += "/".join(region_path) + "/"
|
|
||||||
else:
|
|
||||||
hostgroup += hostgroup_vars[item] + "/"
|
|
||||||
# If the final hostgroup variable is empty
|
|
||||||
if not hostgroup:
|
|
||||||
e = (f"{self.name} has no reliable hostgroup. This is"
|
|
||||||
"most likely due to the use of custom fields that are empty.")
|
|
||||||
logger.error(e)
|
|
||||||
raise SyncInventoryError(e)
|
|
||||||
# Remove final inserted "/" and set hostgroup to class var
|
|
||||||
self.hostgroup = hostgroup.rstrip("/")
|
|
||||||
|
|
||||||
def set_template(self, prefer_config_context, overrule_custom):
|
|
||||||
""" Set Template """
|
|
||||||
self.zbx_template_names = None
|
|
||||||
# Gather templates ONLY from the device specific context
|
|
||||||
if prefer_config_context:
|
|
||||||
try:
|
|
||||||
self.zbx_template_names = self.get_templates_context()
|
|
||||||
except TemplateError as e:
|
|
||||||
logger.warning(e)
|
|
||||||
return True
|
|
||||||
# Gather templates from the custom field but overrule
|
|
||||||
# them should there be any device specific templates
|
|
||||||
if overrule_custom:
|
|
||||||
try:
|
|
||||||
self.zbx_template_names = self.get_templates_context()
|
|
||||||
except TemplateError:
|
|
||||||
pass
|
|
||||||
if not self.zbx_template_names:
|
|
||||||
self.zbx_template_names = self.get_templates_cf()
|
|
||||||
return True
|
|
||||||
# Gather templates ONLY from the custom field
|
|
||||||
self.zbx_template_names = self.get_templates_cf()
|
|
||||||
return True
|
|
||||||
|
|
||||||
def get_templates_cf(self):
|
|
||||||
""" Get template from custom field """
|
|
||||||
# Get Zabbix templates from the device type
|
|
||||||
device_type_cfs = self.nb.device_type.custom_fields
|
|
||||||
# Check if the ZBX Template CF is present
|
|
||||||
if template_cf in device_type_cfs:
|
|
||||||
# Set value to template
|
|
||||||
return [device_type_cfs[template_cf]]
|
|
||||||
# Custom field not found, return error
|
|
||||||
e = (f"Custom field {template_cf} not "
|
|
||||||
f"found for {self.nb.device_type.manufacturer.name}"
|
|
||||||
f" - {self.nb.device_type.display}.")
|
|
||||||
raise TemplateError(e)
|
|
||||||
|
|
||||||
def get_templates_context(self):
|
|
||||||
""" Get Zabbix templates from the device context """
|
|
||||||
if "zabbix" not in self.config_context:
|
|
||||||
e = ("Key 'zabbix' not found in config "
|
|
||||||
f"context for template host {self.name}")
|
|
||||||
raise TemplateError(e)
|
|
||||||
if "templates" not in self.config_context["zabbix"]:
|
|
||||||
e = ("Key 'templates' not found in config "
|
|
||||||
f"context 'zabbix' for template host {self.name}")
|
|
||||||
raise TemplateError(e)
|
|
||||||
return self.config_context["zabbix"]["templates"]
|
|
||||||
|
|
||||||
def set_inventory(self, nbdevice):
|
|
||||||
""" Set host inventory """
|
|
||||||
self.inventory_mode = -1
|
|
||||||
self.inventory = {}
|
|
||||||
if inventory_sync:
|
|
||||||
# Set inventory mode to automatic or manual
|
|
||||||
self.inventory_mode = 1 if inventory_automatic else 0
|
|
||||||
|
|
||||||
# Let's build an inventory dict for each property in the inventory_map
|
|
||||||
for nb_inv_field, zbx_inv_field in inventory_map.items():
|
|
||||||
field_list = nb_inv_field.split("/") # convert str to list based on delimiter
|
|
||||||
# start at the base of the dict...
|
|
||||||
value = nbdevice
|
|
||||||
# ... and step through the dict till we find the needed value
|
|
||||||
for item in field_list:
|
|
||||||
value = value[item] if value else None
|
|
||||||
# Check if the result is usable and expected
|
|
||||||
# We want to apply any int or float 0 values,
|
|
||||||
# even if python thinks those are empty.
|
|
||||||
if ((value and isinstance(value, int | float | str )) or
|
|
||||||
(isinstance(value, int | float) and int(value) ==0)):
|
|
||||||
self.inventory[zbx_inv_field] = str(value)
|
|
||||||
elif not value:
|
|
||||||
# empty value should just be an empty string for API compatibility
|
|
||||||
logger.debug(f"Inventory lookup for '{nb_inv_field}' returned an empty value")
|
|
||||||
self.inventory[zbx_inv_field] = ""
|
|
||||||
else:
|
|
||||||
# Value is not a string or numeral, probably not what the user expected.
|
|
||||||
logger.error(f"Inventory lookup for '{nb_inv_field}' returned"
|
|
||||||
f" an unexpected type: it will be skipped.")
|
|
||||||
return True
|
|
||||||
|
|
||||||
def isCluster(self):
|
|
||||||
"""
|
|
||||||
Checks if device is part of cluster.
|
|
||||||
"""
|
|
||||||
return bool(self.nb.virtual_chassis)
|
|
||||||
|
|
||||||
def getClusterMaster(self):
|
|
||||||
"""
|
|
||||||
Returns chassis master ID.
|
|
||||||
"""
|
|
||||||
if not self.isCluster():
|
|
||||||
e = (f"Unable to proces {self.name} for cluster calculation: "
|
|
||||||
f"not part of a cluster.")
|
|
||||||
logger.warning(e)
|
|
||||||
raise SyncInventoryError(e)
|
|
||||||
if not self.nb.virtual_chassis.master:
|
|
||||||
e = (f"{self.name} is part of a Netbox virtual chassis which does "
|
|
||||||
"not have a master configured. Skipping for this reason.")
|
|
||||||
logger.error(e)
|
|
||||||
raise SyncInventoryError(e)
|
|
||||||
return self.nb.virtual_chassis.master.id
|
|
||||||
|
|
||||||
def promoteMasterDevice(self):
|
|
||||||
"""
|
|
||||||
If device is Primary in cluster,
|
|
||||||
promote device name to the cluster name.
|
|
||||||
Returns True if succesfull, returns False if device is secondary.
|
|
||||||
"""
|
|
||||||
masterid = self.getClusterMaster()
|
|
||||||
if masterid == self.id:
|
|
||||||
logger.debug(f"Device {self.name} is primary cluster member. "
|
|
||||||
f"Modifying hostname from {self.name} to " +
|
|
||||||
f"{self.nb.virtual_chassis.name}.")
|
|
||||||
self.name = self.nb.virtual_chassis.name
|
|
||||||
return True
|
|
||||||
logger.debug(f"Device {self.name} is non-primary cluster member.")
|
|
||||||
return False
|
|
||||||
|
|
||||||
def zbxTemplatePrepper(self, templates):
|
|
||||||
"""
|
|
||||||
Returns Zabbix template IDs
|
|
||||||
INPUT: list of templates from Zabbix
|
|
||||||
OUTPUT: True
|
|
||||||
"""
|
|
||||||
# Check if there are templates defined
|
|
||||||
if not self.zbx_template_names:
|
|
||||||
e = f"No templates found for device {self.name}"
|
|
||||||
logger.info(e)
|
|
||||||
raise SyncInventoryError()
|
|
||||||
# Set variable to empty list
|
|
||||||
self.zbx_templates = []
|
|
||||||
# Go through all templates definded in Netbox
|
|
||||||
for nb_template in self.zbx_template_names:
|
|
||||||
template_match = False
|
|
||||||
# Go through all templates found in Zabbix
|
|
||||||
for zbx_template in templates:
|
|
||||||
# If the template names match
|
|
||||||
if zbx_template['name'] == nb_template:
|
|
||||||
# Set match variable to true, add template details
|
|
||||||
# to class variable and return debug log
|
|
||||||
template_match = True
|
|
||||||
self.zbx_templates.append({"templateid": zbx_template['templateid'],
|
|
||||||
"name": zbx_template['name']})
|
|
||||||
e = (f"Found template {zbx_template['name']}"
|
|
||||||
f" for host {self.name}.")
|
|
||||||
logger.debug(e)
|
|
||||||
# Return error should the template not be found in Zabbix
|
|
||||||
if not template_match:
|
|
||||||
e = (f"Unable to find template {nb_template} "
|
|
||||||
f"for host {self.name} in Zabbix. Skipping host...")
|
|
||||||
logger.warning(e)
|
|
||||||
raise SyncInventoryError(e)
|
|
||||||
|
|
||||||
def getZabbixGroup(self, groups):
|
|
||||||
"""
|
|
||||||
Returns Zabbix group ID
|
|
||||||
INPUT: list of hostgroups
|
|
||||||
OUTPUT: True / False
|
|
||||||
"""
|
|
||||||
# Go through all groups
|
|
||||||
for group in groups:
|
|
||||||
if group['name'] == self.hostgroup:
|
|
||||||
self.group_id = group['groupid']
|
|
||||||
e = f"Found group {group['name']} for host {self.name}."
|
|
||||||
logger.debug(e)
|
|
||||||
return True
|
|
||||||
e = (f"Unable to find group '{self.hostgroup}' "
|
|
||||||
f"for host {self.name} in Zabbix.")
|
|
||||||
logger.warning(e)
|
|
||||||
raise SyncInventoryError(e)
|
|
||||||
|
|
||||||
def cleanup(self):
|
|
||||||
"""
|
|
||||||
Removes device from external resources.
|
|
||||||
Resets custom fields in Netbox.
|
|
||||||
"""
|
|
||||||
if self.zabbix_id:
|
|
||||||
try:
|
|
||||||
self.zabbix.host.delete(self.zabbix_id)
|
|
||||||
self.nb.custom_fields[device_cf] = None
|
|
||||||
self.nb.save()
|
|
||||||
e = f"Deleted host {self.name} from Zabbix."
|
|
||||||
logger.info(e)
|
|
||||||
self.create_journal_entry("warning", "Deleted host from Zabbix")
|
|
||||||
except ZabbixAPIException as e:
|
|
||||||
e = f"Zabbix returned the following error: {str(e)}."
|
|
||||||
logger.error(e)
|
|
||||||
raise SyncExternalError(e) from e
|
|
||||||
|
|
||||||
def _zabbixHostnameExists(self):
|
|
||||||
"""
|
|
||||||
Checks if hostname exists in Zabbix.
|
|
||||||
"""
|
|
||||||
# Validate the hostname or visible name field
|
|
||||||
if not self.use_visible_name:
|
|
||||||
zbx_filter = {'host': self.name}
|
|
||||||
else:
|
|
||||||
zbx_filter = {'name': self.visible_name}
|
|
||||||
host = self.zabbix.host.get(filter=zbx_filter, output=[])
|
|
||||||
return bool(host)
|
|
||||||
|
|
||||||
def setInterfaceDetails(self):
|
|
||||||
"""
|
|
||||||
Checks interface parameters from Netbox and
|
|
||||||
creates a model for the interface to be used in Zabbix.
|
|
||||||
"""
|
|
||||||
try:
|
|
||||||
# Initiate interface class
|
|
||||||
interface = ZabbixInterface(self.nb.config_context, self.ip)
|
|
||||||
# Check if Netbox has device context.
|
|
||||||
# If not fall back to old config.
|
|
||||||
if interface.get_context():
|
|
||||||
# If device is SNMP type, add aditional information.
|
|
||||||
if interface.interface["type"] == 2:
|
|
||||||
interface.set_snmp()
|
|
||||||
else:
|
|
||||||
interface.set_default()
|
|
||||||
return [interface.interface]
|
|
||||||
except InterfaceConfigError as e:
|
|
||||||
e = f"{self.name}: {e}"
|
|
||||||
logger.warning(e)
|
|
||||||
raise SyncInventoryError(e) from e
|
|
||||||
|
|
||||||
def setProxy(self, proxy_list):
|
|
||||||
""" check if Zabbix Proxy has been defined in config context """
|
|
||||||
if "zabbix" in self.nb.config_context:
|
|
||||||
if "proxy" in self.nb.config_context["zabbix"]:
|
|
||||||
proxy = self.nb.config_context["zabbix"]["proxy"]
|
|
||||||
# Try matching proxy
|
|
||||||
for px in proxy_list:
|
|
||||||
if px["name"] == proxy:
|
|
||||||
self.zbxproxy = px["proxyid"]
|
|
||||||
logger.debug(f"Found proxy {proxy}"
|
|
||||||
f" for {self.name}.")
|
|
||||||
return True
|
|
||||||
e = f"{self.name}: Defined proxy {proxy} not found."
|
|
||||||
logger.warning(e)
|
|
||||||
return False
|
|
||||||
return True
|
|
||||||
|
|
||||||
def createInZabbix(self, groups, templates, proxies,
|
|
||||||
description="Host added by Netbox sync script."):
|
|
||||||
"""
|
|
||||||
Creates Zabbix host object with parameters from Netbox object.
|
|
||||||
"""
|
|
||||||
# Check if hostname is already present in Zabbix
|
|
||||||
if not self._zabbixHostnameExists():
|
|
||||||
# Get group and template ID's for host
|
|
||||||
if not self.getZabbixGroup(groups):
|
|
||||||
raise SyncInventoryError()
|
|
||||||
self.zbxTemplatePrepper(templates)
|
|
||||||
templateids = []
|
|
||||||
for template in self.zbx_templates:
|
|
||||||
templateids.append({'templateid': template['templateid']})
|
|
||||||
# Set interface, group and template configuration
|
|
||||||
interfaces = self.setInterfaceDetails()
|
|
||||||
groups = [{"groupid": self.group_id}]
|
|
||||||
# Set Zabbix proxy if defined
|
|
||||||
self.setProxy(proxies)
|
|
||||||
# Add host to Zabbix
|
|
||||||
try:
|
|
||||||
if version.parse(self.zabbix.api_version()) < version.parse("7.0.0"):
|
|
||||||
host = self.zabbix.host.create(host=self.name,
|
|
||||||
name=self.visible_name,
|
|
||||||
status=self.zabbix_state,
|
|
||||||
interfaces=interfaces,
|
|
||||||
groups=groups,
|
|
||||||
templates=templateids,
|
|
||||||
proxy_hostid=self.zbxproxy,
|
|
||||||
description=description,
|
|
||||||
inventory_mode=self.inventory_mode,
|
|
||||||
inventory=self.inventory)
|
|
||||||
else:
|
|
||||||
host = self.zabbix.host.create(host=self.name,
|
|
||||||
name=self.visible_name,
|
|
||||||
status=self.zabbix_state,
|
|
||||||
interfaces=interfaces,
|
|
||||||
groups=groups,
|
|
||||||
templates=templateids,
|
|
||||||
proxyid=self.zbxproxy,
|
|
||||||
description=description,
|
|
||||||
inventory_mode=self.inventory_mode,
|
|
||||||
inventory=self.inventory)
|
|
||||||
self.zabbix_id = host["hostids"][0]
|
|
||||||
except ZabbixAPIException as e:
|
|
||||||
e = f"Couldn't add {self.name}, Zabbix returned {str(e)}."
|
|
||||||
logger.error(e)
|
|
||||||
raise SyncExternalError(e) from e
|
|
||||||
# Set Netbox custom field to hostID value.
|
|
||||||
self.nb.custom_fields[device_cf] = int(self.zabbix_id)
|
|
||||||
self.nb.save()
|
|
||||||
msg = f"Created host {self.name} in Zabbix."
|
|
||||||
logger.info(msg)
|
|
||||||
self.create_journal_entry("success", msg)
|
|
||||||
else:
|
|
||||||
e = f"Unable to add {self.name} to Zabbix: host already present."
|
|
||||||
logger.warning(e)
|
|
||||||
|
|
||||||
def createZabbixHostgroup(self):
|
|
||||||
"""
|
|
||||||
Creates Zabbix host group based on hostgroup format.
|
|
||||||
"""
|
|
||||||
try:
|
|
||||||
groupid = self.zabbix.hostgroup.create(name=self.hostgroup)
|
|
||||||
e = f"Added hostgroup '{self.hostgroup}'."
|
|
||||||
logger.info(e)
|
|
||||||
data = {'groupid': groupid["groupids"][0], 'name': self.hostgroup}
|
|
||||||
return data
|
|
||||||
except ZabbixAPIException as e:
|
|
||||||
e = f"Couldn't add hostgroup, Zabbix returned {str(e)}."
|
|
||||||
logger.error(e)
|
|
||||||
raise SyncExternalError(e) from e
|
|
||||||
|
|
||||||
def updateZabbixHost(self, **kwargs):
|
|
||||||
"""
|
|
||||||
Updates Zabbix host with given parameters.
|
|
||||||
INPUT: Key word arguments for Zabbix host object.
|
|
||||||
"""
|
|
||||||
try:
|
|
||||||
self.zabbix.host.update(hostid=self.zabbix_id, **kwargs)
|
|
||||||
except ZabbixAPIException as e:
|
|
||||||
e = f"Zabbix returned the following error: {str(e)}."
|
|
||||||
logger.error(e)
|
|
||||||
raise SyncExternalError(e) from e
|
|
||||||
logger.info(f"Updated host {self.name} with data {kwargs}.")
|
|
||||||
self.create_journal_entry("info", "Updated host in Zabbix with latest NB data.")
|
|
||||||
|
|
||||||
def ConsistencyCheck(self, groups, templates, proxies, proxy_power):
|
|
||||||
# pylint: disable=too-many-branches, too-many-statements
|
|
||||||
"""
|
|
||||||
Checks if Zabbix object is still valid with Netbox parameters.
|
|
||||||
"""
|
|
||||||
self.getZabbixGroup(groups)
|
|
||||||
self.zbxTemplatePrepper(templates)
|
|
||||||
self.setProxy(proxies)
|
|
||||||
host = self.zabbix.host.get(filter={'hostid': self.zabbix_id},
|
|
||||||
selectInterfaces=['type', 'ip',
|
|
||||||
'port', 'details',
|
|
||||||
'interfaceid'],
|
|
||||||
selectGroups=["groupid"],
|
|
||||||
selectParentTemplates=["templateid"],
|
|
||||||
selectInventory=list(inventory_map.values()))
|
|
||||||
if len(host) > 1:
|
|
||||||
e = (f"Got {len(host)} results for Zabbix hosts "
|
|
||||||
f"with ID {self.zabbix_id} - hostname {self.name}.")
|
|
||||||
logger.error(e)
|
|
||||||
raise SyncInventoryError(e)
|
|
||||||
if len(host) == 0:
|
|
||||||
e = (f"No Zabbix host found for {self.name}. "
|
|
||||||
f"This is likely the result of a deleted Zabbix host "
|
|
||||||
f"without zeroing the ID field in Netbox.")
|
|
||||||
logger.error(e)
|
|
||||||
raise SyncInventoryError(e)
|
|
||||||
host = host[0]
|
|
||||||
if host["host"] == self.name:
|
|
||||||
logger.debug(f"Device {self.name}: hostname in-sync.")
|
|
||||||
else:
|
|
||||||
logger.warning(f"Device {self.name}: hostname OUT of sync. "
|
|
||||||
f"Received value: {host['host']}")
|
|
||||||
self.updateZabbixHost(host=self.name)
|
|
||||||
# Execute check depending on wether the name is special or not
|
|
||||||
if self.use_visible_name:
|
|
||||||
if host["name"] == self.visible_name:
|
|
||||||
logger.debug(f"Device {self.name}: visible name in-sync.")
|
|
||||||
else:
|
|
||||||
logger.warning(f"Device {self.name}: visible name OUT of sync."
|
|
||||||
f" Received value: {host['name']}")
|
|
||||||
self.updateZabbixHost(name=self.visible_name)
|
|
||||||
|
|
||||||
# Check if the templates are in-sync
|
|
||||||
if not self.zbx_template_comparer(host["parentTemplates"]):
|
|
||||||
logger.warning(f"Device {self.name}: template(s) OUT of sync.")
|
|
||||||
# Update Zabbix with NB templates and clear any old / lost templates
|
|
||||||
self.updateZabbixHost(templates_clear=host["parentTemplates"],
|
|
||||||
templates=self.zbx_templates)
|
|
||||||
else:
|
|
||||||
logger.debug(f"Device {self.name}: template(s) in-sync.")
|
|
||||||
|
|
||||||
for group in host["groups"]:
|
|
||||||
if group["groupid"] == self.group_id:
|
|
||||||
logger.debug(f"Device {self.name}: hostgroup in-sync.")
|
|
||||||
break
|
|
||||||
else:
|
|
||||||
logger.warning(f"Device {self.name}: hostgroup OUT of sync.")
|
|
||||||
self.updateZabbixHost(groups={'groupid': self.group_id})
|
|
||||||
|
|
||||||
if int(host["status"]) == self.zabbix_state:
|
|
||||||
logger.debug(f"Device {self.name}: status in-sync.")
|
|
||||||
else:
|
|
||||||
logger.warning(f"Device {self.name}: status OUT of sync.")
|
|
||||||
self.updateZabbixHost(status=str(self.zabbix_state))
|
|
||||||
|
|
||||||
# Check if a proxy has been defined
|
|
||||||
if self.zbxproxy != "0":
|
|
||||||
# Check if expected proxyID matches with configured proxy
|
|
||||||
if (("proxy_hostid" in host and host["proxy_hostid"] == self.zbxproxy) or
|
|
||||||
("proxyid" in host and host["proxyid"] == self.zbxproxy)):
|
|
||||||
logger.debug(f"Device {self.name}: proxy in-sync.")
|
|
||||||
else:
|
|
||||||
# Proxy diff, update value
|
|
||||||
logger.warning(f"Device {self.name}: proxy OUT of sync.")
|
|
||||||
if version.parse(self.zabbix.api_version()) < version.parse("7.0.0"):
|
|
||||||
self.updateZabbixHost(proxy_hostid=self.zbxproxy)
|
|
||||||
else:
|
|
||||||
self.updateZabbixHost(proxyid=self.zbxproxy)
|
|
||||||
else:
|
|
||||||
if (("proxy_hostid" in host and not host["proxy_hostid"] == "0")
|
|
||||||
or ("proxyid" in host and not host["proxyid"] == "0")):
|
|
||||||
if proxy_power:
|
|
||||||
# Variable full_proxy_sync has been enabled
|
|
||||||
# delete the proxy link in Zabbix
|
|
||||||
if version.parse(self.zabbix.api_version()) < version.parse("7.0.0"):
|
|
||||||
self.updateZabbixHost(proxy_hostid=self.zbxproxy)
|
|
||||||
else:
|
|
||||||
self.updateZabbixHost(proxyid=self.zbxproxy)
|
|
||||||
else:
|
|
||||||
# Instead of deleting the proxy config in zabbix and
|
|
||||||
# forcing potential data loss,
|
|
||||||
# an error message is displayed.
|
|
||||||
logger.error(f"Device {self.name} is configured "
|
|
||||||
f"with proxy in Zabbix but not in Netbox. The"
|
|
||||||
" -p flag was ommited: no "
|
|
||||||
"changes have been made.")
|
|
||||||
|
|
||||||
# Check host inventory
|
|
||||||
if inventory_sync:
|
|
||||||
# check inventory mode first, as we need it set to parse
|
|
||||||
# actual inventory values
|
|
||||||
if str(host['inventory_mode']) == str(self.inventory_mode):
|
|
||||||
logger.debug(f"Device {self.name}: inventory_mode in-sync.")
|
|
||||||
else:
|
|
||||||
logger.warning(f"Device {self.name}: inventory_mode OUT of sync.")
|
|
||||||
self.updateZabbixHost(inventory_mode=str(self.inventory_mode))
|
|
||||||
# Now we can check if inventory is in-sync.
|
|
||||||
if host['inventory'] == self.inventory:
|
|
||||||
logger.debug(f"Device {self.name}: inventory in-sync.")
|
|
||||||
else:
|
|
||||||
logger.warning(f"Device {self.name}: inventory OUT of sync.")
|
|
||||||
self.updateZabbixHost(inventory=self.inventory)
|
|
||||||
|
|
||||||
# If only 1 interface has been found
|
|
||||||
# pylint: disable=too-many-nested-blocks
|
|
||||||
if len(host['interfaces']) == 1:
|
|
||||||
updates = {}
|
|
||||||
# Go through each key / item and check if it matches Zabbix
|
|
||||||
for key, item in self.setInterfaceDetails()[0].items():
|
|
||||||
# Check if Netbox value is found in Zabbix
|
|
||||||
if key in host["interfaces"][0]:
|
|
||||||
# If SNMP is used, go through nested dict
|
|
||||||
# to compare SNMP parameters
|
|
||||||
if isinstance(item,dict) and key == "details":
|
|
||||||
for k, i in item.items():
|
|
||||||
if k in host["interfaces"][0][key]:
|
|
||||||
# Set update if values don't match
|
|
||||||
if host["interfaces"][0][key][k] != str(i):
|
|
||||||
# If dict has not been created, add it
|
|
||||||
if key not in updates:
|
|
||||||
updates[key] = {}
|
|
||||||
updates[key][k] = str(i)
|
|
||||||
# If SNMP version has been changed
|
|
||||||
# break loop and force full SNMP update
|
|
||||||
if k == "version":
|
|
||||||
break
|
|
||||||
# Force full SNMP config update
|
|
||||||
# when version has changed.
|
|
||||||
if key in updates:
|
|
||||||
if "version" in updates[key]:
|
|
||||||
for k, i in item.items():
|
|
||||||
updates[key][k] = str(i)
|
|
||||||
continue
|
|
||||||
# Set update if values don't match
|
|
||||||
if host["interfaces"][0][key] != str(item):
|
|
||||||
updates[key] = item
|
|
||||||
if updates:
|
|
||||||
# If interface updates have been found: push to Zabbix
|
|
||||||
logger.warning(f"Device {self.name}: Interface OUT of sync.")
|
|
||||||
if "type" in updates:
|
|
||||||
# Changing interface type not supported. Raise exception.
|
|
||||||
e = (f"Device {self.name}: changing interface type to "
|
|
||||||
f"{str(updates['type'])} is not supported.")
|
|
||||||
logger.error(e)
|
|
||||||
raise InterfaceConfigError(e)
|
|
||||||
# Set interfaceID for Zabbix config
|
|
||||||
updates["interfaceid"] = host["interfaces"][0]['interfaceid']
|
|
||||||
try:
|
|
||||||
# API call to Zabbix
|
|
||||||
self.zabbix.hostinterface.update(updates)
|
|
||||||
e = f"Solved {self.name} interface conflict."
|
|
||||||
logger.info(e)
|
|
||||||
self.create_journal_entry("info", e)
|
|
||||||
except ZabbixAPIException as e:
|
|
||||||
e = f"Zabbix returned the following error: {str(e)}."
|
|
||||||
logger.error(e)
|
|
||||||
raise SyncExternalError(e) from e
|
|
||||||
else:
|
|
||||||
# If no updates are found, Zabbix interface is in-sync
|
|
||||||
e = f"Device {self.name}: interface in-sync."
|
|
||||||
logger.debug(e)
|
|
||||||
else:
|
|
||||||
e = (f"Device {self.name} has unsupported interface configuration."
|
|
||||||
f" Host has total of {len(host['interfaces'])} interfaces. "
|
|
||||||
"Manual interfention required.")
|
|
||||||
logger.error(e)
|
|
||||||
raise SyncInventoryError(e)
|
|
||||||
|
|
||||||
def create_journal_entry(self, severity, message):
|
|
||||||
"""
|
|
||||||
Send a new Journal entry to Netbox. Usefull for viewing actions
|
|
||||||
in Netbox without having to look in Zabbix or the script log output
|
|
||||||
"""
|
|
||||||
if self.journal:
|
|
||||||
# Check if the severity is valid
|
|
||||||
if severity not in ["info", "success", "warning", "danger"]:
|
|
||||||
logger.warning(f"Value {severity} not valid for NB journal entries.")
|
|
||||||
return False
|
|
||||||
journal = {"assigned_object_type": "dcim.device",
|
|
||||||
"assigned_object_id": self.id,
|
|
||||||
"kind": severity,
|
|
||||||
"comments": message
|
|
||||||
}
|
|
||||||
try:
|
|
||||||
self.nb_journals.create(journal)
|
|
||||||
logger.debug(f"Created journal entry in NB for host {self.name}")
|
|
||||||
return True
|
|
||||||
except JournalError(e) as e:
|
|
||||||
logger.warning("Unable to create journal entry for "
|
|
||||||
f"{self.name}: NB returned {e}")
|
|
||||||
return False
|
|
||||||
return False
|
|
||||||
|
|
||||||
def zbx_template_comparer(self, tmpls_from_zabbix):
|
|
||||||
"""
|
|
||||||
Compares the Netbox and Zabbix templates with each other.
|
|
||||||
Should there be a mismatch then the function will return false
|
|
||||||
|
|
||||||
INPUT: list of NB and ZBX templates
|
|
||||||
OUTPUT: Boolean True/False
|
|
||||||
"""
|
|
||||||
succesfull_templates = []
|
|
||||||
# Go through each Netbox template
|
|
||||||
for nb_tmpl in self.zbx_templates:
|
|
||||||
# Go through each Zabbix template
|
|
||||||
for pos, zbx_tmpl in enumerate(tmpls_from_zabbix):
|
|
||||||
# Check if template IDs match
|
|
||||||
if nb_tmpl["templateid"] == zbx_tmpl["templateid"]:
|
|
||||||
# Templates match. Remove this template from the Zabbix templates
|
|
||||||
# and add this NB template to the list of successfull templates
|
|
||||||
tmpls_from_zabbix.pop(pos)
|
|
||||||
succesfull_templates.append(nb_tmpl)
|
|
||||||
logger.debug(f"Device {self.name}: template "
|
|
||||||
f"{nb_tmpl['name']} is present in Zabbix.")
|
|
||||||
break
|
|
||||||
if len(succesfull_templates) == len(self.zbx_templates) and len(tmpls_from_zabbix) == 0:
|
|
||||||
# All of the Netbox templates have been confirmed as successfull
|
|
||||||
# and the ZBX template list is empty. This means that
|
|
||||||
# all of the templates match.
|
|
||||||
return True
|
|
||||||
return False
|
|
||||||
|
|
||||||
|
|
||||||
class ZabbixInterface():
|
|
||||||
"""Class that represents a Zabbix interface."""
|
|
||||||
|
|
||||||
def __init__(self, context, ip):
|
|
||||||
self.context = context
|
|
||||||
self.ip = ip
|
|
||||||
self.skelet = {"main": "1", "useip": "1", "dns": "", "ip": self.ip}
|
|
||||||
self.interface = self.skelet
|
|
||||||
|
|
||||||
def get_context(self):
|
|
||||||
""" check if Netbox custom context has been defined. """
|
|
||||||
if "zabbix" in self.context:
|
|
||||||
zabbix = self.context["zabbix"]
|
|
||||||
if("interface_type" in zabbix and "interface_port" in zabbix):
|
|
||||||
self.interface["type"] = zabbix["interface_type"]
|
|
||||||
self.interface["port"] = zabbix["interface_port"]
|
|
||||||
return True
|
|
||||||
return False
|
|
||||||
return False
|
|
||||||
|
|
||||||
def set_snmp(self):
|
|
||||||
""" Check if interface is type SNMP """
|
|
||||||
# pylint: disable=too-many-branches
|
|
||||||
if self.interface["type"] == 2:
|
|
||||||
# Checks if SNMP settings are defined in Netbox
|
|
||||||
if "snmp" in self.context["zabbix"]:
|
|
||||||
snmp = self.context["zabbix"]["snmp"]
|
|
||||||
self.interface["details"] = {}
|
|
||||||
# Checks if bulk config has been defined
|
|
||||||
if "bulk" in snmp:
|
|
||||||
self.interface["details"]["bulk"] = str(snmp.pop("bulk"))
|
|
||||||
else:
|
|
||||||
# Fallback to bulk enabled if not specified
|
|
||||||
self.interface["details"]["bulk"] = "1"
|
|
||||||
# SNMP Version config is required in Netbox config context
|
|
||||||
if snmp.get("version"):
|
|
||||||
self.interface["details"]["version"] = str(snmp.pop("version"))
|
|
||||||
else:
|
|
||||||
e = "SNMP version option is not defined."
|
|
||||||
raise InterfaceConfigError(e)
|
|
||||||
# If version 1 or 2 is used, get community string
|
|
||||||
if self.interface["details"]["version"] in ['1','2']:
|
|
||||||
if "community" in snmp:
|
|
||||||
# Set SNMP community to confix context value
|
|
||||||
community = snmp["community"]
|
|
||||||
else:
|
|
||||||
# Set SNMP community to default
|
|
||||||
community = "{$SNMP_COMMUNITY}"
|
|
||||||
self.interface["details"]["community"] = str(community)
|
|
||||||
# If version 3 has been used, get all
|
|
||||||
# SNMPv3 Netbox related configs
|
|
||||||
elif self.interface["details"]["version"] == '3':
|
|
||||||
items = ["securityname", "securitylevel", "authpassphrase",
|
|
||||||
"privpassphrase", "authprotocol", "privprotocol",
|
|
||||||
"contextname"]
|
|
||||||
for key, item in snmp.items():
|
|
||||||
if key in items:
|
|
||||||
self.interface["details"][key] = str(item)
|
|
||||||
else:
|
|
||||||
e = "Unsupported SNMP version."
|
|
||||||
raise InterfaceConfigError(e)
|
|
||||||
else:
|
|
||||||
e = "Interface type SNMP but no parameters provided."
|
|
||||||
raise InterfaceConfigError(e)
|
|
||||||
else:
|
|
||||||
e = "Interface type is not SNMP, unable to set SNMP details"
|
|
||||||
raise InterfaceConfigError(e)
|
|
||||||
|
|
||||||
def set_default(self):
|
|
||||||
""" Set default config to SNMPv2, port 161 and community macro. """
|
|
||||||
self.interface = self.skelet
|
|
||||||
self.interface["type"] = "2"
|
|
||||||
self.interface["port"] = "161"
|
|
||||||
self.interface["details"] = {"version": "2",
|
|
||||||
"community": "{$SNMP_COMMUNITY}",
|
|
||||||
"bulk": "1"}
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
parser = argparse.ArgumentParser(
|
parser = argparse.ArgumentParser(
|
||||||
description='A script to sync Zabbix with Netbox device data.'
|
description='A script to sync Zabbix with Netbox device data.'
|
||||||
|
@ -1,2 +1,2 @@
|
|||||||
pynetbox
|
pynetbox
|
||||||
pyzabbix
|
zabbix_utils
|
||||||
|
Loading…
Reference in New Issue
Block a user