Change logic to prevent sys.exit from stopping script processing
This commit is contained in:
parent
5769684c98
commit
5292afaae0
28 changed files with 544 additions and 572 deletions
|
@ -5,8 +5,7 @@ from startup_script_utils import load_yaml, set_permissions
|
||||||
from users.models import Token
|
from users.models import Token
|
||||||
|
|
||||||
users = load_yaml('/opt/netbox/initializers/users.yml')
|
users = load_yaml('/opt/netbox/initializers/users.yml')
|
||||||
if users is None:
|
if not users is None:
|
||||||
sys.exit()
|
|
||||||
|
|
||||||
for username, user_details in users.items():
|
for username, user_details in users.items():
|
||||||
if not User.objects.filter(username=username):
|
if not User.objects.filter(username=username):
|
||||||
|
|
|
@ -4,8 +4,7 @@ from django.contrib.auth.models import Group, User
|
||||||
from startup_script_utils import load_yaml, set_permissions
|
from startup_script_utils import load_yaml, set_permissions
|
||||||
|
|
||||||
groups = load_yaml('/opt/netbox/initializers/groups.yml')
|
groups = load_yaml('/opt/netbox/initializers/groups.yml')
|
||||||
if groups is None:
|
if not groups is None:
|
||||||
sys.exit()
|
|
||||||
|
|
||||||
for groupname, group_details in groups.items():
|
for groupname, group_details in groups.items():
|
||||||
group, created = Group.objects.get_or_create(name=groupname)
|
group, created = Group.objects.get_or_create(name=groupname)
|
||||||
|
|
|
@ -14,8 +14,7 @@ def get_class_for_class_path(class_path):
|
||||||
|
|
||||||
customfields = load_yaml('/opt/netbox/initializers/custom_fields.yml')
|
customfields = load_yaml('/opt/netbox/initializers/custom_fields.yml')
|
||||||
|
|
||||||
if customfields is None:
|
if not customfields is None:
|
||||||
sys.exit()
|
|
||||||
|
|
||||||
for cf_name, cf_details in customfields.items():
|
for cf_name, cf_details in customfields.items():
|
||||||
custom_field, created = CustomField.objects.get_or_create(name = cf_name)
|
custom_field, created = CustomField.objects.get_or_create(name = cf_name)
|
||||||
|
|
|
@ -4,8 +4,7 @@ import sys
|
||||||
|
|
||||||
regions = load_yaml('/opt/netbox/initializers/regions.yml')
|
regions = load_yaml('/opt/netbox/initializers/regions.yml')
|
||||||
|
|
||||||
if regions is None:
|
if not regions is None:
|
||||||
sys.exit()
|
|
||||||
|
|
||||||
optional_assocs = {
|
optional_assocs = {
|
||||||
'parent': (Region, 'name')
|
'parent': (Region, 'name')
|
||||||
|
|
|
@ -6,8 +6,7 @@ import sys
|
||||||
|
|
||||||
sites = load_yaml('/opt/netbox/initializers/sites.yml')
|
sites = load_yaml('/opt/netbox/initializers/sites.yml')
|
||||||
|
|
||||||
if sites is None:
|
if not sites is None:
|
||||||
sys.exit()
|
|
||||||
|
|
||||||
optional_assocs = {
|
optional_assocs = {
|
||||||
'region': (Region, 'name'),
|
'region': (Region, 'name'),
|
||||||
|
|
|
@ -4,8 +4,7 @@ import sys
|
||||||
|
|
||||||
manufacturers = load_yaml('/opt/netbox/initializers/manufacturers.yml')
|
manufacturers = load_yaml('/opt/netbox/initializers/manufacturers.yml')
|
||||||
|
|
||||||
if manufacturers is None:
|
if not manufacturers is None:
|
||||||
sys.exit()
|
|
||||||
|
|
||||||
for params in manufacturers:
|
for params in manufacturers:
|
||||||
manufacturer, created = Manufacturer.objects.get_or_create(**params)
|
manufacturer, created = Manufacturer.objects.get_or_create(**params)
|
||||||
|
|
|
@ -6,8 +6,7 @@ import sys
|
||||||
|
|
||||||
device_types = load_yaml('/opt/netbox/initializers/device_types.yml')
|
device_types = load_yaml('/opt/netbox/initializers/device_types.yml')
|
||||||
|
|
||||||
if device_types is None:
|
if not device_types is None:
|
||||||
sys.exit()
|
|
||||||
|
|
||||||
required_assocs = {
|
required_assocs = {
|
||||||
'manufacturer': (Manufacturer, 'name')
|
'manufacturer': (Manufacturer, 'name')
|
||||||
|
|
|
@ -6,8 +6,7 @@ import sys
|
||||||
|
|
||||||
rack_roles = load_yaml('/opt/netbox/initializers/rack_roles.yml')
|
rack_roles = load_yaml('/opt/netbox/initializers/rack_roles.yml')
|
||||||
|
|
||||||
if rack_roles is None:
|
if not rack_roles is None:
|
||||||
sys.exit()
|
|
||||||
|
|
||||||
for params in rack_roles:
|
for params in rack_roles:
|
||||||
if 'color' in params:
|
if 'color' in params:
|
||||||
|
|
|
@ -4,8 +4,7 @@ import sys
|
||||||
|
|
||||||
rack_groups = load_yaml('/opt/netbox/initializers/rack_groups.yml')
|
rack_groups = load_yaml('/opt/netbox/initializers/rack_groups.yml')
|
||||||
|
|
||||||
if rack_groups is None:
|
if not rack_groups is None:
|
||||||
sys.exit()
|
|
||||||
|
|
||||||
required_assocs = {
|
required_assocs = {
|
||||||
'site': (Site, 'name')
|
'site': (Site, 'name')
|
||||||
|
|
|
@ -6,8 +6,7 @@ import sys
|
||||||
|
|
||||||
racks = load_yaml('/opt/netbox/initializers/racks.yml')
|
racks = load_yaml('/opt/netbox/initializers/racks.yml')
|
||||||
|
|
||||||
if racks is None:
|
if not racks is None:
|
||||||
sys.exit()
|
|
||||||
|
|
||||||
required_assocs = {
|
required_assocs = {
|
||||||
'site': (Site, 'name')
|
'site': (Site, 'name')
|
||||||
|
|
|
@ -5,8 +5,7 @@ import sys
|
||||||
|
|
||||||
device_roles = load_yaml('/opt/netbox/initializers/device_roles.yml')
|
device_roles = load_yaml('/opt/netbox/initializers/device_roles.yml')
|
||||||
|
|
||||||
if device_roles is None:
|
if not device_roles is None:
|
||||||
sys.exit()
|
|
||||||
|
|
||||||
for params in device_roles:
|
for params in device_roles:
|
||||||
|
|
||||||
|
|
|
@ -4,8 +4,7 @@ import sys
|
||||||
|
|
||||||
platforms = load_yaml('/opt/netbox/initializers/platforms.yml')
|
platforms = load_yaml('/opt/netbox/initializers/platforms.yml')
|
||||||
|
|
||||||
if platforms is None:
|
if not platforms is None:
|
||||||
sys.exit()
|
|
||||||
|
|
||||||
optional_assocs = {
|
optional_assocs = {
|
||||||
'manufacturer': (Manufacturer, 'name'),
|
'manufacturer': (Manufacturer, 'name'),
|
||||||
|
|
|
@ -4,8 +4,7 @@ import sys
|
||||||
|
|
||||||
tenant_groups = load_yaml('/opt/netbox/initializers/tenant_groups.yml')
|
tenant_groups = load_yaml('/opt/netbox/initializers/tenant_groups.yml')
|
||||||
|
|
||||||
if tenant_groups is None:
|
if not tenant_groups is None:
|
||||||
sys.exit()
|
|
||||||
|
|
||||||
for params in tenant_groups:
|
for params in tenant_groups:
|
||||||
tenant_group, created = TenantGroup.objects.get_or_create(**params)
|
tenant_group, created = TenantGroup.objects.get_or_create(**params)
|
||||||
|
|
|
@ -5,8 +5,7 @@ import sys
|
||||||
|
|
||||||
tenants = load_yaml('/opt/netbox/initializers/tenants.yml')
|
tenants = load_yaml('/opt/netbox/initializers/tenants.yml')
|
||||||
|
|
||||||
if tenants is None:
|
if not tenants is None:
|
||||||
sys.exit()
|
|
||||||
|
|
||||||
optional_assocs = {
|
optional_assocs = {
|
||||||
'group': (TenantGroup, 'name')
|
'group': (TenantGroup, 'name')
|
||||||
|
|
|
@ -8,8 +8,7 @@ import sys
|
||||||
|
|
||||||
devices = load_yaml('/opt/netbox/initializers/devices.yml')
|
devices = load_yaml('/opt/netbox/initializers/devices.yml')
|
||||||
|
|
||||||
if devices is None:
|
if not devices is None:
|
||||||
sys.exit()
|
|
||||||
|
|
||||||
required_assocs = {
|
required_assocs = {
|
||||||
'device_role': (DeviceRole, 'name'),
|
'device_role': (DeviceRole, 'name'),
|
||||||
|
|
|
@ -4,8 +4,7 @@ import sys
|
||||||
|
|
||||||
cluster_types = load_yaml('/opt/netbox/initializers/cluster_types.yml')
|
cluster_types = load_yaml('/opt/netbox/initializers/cluster_types.yml')
|
||||||
|
|
||||||
if cluster_types is None:
|
if not cluster_types is None:
|
||||||
sys.exit()
|
|
||||||
|
|
||||||
for params in cluster_types:
|
for params in cluster_types:
|
||||||
cluster_type, created = ClusterType.objects.get_or_create(**params)
|
cluster_type, created = ClusterType.objects.get_or_create(**params)
|
||||||
|
|
|
@ -4,8 +4,7 @@ import sys
|
||||||
|
|
||||||
rirs = load_yaml('/opt/netbox/initializers/rirs.yml')
|
rirs = load_yaml('/opt/netbox/initializers/rirs.yml')
|
||||||
|
|
||||||
if rirs is None:
|
if not rirs is None:
|
||||||
sys.exit()
|
|
||||||
|
|
||||||
for params in rirs:
|
for params in rirs:
|
||||||
rir, created = RIR.objects.get_or_create(**params)
|
rir, created = RIR.objects.get_or_create(**params)
|
||||||
|
|
|
@ -8,8 +8,7 @@ import sys
|
||||||
|
|
||||||
aggregates = load_yaml('/opt/netbox/initializers/aggregates.yml')
|
aggregates = load_yaml('/opt/netbox/initializers/aggregates.yml')
|
||||||
|
|
||||||
if aggregates is None:
|
if not aggregates is None:
|
||||||
sys.exit()
|
|
||||||
|
|
||||||
required_assocs = {
|
required_assocs = {
|
||||||
'rir': (RIR, 'name')
|
'rir': (RIR, 'name')
|
||||||
|
|
|
@ -6,8 +6,7 @@ import sys
|
||||||
|
|
||||||
clusters = load_yaml('/opt/netbox/initializers/clusters.yml')
|
clusters = load_yaml('/opt/netbox/initializers/clusters.yml')
|
||||||
|
|
||||||
if clusters is None:
|
if not clusters is None:
|
||||||
sys.exit()
|
|
||||||
|
|
||||||
required_assocs = {
|
required_assocs = {
|
||||||
'type': (ClusterType, 'name')
|
'type': (ClusterType, 'name')
|
||||||
|
|
|
@ -8,8 +8,7 @@ import sys
|
||||||
|
|
||||||
vrfs = load_yaml('/opt/netbox/initializers/vrfs.yml')
|
vrfs = load_yaml('/opt/netbox/initializers/vrfs.yml')
|
||||||
|
|
||||||
if vrfs is None:
|
if not vrfs is None:
|
||||||
sys.exit()
|
|
||||||
|
|
||||||
optional_assocs = {
|
optional_assocs = {
|
||||||
'tenant': (Tenant, 'name')
|
'tenant': (Tenant, 'name')
|
||||||
|
|
|
@ -4,8 +4,7 @@ import sys
|
||||||
|
|
||||||
roles = load_yaml('/opt/netbox/initializers/prefix_vlan_roles.yml')
|
roles = load_yaml('/opt/netbox/initializers/prefix_vlan_roles.yml')
|
||||||
|
|
||||||
if roles is None:
|
if not roles is None:
|
||||||
sys.exit()
|
|
||||||
|
|
||||||
for params in roles:
|
for params in roles:
|
||||||
role, created = Role.objects.get_or_create(**params)
|
role, created = Role.objects.get_or_create(**params)
|
||||||
|
|
|
@ -6,8 +6,7 @@ import sys
|
||||||
|
|
||||||
vlan_groups = load_yaml('/opt/netbox/initializers/vlan_groups.yml')
|
vlan_groups = load_yaml('/opt/netbox/initializers/vlan_groups.yml')
|
||||||
|
|
||||||
if vlan_groups is None:
|
if not vlan_groups is None:
|
||||||
sys.exit()
|
|
||||||
|
|
||||||
optional_assocs = {
|
optional_assocs = {
|
||||||
'site': (Site, 'name')
|
'site': (Site, 'name')
|
||||||
|
|
|
@ -7,8 +7,7 @@ import sys
|
||||||
|
|
||||||
vlans = load_yaml('/opt/netbox/initializers/vlans.yml')
|
vlans = load_yaml('/opt/netbox/initializers/vlans.yml')
|
||||||
|
|
||||||
if vlans is None:
|
if not vlans is None:
|
||||||
sys.exit()
|
|
||||||
|
|
||||||
optional_assocs = {
|
optional_assocs = {
|
||||||
'site': (Site, 'name'),
|
'site': (Site, 'name'),
|
||||||
|
|
|
@ -8,8 +8,7 @@ import sys
|
||||||
|
|
||||||
prefixes = load_yaml('/opt/netbox/initializers/prefixes.yml')
|
prefixes = load_yaml('/opt/netbox/initializers/prefixes.yml')
|
||||||
|
|
||||||
if prefixes is None:
|
if not prefixes is None:
|
||||||
sys.exit()
|
|
||||||
|
|
||||||
optional_assocs = {
|
optional_assocs = {
|
||||||
'site': (Site, 'name'),
|
'site': (Site, 'name'),
|
||||||
|
|
|
@ -7,8 +7,7 @@ import sys
|
||||||
|
|
||||||
virtual_machines = load_yaml('/opt/netbox/initializers/virtual_machines.yml')
|
virtual_machines = load_yaml('/opt/netbox/initializers/virtual_machines.yml')
|
||||||
|
|
||||||
if virtual_machines is None:
|
if not virtual_machines is None:
|
||||||
sys.exit()
|
|
||||||
|
|
||||||
required_assocs = {
|
required_assocs = {
|
||||||
'cluster': (Cluster, 'name')
|
'cluster': (Cluster, 'name')
|
||||||
|
|
|
@ -6,8 +6,7 @@ import sys
|
||||||
|
|
||||||
interfaces = load_yaml('/opt/netbox/initializers/virtualization_interfaces.yml')
|
interfaces = load_yaml('/opt/netbox/initializers/virtualization_interfaces.yml')
|
||||||
|
|
||||||
if interfaces is None:
|
if not interfaces is None:
|
||||||
sys.exit()
|
|
||||||
|
|
||||||
required_assocs = {
|
required_assocs = {
|
||||||
'virtual_machine': (VirtualMachine, 'name')
|
'virtual_machine': (VirtualMachine, 'name')
|
||||||
|
|
|
@ -5,8 +5,7 @@ import sys
|
||||||
|
|
||||||
interfaces= load_yaml('/opt/netbox/initializers/dcim_interfaces.yml')
|
interfaces= load_yaml('/opt/netbox/initializers/dcim_interfaces.yml')
|
||||||
|
|
||||||
if interfaces is None:
|
if not interfaces is None:
|
||||||
sys.exit()
|
|
||||||
|
|
||||||
required_assocs = {
|
required_assocs = {
|
||||||
'device': (Device, 'name')
|
'device': (Device, 'name')
|
||||||
|
|
|
@ -10,8 +10,7 @@ import sys
|
||||||
|
|
||||||
ip_addresses = load_yaml('/opt/netbox/initializers/ip_addresses.yml')
|
ip_addresses = load_yaml('/opt/netbox/initializers/ip_addresses.yml')
|
||||||
|
|
||||||
if ip_addresses is None:
|
if not ip_addresses is None:
|
||||||
sys.exit()
|
|
||||||
|
|
||||||
optional_assocs = {
|
optional_assocs = {
|
||||||
'tenant': (Tenant, 'name'),
|
'tenant': (Tenant, 'name'),
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue