fix(core): get_children with include_data=True uses GetChildren2 types (

#514)
python-zk · Oct 8, 2018 · 4ba3231 · 4ba3231
1 parent de20be9
commit 4ba3231
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/kazoo/protocol/connection.py b/kazoo/protocol/connection.py
@@ -23,6 +23,7 @@
     Connect,
     Exists,
     GetChildren,
+    GetChildren2,
     Ping,
     PingInstance,
     ReplyHeader,
@@ -397,7 +398,7 @@ def _read_response(self, header, buffer, offset):
             # Determine if watchers should be registered
             watcher = getattr(request, 'watcher', None)
             if not client._stopped.is_set() and watcher:
-                if isinstance(request, GetChildren):
+                if isinstance(request, (GetChildren, GetChildren2)):
                     client._child_watchers[request.path].add(watcher)
                 else:
                     client._data_watchers[request.path].add(watcher)