SDK-375: Added support for Qubole-trace-Id (#285)

Added support for Qubole-trace-Id

qds_sdk/connection.py

@@ -5,6 +5,7 @@
 import json
 import pkg_resources
 from requests.adapters import HTTPAdapter
+from datetime import datetime
 try:
     from requests.packages.urllib3.poolmanager import PoolManager
 except ImportError:
@@ -130,7 +131,13 @@ def _handle_error(response):
 
         if 200 <= code < 400:
             return
-
+
+        if 'X-Qubole-Trace-Id' in response.headers:
+            now = datetime.now()
+            time = now.strftime('%Y-%m-%d %H:%M:%S')
+            format_list = [time,response.headers['X-Qubole-Trace-Id']]
+            sys.stderr.write("[{}] Request ID is: {}. Please share it with Qubole Support team for any assistance".format(*format_list) + "\n")
+
         if code == 400:
             sys.stderr.write(response.text + "\n")
             raise BadRequest(response)