Added recursive_delete method to KazooClient

kazoo/client.py

@@ -1,7 +1,7 @@
 import logging
 
 from kazoo.zkclient import ZooKeeperClient, WatchedEvent, KeeperState,\
-    EventType, NodeExistsException
+    EventType, NodeExistsException, NoNodeException
 from kazoo.retry import KazooRetry
 
 log = logging.getLogger(__name__)
@@ -211,6 +211,18 @@ def ensure_path(self, path):
                     # someone else created the node. how sweet!
                     pass
 
+    def recursive_delete(self, path):
+        """Recursively delete a ZNode and all of its children
+        """
+        children = self.get_children(path)
+        if children:
+            for child in children:
+                self.recursive_delete(path + "/" + child)
+        try:
+            self.delete(path)
+        except NoNodeException:
+            pass
+
     def with_retry(self, func, *args, **kwargs):
         """Run a method repeatedly in the face of transient ZK errors
         """

kazoo/zkclient.py

@@ -43,16 +43,16 @@ class WatchedEvent(namedtuple('WatchedEvent', ('type', 'state', 'path'))):
     """
 
 class ZooKeeperClient(object):
-
-    DEFAULT_TIMEOUT = 10000
-
     """A gevent-friendly wrapper of the Apache ZooKeeper zkpython client
 
     TODO lots to do:
     * better handling of ZK client session events
     * disconnected state handling
     * the rest of the operations
     """
+
+    DEFAULT_TIMEOUT = 10000
+
     def __init__(self, hosts, watcher=None, timeout=10000):
         self._hosts = hosts
         self._watcher = watcher