Function main

python/seldon/cli/cli_main.py:332–377

()

        gdata["zk_client"].stop()

def main():
    logging.basicConfig()
    check_conf()
    expand_conf()
    opts = getOpts()
    if opts.setup_config:
        create_default_conf()
        sys.exit(0)
    if opts.print_default_config:
        print dict_to_json(json_to_dict(get_default_conf()), True)
        sys.exit(0)

    # is the conf still not setup
    if gdata['conf_data'] == None:
        print "No config found, run with '--setup-config' to set it up."
        sys.exit(1)

    if opts.version == True:
        from seldon import __version__
        print __version__
        sys.exit(0)

    if len(opts.args) < 1:
        print "Need command"
        sys.exit(1)

    if opts.zk_hosts != None:
        gdata["conf_data"]["zk_hosts"] = opts.zk_hosts
    else:
        opts.zk_hosts = gdata["conf_data"]["zk_hosts"]

    cmd = opts.args[0]
    command_args = opts.args[1:]

    cmds = gdata['cmds']
    if cmds.has_key(cmd):
        start_zk_client(opts)
        command_data = {
                'conf_data' : gdata['conf_data'],
                'zkdetails' : {'zkroot': gdata['conf_data']['zkroot'], 'zk_client': gdata['zk_client']},
        }
        cmds[cmd](opts,command_data, command_args)
        stop_zk_client()
    else:
        print "Invalid command[{}]".format(cmd)
        sys.exit(1)

if __name__ == '__main__':
    main()