mirror of
https://github.com/TheNetworkGuy/netbox-zabbix-sync.git
synced 2025-07-14 01:41:25 -06:00
41 lines
1.5 KiB
Python
41 lines
1.5 KiB
Python
"""Module that hosts all functions for virtual machine processing"""
|
|
|
|
from modules.device import PhysicalDevice
|
|
from modules.hostgroups import Hostgroup
|
|
from modules.exceptions import TemplateError
|
|
try:
|
|
from config import (
|
|
traverse_site_groups,
|
|
traverse_regions,
|
|
template_cf
|
|
)
|
|
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 VirtualMachine(PhysicalDevice):
|
|
"""Model for virtual machines"""
|
|
def set_hostgroup(self, hg_format, nb_site_groups, nb_regions):
|
|
"""Set the hostgroup for this device"""
|
|
# Create new Hostgroup instance
|
|
hg = Hostgroup("vm", self.nb, self.nb_api_version,
|
|
traverse_site_groups, traverse_regions,
|
|
nb_site_groups, nb_regions)
|
|
# Generate hostgroup based on hostgroup format
|
|
self.hostgroup = hg.generate(hg_format)
|
|
|
|
def set_vm_template(self):
|
|
""" Set Template for VMs. Overwrites default class
|
|
to skip a lookup of custom fields."""
|
|
self.zbx_template_names = None
|
|
# Gather templates ONLY from the device specific context
|
|
try:
|
|
self.zbx_template_names = self.get_templates_context()
|
|
except TemplateError as e:
|
|
self.logger.warning(e)
|
|
return True
|
|
|
|
def set_template(self, **kwargs):
|
|
"""Simple wrapper fur underlying functions"""
|
|
self.set_vm_template() |