mirror of
https://github.com/netbox-community/Device-Type-Library-Import.git
synced 2025-02-12 23:49:19 +01:00
98857fc649
* Delete gitcmd.py * Delete nb-dt-import.py * Add files via upload * Logging cleanup (#78) * Removed multiple imports of settings.py * stating to abstract the netbox api calls to their own class * Abstracted away determine features from main script, implemented as part of class initialization * added helper functions to get repos relative & absolute path * renaming gitcmd to repo * starting to abstract away the get_files * fixed issue where spaces and commas in vendor list with/without spaces breaks matching * Added prelim fix for slugs if same issue vendors arg was facing exists. untested * Finished abstracting the get_files function. Reduced fors and ifs to be cleaner and more efficent * abstracted getFiles to repo class. Fixed slug issue not matching because of new slug format. added non-halting log function and renamed exception handler to log handler. * utilized new logging class throughout script to reduce excess logging * Abstracted and optimized create manufacturers * Abstracted the create interfaces for devices to the netbox api class * Fixed regression where check manufactuerers did not have the latest list * Fixed regression caused by externally calling script. Discovered from https://github.com/netbox-community/Device-Type-Library-Import/pull/76 * abstracted all device interfaces to the devicetype class. optimized function calls to reduce duplicate code and reduce extra log calls * Ran against all devices and passed with flying colors * formatting settings.py * formatting repo.py * formatting main file * formatting log_handler.py * added back executable on file (#79) * Add more info to failed device_type creations (#81) --------- Co-authored-by: Philipp Rintz <13933258+p-rintz@users.noreply.github.com>
54 lines
1.9 KiB
Python
Executable File
54 lines
1.9 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
from collections import Counter
|
|
from datetime import datetime
|
|
import yaml
|
|
import pynetbox
|
|
from glob import glob
|
|
import os
|
|
|
|
import settings
|
|
from netbox_api import NetBox
|
|
|
|
|
|
def main():
|
|
startTime = datetime.now()
|
|
args = settings.args
|
|
|
|
netbox = NetBox(settings)
|
|
files, vendors = settings.dtl_repo.get_devices(
|
|
f'{settings.dtl_repo.repo_path}/device-types/', args.vendors)
|
|
|
|
settings.handle.log(f'{len(vendors)} Vendors Found')
|
|
device_types = settings.dtl_repo.parse_files(files, slugs=args.slugs)
|
|
settings.handle.log(f'{len(device_types)} Device-Types Found')
|
|
netbox.create_manufacturers(vendors)
|
|
netbox.create_device_types(device_types)
|
|
|
|
if netbox.modules:
|
|
settings.handle.log("Modules Enabled. Creating Modules...")
|
|
files, vendors = settings.dtl_repo.get_devices(
|
|
f'{settings.dtl_repo.repo_path}/module-types/', args.vendors)
|
|
settings.handle.log(f'{len(vendors)} Module Vendors Found')
|
|
module_types = settings.dtl_repo.parse_files(files, slugs=args.slugs)
|
|
settings.handle.log(f'{len(module_types)} Module-Types Found')
|
|
netbox.create_manufacturers(vendors)
|
|
netbox.create_module_types(module_types)
|
|
|
|
settings.handle.log('---')
|
|
settings.handle.verbose_log(
|
|
f'Script took {(datetime.now() - startTime)} to run')
|
|
settings.handle.log(f'{netbox.counter["added"]} devices created')
|
|
settings.handle.log(
|
|
f'{netbox.counter["updated"]} interfaces/ports updated')
|
|
settings.handle.log(
|
|
f'{netbox.counter["manufacturer"]} manufacturers created')
|
|
if settings.NETBOX_FEATURES['modules']:
|
|
settings.handle.log(
|
|
f'{netbox.counter["module_added"]} modules created')
|
|
settings.handle.log(
|
|
f'{netbox.counter["module_port_added"]} module interface / ports created')
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|