Esempio n. 1
0
def instance_mode(client: MasterApiClient, args):
    if args.action == 'info':
        print(json.dumps(client.read_instance(args.id).to_json(), indent=2, ensure_ascii=False))
    elif args.action == 'ctrl':
        answer = None
        if args.target_state == 'start':
            answer = client.start_instance(args.id)
        elif args.target_state == 'stop':
            answer = client.stop_instance(args.id)
        else:
            logging.exception('Unsupported target state: {}'.format(args.target_state))
            exit(1)
        print('State for instance {} changed from {} to {}.'.format(args.id, answer[0].name, answer[1].name))
    elif args.action == 'logs':
        print(client.instance_logs(args.id, args.task_name, args.host, args.log_type), end='')
    else:
        logging.error('Not supported action for instance mode: %s', args.action)
        exit(1)
Esempio n. 2
0
def graph_mode(client: MasterApiClient, args):
    if args.action == 'create':
        name, rev = client.create_graph(
            graph_struct=_prepare_graph_struct(args.name, args.graph, args.hosts, args.format),
            graph_name=args.name
        )
        print('Created graph "{}", revision {}.'.format(name, rev))
    elif args.action == 'info':
        if args.list_all:
            print(json.dumps([_.to_json() for _ in client.list_graphs(args.name, args.revision)],
                             indent=2, ensure_ascii=False))
        else:
            assert args.name, 'Graph name should be set'
            print(json.dumps(client.read_graph(args.name, args.revision).to_json(), indent=2, ensure_ascii=False))
    elif args.action == 'launch':
        print('Created new graph instance: {}'.format(client.launch_graph(args.name, args.revision)))
    else:
        logging.error('Not supported action for graph mode: %s', args.action)
        exit(1)
Esempio n. 3
0
def main(args):
    cli = MasterApiClient(master_host=args.server, master_port=args.port)
    graph_name, graph_rev = cli.create_graph(graph_struct=json.load(open(args.path)),
                                             graph_name='test_graph')
    print('Created graph: {}, rev: {}'.format(graph_name, graph_rev))
    instance_id = cli.launch_graph(graph_name, int(graph_rev))
    print('Created instance: {}'.format(instance_id))
    old_state, new_state = cli.start_instance(instance_id)
    print('State: {} -> {}'.format(old_state.name, new_state.name))
    while True:
        sleep(1)
        data = cli.read_instance(instance_id)
        state = data.exec_stats.state
        print('#'*10)
        print('State:', state.name)
        for task_name, task_info in data.exec_stats.per_task_execution_info.items():
            print('Task', task_name)
            print('\n'.join('\tHost {}: {}'.format(host, info.state.name)
                            for host, info in task_info.per_host_info.items()))
        if state.is_terminal:
            print('*' * 20)
            if state.name == GraphInstanceState.failed:
                print('Failed!')
                if data.exec_stats.fail_msg:
                    print('Fail msg: {}'.format(data.exec_stats.fail_msg))
            elif state.name == GraphInstanceState.finished:
                print('Success!')
            print('LOGS:')
            for task_name, task_info in data.exec_stats.per_task_execution_info.items():
                print(' Task', task_name)
                for host in task_info.per_host_info.keys():
                    out = cli.instance_logs(instance_id, task_name, host, 'out')
                    err = cli.instance_logs(instance_id, task_name, host, 'err')
                    print('  Host', host)
                    if out:
                        print('   Out:')
                        print(''.join('+++>{}'.format(_) for _ in out.strip().splitlines(keepends=True)))
                    if err:
                        print('   Err:')
                        print(''.join('--->{}'.format(_) for _ in err.strip().splitlines(keepends=True)))
                print('-' * 10)
            break