Fix token authorization for get_version (#734)

Author: arthanson

pynetbox/core/query.py

@@ -295,8 +295,7 @@ def get_version(self):
         RequestError if req.ok returns false.
         """
         headers = {"Content-Type": "application/json"}
-        if self.token:
-            headers["authorization"] = "Token {}".format(self.token)
+        self._add_auth_header(headers)
         req = self.http_session.get(
             self.normalize_url(self.base),
             headers=headers,
@@ -316,11 +315,7 @@ def get_status(self):
         RequestError if request is not successful.
         """
         headers = {"Content-Type": "application/json"}
-        if self.token:
-            if _is_v2_token(self.token):
-                headers["authorization"] = "Bearer {}".format(self.token)
-            else:
-                headers["authorization"] = "Token {}".format(self.token)
+        self._add_auth_header(headers)
         req = self.http_session.get(
             "{}status/".format(self.normalize_url(self.base)),
             headers=headers,
@@ -337,6 +332,18 @@ def normalize_url(self, url):
 
         return url
 
+    def _add_auth_header(self, headers):
+        """Add authorization header to headers dict if token is present.
+
+        ## Parameters
+        * **headers** (dict): Headers dictionary to update with authorization.
+        """
+        if self.token:
+            if _is_v2_token(self.token):
+                headers["authorization"] = "Bearer {}".format(self.token)
+            else:
+                headers["authorization"] = "Token {}".format(self.token)
+
     def _make_call(self, verb="get", url_override=None, add_params=None, data=None):
         # Extract any file-like objects from data
         files = None
@@ -361,11 +368,7 @@ def _make_call(self, verb="get", url_override=None, add_params=None, data=None):
         if should_be_json_body:
             headers["Content-Type"] = "application/json"
 
-        if self.token:
-            if _is_v2_token(self.token):
-                headers["authorization"] = "Bearer {}".format(self.token)
-            else:
-                headers["authorization"] = "Token {}".format(self.token)
+        self._add_auth_header(headers)
 
         params = {}
         if not url_override: