Add module to the table URL segment in ScaleClient.py
[integration/test.git] / csit / libraries / AuthStandalone.py
index 5e0c0eae4b180acc5a04a992224306d20b299a1b..5cb840a4f9054b38e7fc23a12d9cddac7140fd63 100644 (file)
@@ -32,16 +32,16 @@ as URIs not starting with /restconf/ are not supported yet.
 # terms of the Eclipse Public License v1.0 which accompanies this distribution,
 # and is available at http://www.eclipse.org/legal/epl-v10.html
 
+import json
+import requests
+
+
 __author__ = "Vratko Polak"
 __copyright__ = "Copyright(c) 2015, Cisco Systems, Inc."
 __license__ = "Eclipse Public License v1.0"
 __email__ = "vrpolak@cisco.com"
 
 
-import json
-import requests
-
-
 #
 # Karaf Keyword definitions.
 #
@@ -92,7 +92,9 @@ class _BasicReusingSession(object):
         if username:
             self.session.auth = (username, password)  # May work with non-string values
         else:
-            self.session.auth = None  # Supports "no authentication mode" as in odl-restconf-noauth
+            self.session.auth = (
+                None  # Supports "no authentication mode" as in odl-restconf-noauth
+            )
 
     def robust_method(self, method, uri, **kwargs):
         """Try method once using session credentials. Return last response."""
@@ -109,7 +111,9 @@ class _BasicClosingSession(object):
         if username:
             self.auth = (username, password)  # May work with non-string values
         else:
-            self.auth = None  # Supports "no authentication mode" as in odl-restconf-noauth
+            self.auth = (
+                None  # Supports "no authentication mode" as in odl-restconf-noauth
+            )
         self.session = None
 
     def robust_method(self, method, uri, **kwargs):
@@ -144,7 +148,9 @@ class _TokenReusingSession(object):
         if self.session:
             self.session.close()
         self.session = requests.Session()
-        resp = self.session.post(self.auth_url, data=self.auth_data, headers=self.auth_header)
+        resp = self.session.post(
+            self.auth_url, data=self.auth_data, headers=self.auth_header
+        )
         resp_obj = json.loads(resp.text)
         try:
             token = resp_obj["access_token"]
@@ -152,7 +158,6 @@ class _TokenReusingSession(object):
             raise RuntimeError("Parse failed: " + resp.text)
         self.token = token
         # TODO: Use logging so that callers could see token refreshes.
-        # print "DEBUG: token:", token
         # We keep self.session to use for the following restconf requests.
 
     def oneshot_method(self, method, uri, **kwargs):
@@ -192,7 +197,9 @@ class _TokenClosingSession(object):
         if self.session:
             self.session.close()
         self.session = requests.Session()
-        resp = self.session.post(self.auth_url, data=self.auth_data, headers=self.auth_header)
+        resp = self.session.post(
+            self.auth_url, data=self.auth_data, headers=self.auth_header
+        )
         resp_obj = json.loads(resp.text)
         try:
             token = resp_obj["access_token"]
@@ -200,7 +207,6 @@ class _TokenClosingSession(object):
             raise RuntimeError("Parse failed: " + resp.text)
         self.token = token
         # TODO: Use logging so that callers could see token refreshes.
-        # print "DEBUG: token:", token
         # We keep self.session to use for the following restconf requests.
 
     def oneshot_method(self, method, uri, **kwargs):