diff --git a/qds_sdk/connection.py b/qds_sdk/connection.py index 0398acb5..df552da8 100644 --- a/qds_sdk/connection.py +++ b/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)