extern/Device-Type-Library-Import
1
0
Fork 0
mirror of https://github.com/netbox-community/Device-Type-Library-Import.git synced 2025-06-08 19:07:01 +02:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #32 from rsp2k/master

Browse Source 
Import all YAML files
This commit is contained in:
minitriga 2021-10-04 13:27:28 +01:00 committed by GitHub
commit 3aaf98bfe4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
nb-dt-import.py
View File

@ -26,8 +26,10 @@ def update_package(path: str):
def slugFormat(name):
return name.lower().replace(' ', '_')
YAML_EXTENSIONS = ['yml', 'yaml']
def getFiles(vendors=None):
files = []
discoveredVendors = []
base_path = './repo/device-types/'
@ -38,14 +40,16 @@ def getFiles(vendors=None):
if vendor.lower() == folder.lower():
discoveredVendors.append({'name': folder,
'slug': slugFormat(folder)})
files.extend(glob.glob(base_path + folder + '/*.yaml'))
for extension in YAML_EXTENSIONS:
files.extend(glob.glob(base_path + folder + f'/*.{extension}'))
else:
for r, d, f in os.walk(base_path):
for folder in d:
if folder.lower() != "Testing":
discoveredVendors.append({'name': folder,
'slug': slugFormat(folder)})
files.extend(glob.glob(base_path + '[!Testing]*/*.yaml'))
for extension in YAML_EXTENSIONS:
files.extend(glob.glob(base_path + f'[!Testing]*/*.{extension}'))
return files, discoveredVendors