feat: add configuration for stack trace

Author: creynolds15

examples/list_jobs.py

@@ -2,7 +2,7 @@
 import time
 
 start = time.time()
-tamr_client = TamrApiClient('<host-name>', [('x-api-key', '<api-key>')])
+tamr_client = TamrApiClient('<host-name>', [('x-api-key', '<api-key>')], grpc_stack_trace=True)
 one = time.time()
 
 

tamr_sdk/api_client.py

@@ -1,9 +1,10 @@
 from tamr_sdk.jobs.jobs_client import JobsClient
 
 class TamrApiClient:
-    def __init__(self, host, metadata):
+    def __init__(self, host, metadata, grpc_stack_trace=False):
         self.host = host
         self.metadata = metadata
+        self.grpc_stack_trace = grpc_stack_trace
 
     def jobs(self):
-        return JobsClient(self.host, self.metadata)
+        return JobsClient(self.host, self.metadata, self.grpc_stack_trace)

tamr_sdk/jobs/jobs_client.py

@@ -5,11 +5,12 @@
 from tamr_sdk.utils.handler import exception_handler
 
 class JobsClient:
-    def __init__(self, host, metadata):
+    def __init__(self, host, metadata, grpc_stack_trace=False):
         credentials = grpc.ssl_channel_credentials()
         channel = grpc.secure_channel(host, credentials)
         self.metadata = metadata
         self.stub = JobsStub(channel)
+        self.grpc_stack_trace = grpc_stack_trace
 
     @exception_handler
     def get_job(self, job_id):

tamr_sdk/utils/handler.py

@@ -14,17 +14,25 @@ def exception_handler(
     def handle_grpc_status_codes(
         _self: SelfT, *args: ArgT.args, **kwargs: ArgT.kwargs
     ) -> ReturnT:
+        grpc_stack_trace = _self.grpc_stack_trace
         try:
             func(_self, *args, **kwargs)
         except grpc._channel._InactiveRpcError as e:
             if e.code() == grpc.StatusCode.NOT_FOUND:
-                raise ValueError(e.details())
+                if not grpc_stack_trace:
+                    raise ValueError(e.details()) from None
+                else:
+                    raise ValueError(e.details())
             elif e.code() == grpc.StatusCode.UNAVAILABLE:
-                raise ConnectionError(e.details())
+                if not grpc_stack_trace:
+                    raise ConnectionError(e.details()) from None
+                else:
+                    raise ConnectionError(e.details())
             elif e.code() == grpc.StatusCode.UNAUTHENTICATED:
-                raise ConnectionRefusedError(
-                    "Error connecting to client. Authentication is not valid"
-                )
+                if not grpc_stack_trace:
+                    raise ConnectionRefusedError("Error connecting to client. Authentication is not valid") from None
+                else:
+                    raise ConnectionRefusedError("Error connecting to client. Authentication is not valid")
             else:
                 raise e