From 901cba7a40d67ec96c06abe109e3cf51a992b24d Mon Sep 17 00:00:00 2001 From: Jean-Marc Saffroy Date: Tue, 17 Jul 2018 20:20:50 +0200 Subject: [PATCH] fix(core): get_children with include_data=True uses GetChildren2 types (#514) --- kazoo/protocol/connection.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/kazoo/protocol/connection.py b/kazoo/protocol/connection.py index 7f686c09..27388e12 100644 --- a/kazoo/protocol/connection.py +++ b/kazoo/protocol/connection.py @@ -23,6 +23,7 @@ Connect, Exists, GetChildren, + GetChildren2, Ping, PingInstance, ReplyHeader, @@ -398,7 +399,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)