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
|
||||
|
||||
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):
|
||||
|
|
|
@ -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)
|
||||
|
|
|
@ -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)
|
||||
|
|
|
@ -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')
|
||||
|
|
|
@ -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'),
|
||||
|
|
|
@ -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)
|
||||
|
|
|
@ -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')
|
||||
|
|
|
@ -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:
|
||||
|
|
|
@ -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')
|
||||
|
|
|
@ -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')
|
||||
|
|
|
@ -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:
|
||||
|
||||
|
|
|
@ -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'),
|
||||
|
|
|
@ -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)
|
||||
|
|
|
@ -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')
|
||||
|
|
|
@ -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'),
|
||||
|
|
|
@ -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)
|
||||
|
|
|
@ -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)
|
||||
|
|
|
@ -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')
|
||||
|
|
|
@ -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')
|
||||
|
|
|
@ -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')
|
||||
|
|
|
@ -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)
|
||||
|
|
|
@ -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')
|
||||
|
|
|
@ -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'),
|
||||
|
|
|
@ -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'),
|
||||
|
|
|
@ -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')
|
||||
|
|
|
@ -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')
|
||||
|
|
|
@ -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')
|
||||
|
|
|
@ -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'),
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue