Improve error message for get signed url failure

flytekit/clients/friendly.py

@@ -1021,7 +1021,7 @@ def get_upload_signed_url(
                 )
             )
         except Exception as e:
-            raise RuntimeError(f"Failed to get signed url for {filename}, reason: {e}")
+            raise RuntimeError(f"Failed to get signed url for {filename}.") from e
 
     def get_download_signed_url(
         self, native_url: str, expires_in: datetime.timedelta = None

flytekit/clients/grpc_utils/wrap_exception_interceptor.py

@@ -4,7 +4,7 @@
 import grpc
 
 from flytekit.exceptions.base import FlyteException
-from flytekit.exceptions.system import FlyteSystemException
+from flytekit.exceptions.system import FlyteSystemException, FlyteSystemUnavailableException
 from flytekit.exceptions.user import (
     FlyteAuthenticationException,
     FlyteEntityAlreadyExistsException,
@@ -28,6 +28,8 @@ def _raise_if_exc(request: typing.Any, e: Union[grpc.Call, grpc.Future]):
                 raise FlyteEntityNotExistException() from e
             elif e.code() == grpc.StatusCode.INVALID_ARGUMENT:
                 raise FlyteInvalidInputException(request) from e
+            elif e.code() == grpc.StatusCode.UNAVAILABLE:
+                raise FlyteSystemUnavailableException() from e
         raise FlyteSystemException() from e
 
     def intercept_unary_unary(self, continuation, client_call_details, request):

flytekit/clis/sdk_in_container/utils.py

@@ -81,7 +81,7 @@ def pretty_print_grpc_error(e: grpc.RpcError):
     """
     if isinstance(e, grpc._channel._InactiveRpcError):  # noqa
         click.secho(f"RPC Failed, with Status: {e.code()}", fg="red", bold=True)
-        click.secho(f"\tdetails: {e.details()}", fg="magenta", bold=True)
+        click.secho(f"\tDetails: {e.details()}", fg="magenta", bold=True)
     return
 
 
@@ -113,7 +113,6 @@ def pretty_print_traceback(e: Exception, verbosity: int = 1):
     Print the traceback in a nice formatted way if verbose is set to True.
     """
     console = Console()
-    tb = e.__cause__.__traceback__ if e.__cause__ else e.__traceback__
 
     if verbosity == 0:
         console.print(Traceback.from_exception(type(e), e, None))
@@ -124,10 +123,11 @@ def pretty_print_traceback(e: Exception, verbosity: int = 1):
             f" For more verbose output, use the flags -vv or -vvv.",
             fg="yellow",
         )
-        new_tb = remove_unwanted_traceback_frames(tb, unwanted_module_names)
+
+        new_tb = remove_unwanted_traceback_frames(e.__traceback__, unwanted_module_names)
         console.print(Traceback.from_exception(type(e), e, new_tb))
     elif verbosity >= 2:
-        console.print(Traceback.from_exception(type(e), e, tb))
+        console.print(Traceback.from_exception(type(e), e, e.__traceback__))
     else:
         raise ValueError(f"Verbosity level must be between 0 and 2. Got {verbosity}")
 

flytekit/exceptions/system.py

@@ -5,6 +5,13 @@ class FlyteSystemException(_base_exceptions.FlyteRecoverableException):
     _ERROR_CODE = "SYSTEM:Unknown"
 
 
+class FlyteSystemUnavailableException(FlyteSystemException):
+    _ERROR_CODE = "SYSTEM:Unavailable"
+
+    def __str__(self):
+        return "Flyte cluster is currently unavailable. Please make sure the cluster is up and running."
+
+
 class FlyteNotImplementedException(FlyteSystemException, NotImplementedError):
     _ERROR_CODE = "SYSTEM:NotImplemented"