adding token auth

Author: Naor Livne

NebulaPythonSDK/sdk.py

@@ -10,18 +10,22 @@ class Nebula:
 
     # the nebula class init module serves as the login against the nebula API as it's the only shared thing among the
     # class functions
-    def __init__(self, username=None, password=None, host="127.0.0.1", port=80, protocol="http", request_timeout=60):
+    def __init__(self, username=None, password=None, token=None, host="127.0.0.1", port=80, protocol="http",
+                 request_timeout=60):
         self.request_timeout = request_timeout
         self.username = username
         self.password = password
+        self.token = token
         self.protocol = protocol
         self.port = port
         self.host = protocol + "://" + host + ":" + str(port)
         self.headers = {
             'content-type': "application/json",
             'cache-control': "no-cache"
         }
-        if username is not None and password is not None:
+        if token is not None:
+            self.headers["authorization"] = "Bearer " + self.token
+        elif username is not None and password is not None:
             user_pass_combo = username + ":" + password
             if six.PY3 is True:
                 user_pass_combo = user_pass_combo.encode('utf-8')

test/test_nebula_python_sdk.py

@@ -8,12 +8,13 @@
 def nebula_connection():
     nebula_user = os.getenv("NEBULA_TEST_USERNAME", "nebula")
     nebula_password = os.getenv("NEBULA_TEST_PASSWORD", "nebula")
+    nebula_token = os.getenv("NEBULA_TEST_TOKEN", None)
     nebula_hostname = os.getenv("NEBULA_TEST_HOST", "127.0.0.1")
     nebula_port = int(os.getenv("NEBULA_TEST_PORT", "80"))
     nebula_protocol = os.getenv("NEBULA_TEST_PROTOCOL", "http")
     nebula_request_timeout = int(os.getenv("NEBULA_TEST_REQUEST_TIMEOUT", "60"))
     connection = Nebula(username=nebula_user, password=nebula_password, host=nebula_hostname, port=nebula_port,
-                        protocol=nebula_protocol, request_timeout=nebula_request_timeout)
+                        protocol=nebula_protocol, request_timeout=nebula_request_timeout, token=nebula_token)
     return connection
 
 

test/travis_ci_scripts/docker-compose.yml

@@ -26,3 +26,4 @@ services:
        SCHEMA_NAME: nebula
        BASIC_AUTH_PASSWORD: nebula
        BASIC_AUTH_USER: nebula
+       AUTH_TOKEN: nebula