@@ -48,38 +48,38 @@ def handle_exception(exc: Exception, console: Console, exit_code: int) -> NoRetu
4848 if isinstance (exc , Exit ):
4949 raise typer .Exit (code = exc .exit_code )
5050 if isinstance (exc , AuthenticationError ):
51- console .print (f"[red]Authentication failure: { str ( exc ) } " )
51+ console .print (f"[red]Authentication failure: { exc !s } " )
5252 raise typer .Exit (code = exit_code )
5353 if isinstance (exc , (ServerNotReachableError , ServerNotResponsiveError )):
54- console .print (f"[red]{ str ( exc ) } " )
54+ console .print (f"[red]{ exc !s } " )
5555 raise typer .Exit (code = exit_code )
5656 if isinstance (exc , HTTPError ):
57- console .print (f"[red]HTTP communication failure: { str ( exc ) } on { exc .request .method } to { exc .request .url } " )
57+ console .print (f"[red]HTTP communication failure: { exc !s } on { exc .request .method } to { exc .request .url } " )
5858 raise typer .Exit (code = exit_code )
5959 if isinstance (exc , GraphQLError ):
6060 print_graphql_errors (console = console , errors = exc .errors )
6161 raise typer .Exit (code = exit_code )
6262 if isinstance (exc , (SchemaNotFoundError , NodeNotFoundError )):
63- console .print (f"[red]Error: { str ( exc ) } " )
63+ console .print (f"[red]Error: { exc !s } " )
6464 raise typer .Exit (code = exit_code )
6565
66- console .print (f"[red]Error: { str ( exc ) } " )
66+ console .print (f"[red]Error: { exc !s } " )
6767 console .print (traceback .format_exc ())
6868 raise typer .Exit (code = exit_code )
6969
7070
7171def catch_exception (
7272 console : Optional [Console ] = None , exit_code : int = 1
73- ) -> Callable [[Callable [..., T ]], Callable [..., Union [T , Coroutine [Any , Any , T ], NoReturn ]]]:
73+ ) -> Callable [[Callable [..., T ]], Callable [..., Union [T , Coroutine [Any , Any , T ]]]]:
7474 """Decorator to handle exception for commands."""
7575 if not console :
7676 console = Console ()
7777
78- def decorator (func : Callable [..., T ]) -> Callable [..., Union [T , Coroutine [Any , Any , T ], NoReturn ]]:
78+ def decorator (func : Callable [..., T ]) -> Callable [..., Union [T , Coroutine [Any , Any , T ]]]:
7979 if asyncio .iscoroutinefunction (func ):
8080
8181 @wraps (func )
82- async def async_wrapper (* args : Any , ** kwargs : Any ) -> Union [ T , NoReturn ] :
82+ async def async_wrapper (* args : Any , ** kwargs : Any ) -> T :
8383 try :
8484 return await func (* args , ** kwargs )
8585 except (Error , Exception ) as exc : # pylint: disable=broad-exception-caught
@@ -88,7 +88,7 @@ async def async_wrapper(*args: Any, **kwargs: Any) -> Union[T, NoReturn]:
8888 return async_wrapper
8989
9090 @wraps (func )
91- def wrapper (* args : Any , ** kwargs : Any ) -> Union [ T , NoReturn ] :
91+ def wrapper (* args : Any , ** kwargs : Any ) -> T :
9292 try :
9393 return func (* args , ** kwargs )
9494 except (Error , Exception ) as exc : # pylint: disable=broad-exception-caught
0 commit comments