From 3bad3de7206f87ba09478a93ae6bb9edcdc974d6 Mon Sep 17 00:00:00 2001 From: "Joshua M. Boniface" Date: Sun, 13 Jun 2021 15:39:43 -0400 Subject: [PATCH] Verify if key exists before reading --- daemon-common/zkhandler.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/daemon-common/zkhandler.py b/daemon-common/zkhandler.py index e3f2a63e..129905c1 100644 --- a/daemon-common/zkhandler.py +++ b/daemon-common/zkhandler.py @@ -208,8 +208,13 @@ class ZKHandler(object): """ Read data from a key """ - path = self.get_schema_path(key) - return self.zk_conn.get(path)[0].decode(self.encoding) + if self.exists(key): + path = self.get_schema_path(key) + data = self.zk_conn.get(path)[0].decode(self.encoding) + else: + data = None + + return data def write(self, kvpairs): """