Skip to content

Commit 08f14a0

Browse files
authored
[PECO-1961] On non-retryable error, ensure PySQL includes useful information in error (databricks#447)
* added error info on non-retryable error
1 parent a151df2 commit 08f14a0

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

src/databricks/sql/utils.py

+7-1
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,7 @@
3131
import logging
3232

3333
BIT_MASKS = [1, 2, 4, 8, 16, 32, 64, 128]
34+
DEFAULT_ERROR_CONTEXT = "Unknown error"
3435

3536
logger = logging.getLogger(__name__)
3637

@@ -407,7 +408,12 @@ def user_friendly_error_message(self, no_retry_reason, attempt, elapsed):
407408
user_friendly_error_message = "{}: {}".format(
408409
user_friendly_error_message, self.error_message
409410
)
410-
return user_friendly_error_message
411+
try:
412+
error_context = str(self.error)
413+
except:
414+
error_context = DEFAULT_ERROR_CONTEXT
415+
416+
return user_friendly_error_message + ". " + error_context
411417

412418

413419
# Taken from PyHive

0 commit comments

Comments
 (0)