diff --git a/module/netbox/connection.py b/module/netbox/connection.py index 5ff6dfd..210deaf 100644 --- a/module/netbox/connection.py +++ b/module/netbox/connection.py @@ -66,10 +66,14 @@ class NetBoxHandler: # keep track of already resolved dependencies resolved_dependencies = set() - def __init__(self, settings=None, inventory=None): + # set bogus default version + version = "0.0.1" + + def __init__(self, settings=None, inventory=None, version=None): self.settings = settings self.inventory = inventory + self.version = version self.parse_config_settings(settings) @@ -191,7 +195,8 @@ class NetBoxHandler: header = { "Authorization": f"Token {self.api_token}", - "User-Agent": "netbox-sync/0.0.1" + "User-Agent": f"netbox-sync/{self.version}", + "Content-Type": "application/json" } session = requests.Session() diff --git a/netbox-sync.py b/netbox-sync.py index 744e6a0..b2eb5f4 100755 --- a/netbox-sync.py +++ b/netbox-sync.py @@ -79,7 +79,7 @@ def main(): netbox_settings = get_config(config_handler, section="netbox", valid_settings=NetBoxHandler.settings) # establish NetBox connection - nb_handler = NetBoxHandler(settings=netbox_settings, inventory=inventory) + nb_handler = NetBoxHandler(settings=netbox_settings, inventory=inventory, version=__version__) # if purge was selected we go ahead and remove all items which were managed by this tools if args.purge is True: