mirror of
https://github.com/TheNetworkGuy/netbox-zabbix-sync.git
synced 2025-07-13 23:27:32 -06:00
82 lines
3.9 KiB
Python
82 lines
3.9 KiB
Python
#!/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 pprint import pprint
|
|
from logging import getLogger
|
|
from zabbix_utils import APIRequestError
|
|
from modules.exceptions import (SyncInventoryError, TemplateError, SyncExternalError,
|
|
InterfaceConfigError, JournalError)
|
|
try:
|
|
from config import (
|
|
inventory_sync,
|
|
inventory_mode,
|
|
device_inventory_map,
|
|
vm_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 Inventory():
|
|
# pylint: disable=too-many-instance-attributes, too-many-arguments, too-many-positional-arguments
|
|
"""
|
|
Represents Network device.
|
|
INPUT: (NetBox device class, ZabbixAPI class, journal flag, NB journal class)
|
|
"""
|
|
|
|
# def __init__(self, nb, logger=None):
|
|
# self.nb = nb
|
|
|
|
def set_inventory(self, nbobject):
|
|
if hasattr(nbobject, 'device_type'):
|
|
inventory_map = device_inventory_map
|
|
else:
|
|
inventory_map = vm_inventory_map
|
|
""" Set host inventory """
|
|
# Set inventory mode. Default is disabled (see class init function).
|
|
if inventory_mode == "disabled":
|
|
if inventory_sync:
|
|
self.logger.error(f"Host {self.name}: Unable to map NetBox inventory to Zabbix. "
|
|
"Inventory sync is enabled in config but inventory mode is disabled.")
|
|
return True
|
|
if inventory_mode == "manual":
|
|
self.inventory_mode = 0
|
|
elif inventory_mode == "automatic":
|
|
self.inventory_mode = 1
|
|
else:
|
|
self.logger.error(f"Host {self.name}: Specified value for inventory mode in"
|
|
f" config is not valid. Got value {inventory_mode}")
|
|
return False
|
|
self.inventory = {}
|
|
if inventory_sync and self.inventory_mode in [0,1]:
|
|
self.logger.debug(f"Host {self.name}: Starting inventory mapper")
|
|
# 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 = nbobject
|
|
# ... 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"Host {self.name}: NetBox inventory lookup for "
|
|
f"'{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"Host {self.name}: Inventory lookup for '{nb_inv_field}'"
|
|
" returned an unexpected type: it will be skipped.")
|
|
self.logger.debug(f"Host {self.name}: Inventory mapping complete. "
|
|
f"Mapped {len(list(filter(None, self.inventory.values())))} field(s)")
|
|
# return True
|