Allow the zk_conn to be none properly

This commit is contained in:
Joshua Boniface 2019-04-09 16:25:45 -04:00
parent 989c5f6bed
commit ba4a44f5e7
1 changed files with 3 additions and 2 deletions

View File

@ -42,7 +42,8 @@ zk_host = ''
CONTEXT_SETTINGS = dict(help_option_names=['-h', '--help'], max_content_width=120)
def cleanup(retcode, retmsg, zk_conn):
def cleanup(retcode, retmsg, zk_conn=None):
if zk_conn:
pvc_common.stopZKConnection(zk_conn)
if retcode == True:
if retmsg != '':