Allow the zk_conn to be none properly
This commit is contained in:
parent
989c5f6bed
commit
ba4a44f5e7
|
@ -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)
|
||||
|
|
Loading…
Reference in New Issue