Mascali/netbox-docker
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Ignore missing initializers

Browse source 
Previously, the startup_scripts would fail if an initializer file was
not present. Now they just ignore missing files.
This commit is contained in:
Christian Mäder 2018-12-19 14:25:58 +01:00
parent 03eb153da4
commit e46a7d2f7f
No known key found for this signature in database
GPG key ID: 92FFD0A711F196BB
12 changed files with 88 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

8
startup_scripts/020_custom_fields.py
View file

@ -2,6 +2,8 @@ from extras.constants import CF_TYPE_TEXT, CF_TYPE_INTEGER, CF_TYPE_BOOLEAN, CF_
from extras.models import CustomField, CustomFieldChoice
from ruamel.yaml import YAML
from pathlib import Path
import sys
text_to_fields = {
'boolean': CF_TYPE_BOOLEAN,
@ -21,7 +23,11 @@ def get_class_for_class_path(class_path):
clazz = getattr(module, class_name)
return ContentType.objects.get_for_model(clazz)
with open('/opt/netbox/initializers/custom_fields.yml', 'r') as stream:
file = Path('/opt/netbox/initializers/custom_fields.yml')
if not file.is_file():
sys.exit()
with file.open('r') as stream:
yaml = YAML(typ='safe')
customfields = yaml.load(stream)