tableau / TabPy

                        help="Address or hostname to connect to")

def main():

    parser = argparse.ArgumentParser()

    subcommands = parser.add_subparsers()

    cmd_list = subcommands.add_parser('list')

    cmd_list.set_defaults(action='list')

    _add_common_arguments(cmd_list)

    cmd_list.add_argument('-l', '--list', action='store_true',

                          help="Print more details.")

    cmd_do = subcommands.add_parser('do')

    cmd_do.set_defaults(action='do')

    _add_common_arguments(cmd_do)

    cmd_do.add_argument('action_type', type=str,

                        help="The action type to run.")

    cmd_put = subcommands.add_parser('put')

    cmd_put.set_defaults(action='put')

    _add_common_arguments(cmd_put)

    cmd_put.add_argument('file', type=str,

                         help="CSV file to upload.")

    cmd_get = subcommands.add_parser('get')

    cmd_get.set_defaults(action='get')

    _add_common_arguments(cmd_get)

    cmd_get_descriptor = cmd_get.add_mutually_exclusive_group(required=True)

    cmd_get_descriptor.add_argument('-p', '--path', type=str, action='append',

                                    help="The path for the descriptor.")

    cmd_get_descriptor.add_argument('-c', '--command', type=str,

                                    help="The command for the descriptor.")

    args = parser.parse_args()

    if not hasattr(args, 'action'):

        parser.print_help()

        sys.exit(1)

    commands = {

        'list': list_flights,

        'do': do_action,

        'get': get_flight_by_path,

        'put': push_data,

    }

    host, port = args.host.split(':')

    port = int(port)

    scheme = "grpc+tcp"

    connection_args = {}

    if args.tls:

        scheme = "grpc+tls"

        if args.tls_roots:

            with open(args.tls_roots, "rb") as root_certs:

                connection_args["tls_root_certs"] = root_certs.read()

    if args.mtls:

        with open(args.mtls[0], "rb") as cert_file:

            tls_cert_chain = cert_file.read()

        with open(args.mtls[1], "rb") as key_file:

            tls_private_key = key_file.read()

        connection_args["cert_chain"] = tls_cert_chain

        connection_args["private_key"] = tls_private_key

    client = pyarrow.flight.FlightClient(f"{scheme}://{host}:{port}",

                                         **connection_args)

    while True:

        try:

            action = pyarrow.flight.Action("healthcheck", b"")

            options = pyarrow.flight.FlightCallOptions(timeout=1)

            list(client.do_action(action, options=options))

            break

        except pyarrow.ArrowIOError as e:

            if "Deadline" in str(e):

                print("Server is not ready, waiting...")

    commands[args.action](args, client, connection_args)

                        help="Address or hostname to connect to")

def main():

    parser = argparse.ArgumentParser()

    subcommands = parser.add_subparsers()

    cmd_list = subcommands.add_parser('list')

    cmd_list.set_defaults(action='list')

    _add_common_arguments(cmd_list)

    cmd_list.add_argument('-l', '--list', action='store_true',

                          help="Print more details.")

    cmd_do = subcommands.add_parser('do')

    cmd_do.set_defaults(action='do')

    _add_common_arguments(cmd_do)

    cmd_do.add_argument('action_type', type=str,

                        help="The action type to run.")

    cmd_put = subcommands.add_parser('put')

    cmd_put.set_defaults(action='put')

    _add_common_arguments(cmd_put)

    cmd_put.add_argument('file', type=str,

                         help="CSV file to upload.")

    cmd_get = subcommands.add_parser('get')

    cmd_get.set_defaults(action='get')

    _add_common_arguments(cmd_get)

    cmd_get_descriptor = cmd_get.add_mutually_exclusive_group(required=True)

    cmd_get_descriptor.add_argument('-p', '--path', type=str, action='append',

                                    help="The path for the descriptor.")

    cmd_get_descriptor.add_argument('-c', '--command', type=str,

                                    help="The command for the descriptor.")

    args = parser.parse_args()

    if not hasattr(args, 'action'):

        parser.print_help()

        sys.exit(1)

    commands = {

        'list': list_flights,

        'do': do_action,

        'get': get_flight_by_path,

        'put': push_data,

    }

    host, port = args.host.split(':')

    port = int(port)

    scheme = "grpc+tcp"

    connection_args = {}

    if args.tls:

        scheme = "grpc+tls"

        if args.tls_roots:

            with open(args.tls_roots, "rb") as root_certs:

                connection_args["tls_root_certs"] = root_certs.read()

    if args.mtls:

        with open(args.mtls[0], "rb") as cert_file:

            tls_cert_chain = cert_file.read()

        with open(args.mtls[1], "rb") as key_file:

            tls_private_key = key_file.read()

        connection_args["cert_chain"] = tls_cert_chain

        connection_args["private_key"] = tls_private_key

    client = pyarrow.flight.FlightClient(f"{scheme}://{host}:{port}",

                                         **connection_args)

    while True:

        try:

            action = pyarrow.flight.Action("healthcheck", b"")

            options = pyarrow.flight.FlightCallOptions(timeout=1)

            list(client.do_action(action, options=options))

            break

        except pyarrow.ArrowIOError as e:

            if "Deadline" in str(e):

                print("Server is not ready, waiting...")

    commands[args.action](args, client, connection_args)