diff --git a/rcon/rconclt.py b/rcon/rconclt.py index fe98e36..dac4190 100644 --- a/rcon/rconclt.py +++ b/rcon/rconclt.py @@ -21,6 +21,7 @@ __all__ = ['get_credentials', 'main'] ERRORS = ( (ConnectionRefusedError, 'Connection refused.', 3), (timeout, 'Connection timeout.', 4), + (TimeoutError, 'Connection timeout.', 4), (RequestIdMismatch, 'Unexpected request ID mismatch.', 5), (WrongPassword, 'Wrong password.', 6) ) diff --git a/rcon/rconshell.py b/rcon/rconshell.py index bf107ca..9296b14 100644 --- a/rcon/rconshell.py +++ b/rcon/rconshell.py @@ -19,6 +19,7 @@ __all__ = ['get_args', 'main'] ERRORS = ( (ConnectionRefusedError, 'Connection refused.', 3), (timeout, 'Connection timeout.', 4), + (TimeoutError, 'Connection timeout.', 4), (RequestIdMismatch, 'Unexpected request ID mismatch.', 5) ) LOGGER = getLogger('rconshell')