Change logic to prevent sys.exit from stopping script processing

This commit is contained in:
Hank Preston 2020-05-06 13:25:09 -04:00
parent 5769684c98
commit 5292afaae0
28 changed files with 544 additions and 572 deletions

View file

@ -5,8 +5,7 @@ from startup_script_utils import load_yaml, set_permissions
from users.models import Token
users = load_yaml('/opt/netbox/initializers/users.yml')
if users is None:
sys.exit()
if not users is None:
for username, user_details in users.items():
if not User.objects.filter(username=username):

View file

@ -4,8 +4,7 @@ from django.contrib.auth.models import Group, User
from startup_script_utils import load_yaml, set_permissions
groups = load_yaml('/opt/netbox/initializers/groups.yml')
if groups is None:
sys.exit()
if not groups is None:
for groupname, group_details in groups.items():
group, created = Group.objects.get_or_create(name=groupname)

View file

@ -14,8 +14,7 @@ def get_class_for_class_path(class_path):
customfields = load_yaml('/opt/netbox/initializers/custom_fields.yml')
if customfields is None:
sys.exit()
if not customfields is None:
for cf_name, cf_details in customfields.items():
custom_field, created = CustomField.objects.get_or_create(name = cf_name)

View file

@ -4,8 +4,7 @@ import sys
regions = load_yaml('/opt/netbox/initializers/regions.yml')
if regions is None:
sys.exit()
if not regions is None:
optional_assocs = {
'parent': (Region, 'name')

View file

@ -6,8 +6,7 @@ import sys
sites = load_yaml('/opt/netbox/initializers/sites.yml')
if sites is None:
sys.exit()
if not sites is None:
optional_assocs = {
'region': (Region, 'name'),

View file

@ -4,8 +4,7 @@ import sys
manufacturers = load_yaml('/opt/netbox/initializers/manufacturers.yml')
if manufacturers is None:
sys.exit()
if not manufacturers is None:
for params in manufacturers:
manufacturer, created = Manufacturer.objects.get_or_create(**params)

View file

@ -6,8 +6,7 @@ import sys
device_types = load_yaml('/opt/netbox/initializers/device_types.yml')
if device_types is None:
sys.exit()
if not device_types is None:
required_assocs = {
'manufacturer': (Manufacturer, 'name')

View file

@ -6,8 +6,7 @@ import sys
rack_roles = load_yaml('/opt/netbox/initializers/rack_roles.yml')
if rack_roles is None:
sys.exit()
if not rack_roles is None:
for params in rack_roles:
if 'color' in params:

View file

@ -4,8 +4,7 @@ import sys
rack_groups = load_yaml('/opt/netbox/initializers/rack_groups.yml')
if rack_groups is None:
sys.exit()
if not rack_groups is None:
required_assocs = {
'site': (Site, 'name')

View file

@ -6,8 +6,7 @@ import sys
racks = load_yaml('/opt/netbox/initializers/racks.yml')
if racks is None:
sys.exit()
if not racks is None:
required_assocs = {
'site': (Site, 'name')

View file

@ -5,8 +5,7 @@ import sys
device_roles = load_yaml('/opt/netbox/initializers/device_roles.yml')
if device_roles is None:
sys.exit()
if not device_roles is None:
for params in device_roles:

View file

@ -4,8 +4,7 @@ import sys
platforms = load_yaml('/opt/netbox/initializers/platforms.yml')
if platforms is None:
sys.exit()
if not platforms is None:
optional_assocs = {
'manufacturer': (Manufacturer, 'name'),

View file

@ -4,8 +4,7 @@ import sys
tenant_groups = load_yaml('/opt/netbox/initializers/tenant_groups.yml')
if tenant_groups is None:
sys.exit()
if not tenant_groups is None:
for params in tenant_groups:
tenant_group, created = TenantGroup.objects.get_or_create(**params)

View file

@ -5,8 +5,7 @@ import sys
tenants = load_yaml('/opt/netbox/initializers/tenants.yml')
if tenants is None:
sys.exit()
if not tenants is None:
optional_assocs = {
'group': (TenantGroup, 'name')

View file

@ -8,8 +8,7 @@ import sys
devices = load_yaml('/opt/netbox/initializers/devices.yml')
if devices is None:
sys.exit()
if not devices is None:
required_assocs = {
'device_role': (DeviceRole, 'name'),

View file

@ -4,8 +4,7 @@ import sys
cluster_types = load_yaml('/opt/netbox/initializers/cluster_types.yml')
if cluster_types is None:
sys.exit()
if not cluster_types is None:
for params in cluster_types:
cluster_type, created = ClusterType.objects.get_or_create(**params)

View file

@ -4,8 +4,7 @@ import sys
rirs = load_yaml('/opt/netbox/initializers/rirs.yml')
if rirs is None:
sys.exit()
if not rirs is None:
for params in rirs:
rir, created = RIR.objects.get_or_create(**params)

View file

@ -8,8 +8,7 @@ import sys
aggregates = load_yaml('/opt/netbox/initializers/aggregates.yml')
if aggregates is None:
sys.exit()
if not aggregates is None:
required_assocs = {
'rir': (RIR, 'name')

View file

@ -6,8 +6,7 @@ import sys
clusters = load_yaml('/opt/netbox/initializers/clusters.yml')
if clusters is None:
sys.exit()
if not clusters is None:
required_assocs = {
'type': (ClusterType, 'name')

View file

@ -8,8 +8,7 @@ import sys
vrfs = load_yaml('/opt/netbox/initializers/vrfs.yml')
if vrfs is None:
sys.exit()
if not vrfs is None:
optional_assocs = {
'tenant': (Tenant, 'name')

View file

@ -4,8 +4,7 @@ import sys
roles = load_yaml('/opt/netbox/initializers/prefix_vlan_roles.yml')
if roles is None:
sys.exit()
if not roles is None:
for params in roles:
role, created = Role.objects.get_or_create(**params)

View file

@ -6,8 +6,7 @@ import sys
vlan_groups = load_yaml('/opt/netbox/initializers/vlan_groups.yml')
if vlan_groups is None:
sys.exit()
if not vlan_groups is None:
optional_assocs = {
'site': (Site, 'name')

View file

@ -7,8 +7,7 @@ import sys
vlans = load_yaml('/opt/netbox/initializers/vlans.yml')
if vlans is None:
sys.exit()
if not vlans is None:
optional_assocs = {
'site': (Site, 'name'),

View file

@ -8,8 +8,7 @@ import sys
prefixes = load_yaml('/opt/netbox/initializers/prefixes.yml')
if prefixes is None:
sys.exit()
if not prefixes is None:
optional_assocs = {
'site': (Site, 'name'),

View file

@ -7,8 +7,7 @@ import sys
virtual_machines = load_yaml('/opt/netbox/initializers/virtual_machines.yml')
if virtual_machines is None:
sys.exit()
if not virtual_machines is None:
required_assocs = {
'cluster': (Cluster, 'name')

View file

@ -6,8 +6,7 @@ import sys
interfaces = load_yaml('/opt/netbox/initializers/virtualization_interfaces.yml')
if interfaces is None:
sys.exit()
if not interfaces is None:
required_assocs = {
'virtual_machine': (VirtualMachine, 'name')

View file

@ -5,8 +5,7 @@ import sys
interfaces= load_yaml('/opt/netbox/initializers/dcim_interfaces.yml')
if interfaces is None:
sys.exit()
if not interfaces is None:
required_assocs = {
'device': (Device, 'name')

View file

@ -10,8 +10,7 @@ import sys
ip_addresses = load_yaml('/opt/netbox/initializers/ip_addresses.yml')
if ip_addresses is None:
sys.exit()
if not ip_addresses is None:
optional_assocs = {
'tenant': (Tenant, 'name'),