From b60f65a2caceaeed9bba6d87b0056ce7d7f1adfa Mon Sep 17 00:00:00 2001
From: jouir <julien.riou@outlook.com>
Date: Tue, 4 Jul 2017 16:13:18 +0200
Subject: [PATCH] Compatibility with old kazoo (#468)

```python
AttributeError: 'KazooClient' object has no attribute '_retry'
```

Fixes #467
---
 patroni/dcs/zookeeper.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/patroni/dcs/zookeeper.py b/patroni/dcs/zookeeper.py
index 1215d820..82cdff4b 100644
--- a/patroni/dcs/zookeeper.py
+++ b/patroni/dcs/zookeeper.py
@@ -116,7 +116,8 @@ class ZooKeeper(AbstractDCS):
             return True
 
     def set_retry_timeout(self, retry_timeout):
-        self._client._retry.deadline = retry_timeout
+        retry = self._client.retry if isinstance(self._client.retry, KazooRetry) else self._client._retry
+        retry.deadline = retry_timeout
 
     def get_node(self, key, watch=None):
         try: