Added TimeoutError handling for Windows systems.

4 years ago · 305fc8f34a
2 changed files with 2 additions and 0 deletions
--- 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)
 )
--- 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')