From 59e6fa806302629dcbdbeb75d9edc041934ffd7a Mon Sep 17 00:00:00 2001 From: Grant Orchard Date: Sat, 11 May 2019 10:05:32 +1000 Subject: [PATCH] Initial commit --- examples/get_cluster_ips.py | 31 +++++++++++++++++++++++++++++++ 1 file changed, 31 insertions(+) create mode 100644 examples/get_cluster_ips.py diff --git a/examples/get_cluster_ips.py b/examples/get_cluster_ips.py new file mode 100644 index 0000000..cc413be --- /dev/null +++ b/examples/get_cluster_ips.py @@ -0,0 +1,31 @@ +import argparse +import os +from caspyr import Session, Machine + + +def getargs(): + parser = argparse.ArgumentParser() + parser.add_argument('-t', '--token', + required=True, + action='store' + ) + parser.add_argument('-n', '--cluster_name', + required=True, + help='The value of the tag assigned to the machine' + ) + args = parser.parse_args() + return args + + +def get_ips(token, cluster_name): + s = Session.login(token) + ips = [Machine.get_ip(s, master['id']) for master in Machine.find_by_tag(s, key=cluster_name, value='master')] + print(ips) + + +def main(): + args = getargs() + get_ips(token=args.token, cluster_name=args.cluster_name) + +if __name__ == '__main__': + main()