Handle zkpython binding bugs properly for older versions of the lib.

1 files changed, 7 insertions, 0 deletions

diff --git a/kazoo/retry.py b/kazoo/retry.py
index b5a52a0..60381ce 100644
--- a/kazoo/retry.py
+++ b/kazoo/retry.py
@@ -1,6 +1,7 @@
 from zookeeper import (
     ClosingException,
     ConnectionLossException,
+    InvalidStateException,
     OperationTimeoutException,
     SessionExpiredException
 )
@@ -17,6 +18,12 @@ class KazooRetry(object):
         ConnectionLossException,
         OperationTimeoutException,
         SessionExpiredException,
+
+        # Occurs when a command is run on a session handle that expired, if it
+        # manages to run exactly when it expired but before the handle was
+        # removed for reconnection
+        InvalidStateException,
+
         ForceRetryError
     )