import device-roles if exists

nb-dt-import.py

@@ -21,7 +21,12 @@ def main():
  settings.handle.log(f'{len(vendors)} Vendors Found')
  device_types = settings.dtl_repo.parse_files(files, slugs=args.slugs)
  settings.handle.log(f'{len(device_types)} Device-Types Found')
+ files = settings.dtl_repo.get_device_roles(
+ f'{settings.dtl_repo.repo_path}/device-roles/')
+ device_roles = settings.dtl_repo.parse_device_roles_files(files)
+ settings.handle.log(f'{len(device_roles)} Device Roles Found')
  netbox.create_manufacturers(vendors)
+ netbox.create_device_roles(device_roles)
  netbox.create_device_types(device_types)
 
  if netbox.modules:
@@ -43,6 +48,8 @@ def main():
  f'{netbox.counter["updated"]} interfaces/ports updated')
  settings.handle.log(
  f'{netbox.counter["manufacturer"]} manufacturers created')
+ settings.handle.log(
+ f'{netbox.counter["device_role"]} device roles created')
  if settings.NETBOX_FEATURES['modules']:
  settings.handle.log(
  f'{netbox.counter["module_added"]} modules created')

netbox_api.py

@@ -14,6 +14,7 @@ def __init__(self, settings):
  added=0,
  updated=0,
  manufacturer=0,
+ device_role=0,
  module_added=0,
  module_port_added=0,
  images=0,
@@ -79,6 +80,31 @@ def create_manufacturers(self, vendors):
  self.handle.log("Error creating manufacturers")
  self.handle.verbose_log(f"Error during manufacturer creation. - {request_error.error}")
 
+ def get_device_roles(self):
+ return {str(item): item for item in self.netbox.dcim.device_roles.all()}
+
+ def create_device_roles(self, roles):
+ to_create = []
+ self.existing_device_roles = self.get_device_roles()
+ for role in roles:
+ try:
+ rolGet = self.existing_device_roles[role["name"]]
+ self.handle.verbose_log(f'Device Roles Exists: {rolGet.name} - {rolGet.id}')
+ except KeyError:
+ to_create.append(role)
+ self.handle.verbose_log(f"Device Role queued for addition: {role['name']}")
+
+ if to_create:
+ try:
+ created_device_roles = self.netbox.dcim.device_roles.create(to_create)
+ for role in created_device_roles:
+ self.handle.verbose_log(f'Device Role Created: {role.name} - '
+ + f'{role.id}')
+ self.counter.update({'device_role': 1})
+ except pynetbox.RequestError as request_error:
+ self.handle.log("Error creating device role")
+ self.handle.verbose_log(f"Error during device role creation. - {request_error.error}")
+
  def create_device_types(self, device_types_to_add):
  for device_type in device_types_to_add:
 

repo.py

@@ -82,6 +82,24 @@ def get_devices(self, base_path, vendors: list = None):
  files.extend(glob(base_path + folder + f'/*.{extension}'))
  return files, discovered_vendors
 
+ def get_device_roles(self, base_path):
+ files = []
+ for extension in self.yaml_extensions:
+ files.extend(glob(base_path + f'/*.{extension}'))
+ return files
+
+ def parse_device_roles_files(self, files):
+ deviceRoles = []
+ for file in files:
+ with open(file, 'r') as stream:
+ try:
+ data = yaml.safe_load(stream)
+ except yaml.YAMLError as excep:
+ self.handle.verbose_log(excep)
+ continue
+ deviceRoles += data
+ return deviceRoles
+
  def parse_files(self, files: list, slugs: list = None):
  deviceTypes = []
  for file in files: