Credentials attribute in onboarding class

Author: mzbroch

netbox_onboarding/onboard.py

@@ -87,20 +87,20 @@ class OnboardingManager:
 
     def __init__(self, ot, username, password, secret):
         """Inits class."""
-        self.username = username
-        self.password = password
-        self.secret = secret
-
         # Create instance of Onboarding Task Manager class:
         otm = OnboardingTaskManager(ot)
 
+        self.username = username or settings.NAPALM_USERNAME
+        self.password = password or settings.NAPALM_PASSWORD
+        self.secret = secret or otm.optional_args.get("secret", None) or settings.NAPALM_ARGS.get("secret", None)
+
         netdev = NetdevKeeper(
             hostname=otm.ip_address,
             port=otm.port,
             timeout=otm.timeout,
-            username=self.username or settings.NAPALM_USERNAME,
-            password=self.password or settings.NAPALM_PASSWORD,
-            secret=self.secret or otm.optional_args.get("secret", None) or settings.NAPALM_ARGS.get("secret", None),
+            username=self.username,
+            password=self.password,
+            secret=self.secret,
             napalm_driver=otm.napalm_driver,
             optional_args=otm.optional_args or settings.NAPALM_ARGS,
         )
@@ -129,6 +129,7 @@ def __init__(self, ot, username, password, secret):
         }
 
         onboarding_cls = netdev_dict["onboarding_class"]()
+        onboarding_cls.credentials = {"username": self.username, "password": self.password, "secret": self.secret}
         onboarding_cls.run(onboarding_kwargs=onboarding_kwargs)
 
         self.created_device = onboarding_cls.created_device

netbox_onboarding/onboarding/onboarding.py

@@ -21,6 +21,7 @@ class Onboarding:
     def __init__(self):
         """Init the class."""
         self.created_device = None
+        self.credentials = None
 
     def run(self, onboarding_kwargs):
         """Implement run method."""