Allow the zk_conn to be none properly

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

View File

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