netbox_zabbix_sync.py

#!/usr/bin/python3
"""Sync Netbox devices to Zabbix for monitoring.



"""

# System imports
import logging
import pprint

# External imports
import pynetbox
import pyzabbix

# Local imports
import exceptions
import utils

# Set template and device Netbox "custom field" names
CF = {
    'TEMPLATE' : "zabbix_template",
    'HOSTID'   : "zabbix_hostid",
    'SNMP_VER' : "snmp_version",
}

# Netbox to Zabbix device state convertion
zabbix_device_removal = ["Decommissioning", "Inventory"]
zabbix_device_disable = ["Offline", "Planned", "Staged", "Failed"]

logger = logging.getLogger(__name__)

# Main code starts here.

class NetworkDevice:
    """Class representing a Network device.
    INPUT: (
        Netbox device class,
        ZabbixAPI class,
        journal flag,
        Netbox journal class,
    )
    """

    _host_translations = str.maketrans({
        '>': '_',
    })

    def __init__(self, nb_device, zabbix, nb_journal_class, journal=None):
        '''NetworkDevice Constructor.
        '''
        self.nb = nb_device
        self.id = nb_device.id
        self.name = NetworkDevice._clean_hostname(nb_device)
        self.status = nb_device.status.label
        self.zabbix = zabbix
        self.zabbix_id = None
        self.tenant = nb_device.tenant
        self.snmp_version = "2c" # Can be overridden by custom field.
        self.template_id = None
        self.hostgroup = None # Should no longer be needed
        self.hostgroups = []
        self.zbxproxy = "0"
        self.zabbix_state = 0
        self.hg_format = [ # Should no longer be needed
            self.nb.site.name,
            self.nb.device_type.manufacturer.name,
            self.nb.role.name
        ]
        self.group_id = None
        self.journal = journal
        self.nb_journals = nb_journal_class
        self._set_basics()
        self.set_hostgroup()
        self.set_host_groups()

        logger.info(f"Host groups: {pprint.pformat(self.hostgroups)}")

    @staticmethod
    def _clean_hostname(nb_device) -> str:
        '''Returns a cleaned name that Zabbix will accept.

        Per:
        https://www.zabbix.com/documentation/current/en/manual/config/hosts/host

        Alphanumerics, spaces, dots, dashes and underscores are allowed
        '''
        # The name may not always be set.
        if nb_device.name is None:
            return str(nb_device).translate(NetworkDevice._host_translations)
        else:
            return nb_device.name.translate(NetworkDevice._host_translations)

    def _set_basics(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.warning(e)
            raise exceptions.SyncInventoryError(e)

        # Check if device_type has custom field
        device_type_cf = self.nb.device_type.custom_fields
        if CF['TEMPLATE'] in device_type_cf:
            self.template_name = device_type_cf[CF['TEMPLATE']]
        else:
            e = (f"Custom field {CF['TEMPLATE']} not "
                 f"found for {self.nb.device_type.name}.")
            logger.warning(e)
            raise exceptions.SyncInventoryError(e)

        # Check if device has custom field
        if CF['HOSTID'] in self.nb.custom_fields:
            self.zabbix_id = self.nb.custom_fields[CF['HOSTID']]
        else:
            e = f"Custom field {CF['TEMPLATE']} not found for {self.name}."
            logger.warning(e)
            raise exceptions.SyncInventoryError(e)

    def set_hostgroup(self):
        """Sets hostgroup to a string with hg_format parameters.

        Depreciated
        """
        self.hostgroup = "/".join(self.hg_format)

    def set_host_groups(self):
        '''
        Sets the various hostgroups desired

        These will be of the format:
        Location - [Location Name]
        Rack - [Rack Name]
        Site - [Site Name]
        Role - [Device Role]
        Tenant - [Tenant Name]

        '''
        groups = {
            'Location': 'location.name',
            'Rack': 'rack.name',
            'Site': 'site.name',
            'Role': 'device_role.name',
            'Tenant': 'tenant.name',
        }
        for name, loc in groups.items():
            value = utils.rgetattr(self.nb, loc, None)
            if value:
                self.hostgroups.append(f"{name} - {value}")

        # Handle custom tags
        for curr_tag in self.nb.tags:
            if curr_tag['name'][:12] == 'ZabbixGroup#':
                self.hostgroups.append(curr_tag['name'][12:])



    def create_zabbix_hostgroups(self, zabbix_groups_map):
        """Creates Zabbix host group based on hostgroup format.
        """
        group_data = []
        for curr_hostgroup in self.hostgroups:
            try:
                # test if the group exists.
                if curr_hostgroup in zabbix_groups_map:
                    # The group already exists
                    continue
                groupid = self.zabbix.hostgroup.create(name=curr_hostgroup)
                logger.info(f"Added hostgroup '{self.hostgroup}'.")
                group_data.append({
                    'groupid': groupid["groupids"][0],
                    'name': self.hostgroup
                })
            except pyzabbix.ZabbixAPIException as exc:
                exc_msg = f"Couldn't add hostgroup {curr_hostgroup}, Zabbix returned {str(exc)}."
                logger.error(exc_msg)
                raise exceptions.SyncExternalError(exc_msg) from exc

        return group_data

    def is_cluster(self):
        """Checks if device is part of cluster.
        """
        if self.nb.virtual_chassis:
            return True
        else:
            return False

    def get_cluster_master(self):
        """
        Returns chassis master ID.
        """
        if not self.is_cluster():
            e = f"Unable to proces {self.name} for cluster calculation: not part of a cluster."
            logger.warning(e)
            raise exceptions.SyncInventoryError(e)
        else:
            return self.nb.virtual_chassis.master.id

    def promote_master_device(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.get_cluster_master()
        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
        else:
            logger.debug(f"Device {self.name} is non-primary cluster member.")
            return False

    def get_zabbix_template(self, templates) -> bool:
        """Get Zabbix template ID.
        INPUT: list of templates
        OUTPUT: True
        """
        if not self.template_name:
            e = (
                f"Device template '{self.nb.device_type.display}' "
                "has no Zabbix template defined."
            )
            logger.error(e)
            raise exceptions.SyncInventoryError()
        for template in templates:
            if template['name'] == self.template_name:
                self.template_id = template['templateid']
                e = (
                    f"Found template ID {str(template['templateid'])} for "
                    f"host {self.name}.")
                logger.debug(e)
                return True
        # No match was found.
        err_msg = (
            f"Unable to find template {self.template_name} "
            f"for host {self.name} in Zabbix."
        )
        logger.warning(err_msg)
        raise exceptions.SyncInventoryError(err_msg)

    def get_zabbix_group(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
        # No match found
        err_msg = (
            f"Unable to find group '{self.hostgroup}' "
            f"for host {self.name} in Zabbix."
        )
        logger.warning(err_msg)
        raise exceptions.SyncInventoryError(err_msg)

    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[CF['HOSTID']] = 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 pyzabbix.ZabbixAPIException as exc:
                err_msg = f"Zabbix returned the following error: {str(exc)}."
                logger.error(err_msg)
                raise exceptions.SyncExternalError(err_msg) from exc

    def _zabbix_hostname_exists(self):
        """
        Checks if hostname exists in Zabbix.
        """
        host = self.zabbix.host.get(filter={'name': self.name}, output=[])
        if host:
            return True
        else:
            return False

    def set_interface_details(self):
        """Set appropriate interface details.
        """
        try:
            interface = ZabbixInterface(self.nb, self.ip)
            return [interface.interface]
        except exceptions.InterfaceConfigError as exc:
            exc = f"{self.name}: {exc}"
            logger.warning(exc)
            raise exceptions.SyncInventoryError(exc) from None

    def set_proxy(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["host"] == proxy:
                        self.zbxproxy = px["proxyid"]
                        logger.debug(f"Found proxy {proxy}"
                                     f" for {self.name}.")
                        return True
                # No match found
                err_msg = f"{self.name}: Defined proxy {proxy} not found."
                logger.warning(err_msg)
                return False

    def create_in_zabbix(self, groups, zabbix_groups_map, templates, proxys,
                       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._zabbix_hostname_exists():
            # Get group IDs for host
            n_host_group_ids = []
            for curr_group in self.hostgroups:
                try:
                    n_host_group_ids.append(zabbix_groups_map[curr_group]['groupid'])
                except KeyError:
                    logger.error(f"Missing 'groupid' on: {curr_group}")
                    raise
            n_host_group_ids = sorted(map(lambda x: int(x), n_host_group_ids))
            groups = list(map(lambda x: {'groupid': x}, n_host_group_ids))

            # Set template IDs
            self.get_zabbix_template(templates)

            # Set interface, group and template configuration
            interfaces = self.set_interface_details()
            templates = [{"templateid": self.template_id}]

            # Set Zabbix proxy if defined
            self.set_proxy(proxys)

            # Add host to Zabbix
            try:
                host = self.zabbix.host.create(
                    host=self.name,
                    status=self.zabbix_state,
                    interfaces=interfaces,
                    groups=groups,
                    templates=templates,
                    proxy_hostid=self.zbxproxy,
                    description=description,
                )
                self.zabbix_id = host["hostids"][0]
            except pyzabbix.ZabbixAPIException as exc:
                err_msg = f"Couldn't add {self.name}, Zabbix returned {str(exc)}."
                logger.error(err_msg)
                raise exceptions.SyncExternalError(err_msg) from exc
            # Set Netbox custom field to hostID value.
            self.nb.custom_fields[CF['HOSTID']] = 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:
            logger.warning(f"Unable to add {self.name} to Zabbix: host already present.")

    def create_zabbix_hostgroup(self):
        """
        Creates Zabbix host group based on hostgroup format.
        """
        try:
            groupid = self.zabbix.hostgroup.create(name=self.hostgroup)
            logger.info(f"Added hostgroup '{self.hostgroup}'.")
            data = {'groupid': groupid["groupids"][0], 'name': self.hostgroup}
            return data
        except pyzabbix.ZabbixAPIException as exc:
            err_msg = f"Couldn't add hostgroup, Zabbix returned {str(exc)}."
            logger.error(err_msg)
            raise exceptions.SyncExternalError(err_msg) from exc

    def update_zabbix_host(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 pyzabbix.ZabbixAPIException as exc:
            err_msg = f"Zabbix returned the following error: {str(exc)}."
            logger.error(err_msg)
            raise exceptions.SyncExternalError(err_msg) from exc
        logger.info(f"Updated host {self.name} with data {kwargs}.")
        self.create_journal_entry("info", "Updated host in Zabbix with latest NB data.")

    def consistency_check(self, zabbix_groups_map, templates, proxys, proxy_power):
        """
        Checks if Zabbix object is still valid with Netbox parameters.
        """
        self.get_zabbix_template(templates)
        self.set_proxy(proxys)
        host = self.zabbix.host.get(
            filter={
                'hostid': self.zabbix_id,
            },
            selectInterfaces=[
                'type',
                'ip',
                'port',
                'details',
                'interfaceid',
            ],
            selectHostGroups=["groupid"],
            selectParentTemplates=["templateid"],
        )
        if len(host) > 1:
            err_msg = (
                f"Got {len(host)} results for Zabbix hosts "
                f"with ID {self.zabbix_id} - hostname {self.name}."
            )
            logger.error(err_msg)
            raise exceptions.SyncInventoryError(err_msg)
        elif 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 exceptions.SyncInventoryError(e)
        else:
            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.update_zabbix_host(host=self.name)

        for template in host["parentTemplates"]:
            if template["templateid"] == self.template_id:
                logger.debug(f"Device {self.name}: template in-sync.")
                break
        else:
            logger.warning(f"Device {self.name}: template OUT of sync.")
            self.update_zabbix_host(templates=self.template_id)

        # Sync the hostgroups
        n_host_group_ids = []
        for curr_group in self.hostgroups:
            n_host_group_ids.append(zabbix_groups_map[curr_group]['groupid'])
        n_host_group_ids = sorted(map(lambda x: int(x), n_host_group_ids))
        z_host_group_ids = sorted(map(lambda x: int(x['groupid']), host['hostgroups']))

        logger.debug(f"Groups z: {z_host_group_ids} - n: {n_host_group_ids}")

        if z_host_group_ids != n_host_group_ids:
            group_list = list(map(lambda x: {'groupid': x}, n_host_group_ids))
            logger.debug(f"Updating host '{self.name}' with groups {group_list}")
            self.update_zabbix_host(groups=group_list)
        else:
            logger.debug(f"Device {self.name}: hostgroups in-sync.")

        # Update host status
        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.update_zabbix_host(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 host["proxy_hostid"] == 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.")
                self.update_zabbix_host(proxy_hostid=self.zbxproxy)
        else:
            if not host["proxyid"] == "0":
                if proxy_power:
                    # If the -p flag has been issued,
                    # delete the proxy link in Zabbix
                    self.update_zabbix_host(proxy_hostid=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.")
        # If only 1 interface has been found
        if len(host['interfaces']) == 1:
            updates = {}
            # Go through each key / item and check if it matches Zabbix
            for key, item in self.set_interface_details()[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 exceptions.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 pyzabbix.ZabbixAPIException as exc:
                    err_msg = f"Zabbix returned the following error: {str(exc)}."
                    logger.error(err_msg)
                    raise exceptions.SyncExternalError(err_msg) from exc
            else:
                # If no updates are found, Zabbix interface is in-sync
                debug_msg = f"Device {self.name}: interface in-sync."
                logger.debug(debug_msg)
        else:
            err_msg = (
                f"Device {self.name} has unsupported interface configuration."
                f" Host has total of {len(host['interfaces'])} interfaces. "
                "Manual interfention required."
            )
            logger.error(err_msg)
            raise exceptions.SyncInventoryError(err_msg)

    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 pynetbox.RequestError as exc:
                logger.warning(
                    "Unable to create journal entry for "
                    f"{self.name}: NB returned {exc}"
                )


class ZabbixInterface:
    """Class that represents a Zabbix interface"""

    def __init__(self, device, ip: str):
        """Zabbix interface constructor.

        Args:
            device (_type_): Netbox device object.
            ip (str): _description_
        """
        self.device = device
        self.context = device.config_context
        self.ip = ip
        self.skelet = {"main": "1", "useip": "1", "dns": "", "ip": self.ip}
        self.interface = self.skelet

        # Check if Netbox has device context.
        # If not fall back to default config.
        if self.get_context():
            # If device is SNMP type, add aditional information.
            if self.interface["type"] == 2:
                self.set_interface_snmp()
        else:
            self.set_snmp_default()

    def get_context(self):
        '''Check if Netbox custom context has been defined.
        '''
        if "zabbix" not in self.context:
            return False
        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
        else:
            return False

    def set_interface_snmp(self):
        '''Set SNMP parameters from host config context.
        '''
        if self.interface["type"] != 2:
            raise exceptions.InterfaceConfigError(
                "Interface type is not SNMP, unable to set SNMP details"
            )
        if "snmp" not in self.context["zabbix"]:
            # Checks if SNMP settings are defined in Netbox
            raise exceptions.InterfaceConfigError(
                "Interface type SNMP but no parameters provided."
            )

        snmp = self.context["zabbix"]["snmp"]
        self.interface["details"] = {}
        # Checks if bulk config has been defined, default to 1.
        self.interface["details"]["bulk"] = snmp.get('bulk', '1')

        # SNMP Version config is required in Netbox config context
        self.interface["details"]["version"] = str(snmp.get("version", None))
        if self.interface["details"]["version"] in ['1','2']:
            # If version 1 or 2 is used, get community string
            if "community" in snmp:
                community = snmp["community"]
                self.interface["details"]["community"] = str(community)
            else:
                logger.debug("No SNMP community string "
                    "defined in custom context, using default")
        elif self.interface["details"]["version"] == '3':
            # If version 3 has been used, get all
            # SNMPv3 Netbox related configs
            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:
            raise exceptions.InterfaceConfigError(
                'Unsupported SNMP version '
                f'{self.interface["details"]["version"]}'
            )

    def set_snmp_default(self):
        '''Set SNMP defaults based on the device type.
        '''
        self.interface = self.skelet
        self.interface["type"] = "2"
        self.interface["port"] = "161"
        self.interface["details"] = {
            "version": "2",
            "community": "{$SNMP_COMMUNITY}",
            "bulk": "1"
        }
        # Pull in data from the device_type
        if self.device.device_type.custom_fields['snmp_version'] is not None:
            self.interface["details"]["version"] = self.device.device_type.\
                custom_fields['snmp_version']


def main():
    """Run the sync process.
    """
    # Get CLI args
    arguments = utils.parse_args()

    utils.setup_logging(logger, arguments,'sync')

    config = utils.fetch_sync_config()

    # Connect to netbox & zabbix
    zabbix = utils.connect_zabbix(config)
    netbox = utils.connect_netbox(config)

    # Fetch zabbix data
    zabbix_templates = zabbix.template.get(output=['name'])
    zabbix_proxies   = zabbix.proxy.get(output=['name','proxyid'])

    # Fetch netbox data
    netbox_devices  = netbox.dcim.devices.all()
    netbox_journals = netbox.extras.journal_entries

    # Go through all Netbox devices
    for nb_device in netbox_devices:
        try:
            # Ignore rules
            if nb_device.role.slug in config['NETBOX_ROLE_IGNORE']:
                logger.debug(f"Skipping host: {nb_device.name}")
                continue

            # Lets start dealing with the device
            device = NetworkDevice(
                nb_device,
                zabbix,
                netbox_journals,
                arguments.journal
            )
            # Checks if device is part of cluster.
            # Requires the cluster argument.
            if device.is_cluster() and arguments.cluster:
                # Check if device is master or slave
                if device.promote_master_device():
                    err_msg = f"Device {device.name} is part of " +\
                        "cluster and primary."
                    logger.info(err_msg)
                else:
                    # Device is secondary in cluster.
                    # Don't continue with this device.
                    err_msg = f"Device {device.name} is part of cluster " \
                              f"but not primary. Skipping this host..."
                    logger.info(err_msg)
                    continue

            # Checks if device is in cleanup state
            if device.status in zabbix_device_removal:
                if device.zabbix_id :
                    # Delete device from Zabbix
                    # and remove hostID from Netbox.
                    device.cleanup()
                    logger.info(f"Cleaned up host {device.name}.")

                else:
                    # Device has been added to Netbox
                    # but is not in Activate state
                    logger.info(
                        f"Skipping host {device.name} since its "
                        f"not in the active state."
                    )
                continue
            elif device.status in zabbix_device_disable:
                device.zabbix_state = 1

            # Setup the host groups, wasteful for the time being
            # to handle the creation of new groups, might just be best to
            # query the groups I want for this host in the create hostgroups function
            zabbix_groups    = zabbix.hostgroup.get(output=['name'])
            zabbix_groups_map = {v['name']:v for v in zabbix_groups}
            host_group_data = device.create_zabbix_hostgroups(zabbix_groups_map)

            zabbix_groups    = zabbix.hostgroup.get(output=['groupid', 'name'])
            zabbix_groups_map = {v['name']:v for v in zabbix_groups}

            if device.zabbix_id: # Update Zabbix
                device.consistency_check(
                    zabbix_groups_map,
                    zabbix_templates,
                    zabbix_proxies,
                    arguments.proxy_power,
                )
            else: # Add to Zabbix
                device.create_in_zabbix(
                    zabbix_groups,
                    zabbix_groups_map,
                    zabbix_templates,
                    zabbix_proxies
                )
        except exceptions.SyncError:
            pass
    logger.info("Done")

if __name__ == "__main__":
    main()