Skip to content

Commit ce0d159

Browse files
Apply suggestions from code review
Co-authored-by: Favian (Ian) Samatha <[email protected]>
1 parent 13cdf2b commit ce0d159

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

aws_advanced_python_wrapper/utils/pg_exception_handler.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@ class PgExceptionHandler(ExceptionHandler):
4949
_READ_ONLY_ERROR_CODE: str = "25006" # read only sql transaction
5050

5151
def is_network_exception(self, error: Optional[Exception] = None, sql_state: Optional[str] = None) -> bool:
52-
if isinstance(error, AwsConnectError) or isinstance(error, QueryTimeoutError):
52+
if isinstance(error, (AwsConnectError, QueryTimeoutError)):
5353
return True
5454

5555
if isinstance(error, AwsWrapperError):
@@ -61,7 +61,7 @@ def _is_network_error(self, error: Optional[BaseException], sql_state: Optional[
6161
if error is None:
6262
return False
6363

64-
if isinstance(error, AwsConnectError) or isinstance(error, QueryTimeoutError) or isinstance(error, ConnectionTimeout):
64+
if isinstance(error, (AwsConnectError, QueryTimeoutError, ConnectionTimeout)):
6565
return True
6666

6767
if sql_state is None:
@@ -89,7 +89,7 @@ def _is_login_error(self, error: Optional[BaseException] = None, sql_state: Opti
8989
if error is None:
9090
return False
9191

92-
if isinstance(error, InvalidAuthorizationSpecification) or isinstance(error, InvalidPassword):
92+
if isinstance(error, (InvalidAuthorizationSpecification, InvalidPassword)):
9393
return True
9494

9595
if sql_state is None:

0 commit comments

Comments
 (0)