Verify if key exists before reading

This commit is contained in:
Joshua Boniface 2021-06-13 15:39:43 -04:00
parent d2f93b3a2e
commit 3bad3de720
1 changed files with 7 additions and 2 deletions

View File

@ -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):
"""