Lookup: Updates to look into the environment for NetBox info (#391)

Author: jvanderaa

plugins/lookup/nb_lookup.py

@@ -11,6 +11,7 @@
 
 from __future__ import absolute_import, division, print_function
 
+import os
 import functools
 from pprint import pformat
 
@@ -41,6 +42,10 @@
         api_endpoint:
             description:
                 - The URL to the Netbox instance to query
+            env:
+                # in order of precendence
+                - name: NETBOX_API
+                - name: NETBOX_URL
             required: True
         api_filter:
             description:
@@ -54,6 +59,10 @@
             description:
                 - The API token created through Netbox
                 - This may not be required depending on the Netbox setup.
+            env:
+                # in order of precendence
+                - name: NETBOX_TOKEN
+                - name: NETBOX_API_TOKEN
             required: False
         validate_certs:
             description:
@@ -283,8 +292,16 @@ class LookupModule(LookupBase):
 
     def run(self, terms, variables=None, **kwargs):
 
-        netbox_api_token = kwargs.get("token")
-        netbox_api_endpoint = kwargs.get("api_endpoint")
+        netbox_api_token = (
+            kwargs.get("token")
+            or os.getenv("NETBOX_TOKEN")
+            or os.getenv("NETBOX_API_TOKEN")
+        )
+        netbox_api_endpoint = (
+            kwargs.get("api_endpoint")
+            or os.getenv("NETBOX_API")
+            or os.getenv("NETBOX_URL")
+        )
         netbox_ssl_verify = kwargs.get("validate_certs", True)
         netbox_private_key_file = kwargs.get("key_file")
         netbox_api_filter = kwargs.get("api_filter")