3 from opnfv.deployment import factory
7 parser = argparse.ArgumentParser(description='OPNFV POD Info Generator')
9 parser.add_argument("-t", "--INSTALLER_TYPE", help="Give INSTALLER_TYPE")
10 parser.add_argument("-i", "--INSTALLER_IP", help="Give INSTALLER_IP")
11 parser.add_argument("-u", "--user", help="Give username of this pod")
12 parser.add_argument("-k", "--key", help="Give key file of the user")
13 parser.add_argument("-p", "--password", help="Give password of the user")
14 parser.add_argument("-f", "--filepath", help="Give dest path of output file")
15 args = parser.parse_args()
20 Check all the CLI inputs. Must give INSTALLER_TYPE, INSTALLER_IP, user
21 and filepath of the output file.
22 Need to give key or password.
24 if not args.INSTALLER_TYPE or not args.INSTALLER_IP or not args.user:
25 print("INSTALLER_TYPE, INSTALLER_IP and user are all needed.")
27 if not args.key and not args.password:
28 print("key and password are all None. At least providing one.")
31 print("Must give the dest path of the output file.")
38 Get handler of the nodes info with key file.
40 return factory.Factory.get_handler(args.INSTALLER_TYPE, args.INSTALLER_IP,
41 args.user, pkey_file=args.key)
44 def get_with_passwd():
46 Get handler of the nodes info with password.
48 return factory.Factory.get_handler(args.INSTALLER_TYPE, args.INSTALLER_IP,
49 args.user, installer_pwd=args.password)
52 def create_file(handler, INSTALLER_TYPE):
54 Create the yaml file of nodes info.
55 As Yardstick required, node name must be node1, node2, ... and node1 must
57 Compass uses password of each node.
58 Other installers use key file of each node.
60 if not os.path.exists(os.path.dirname(args.filepath)):
61 os.makedirs(os.path.dirname(args.filepath))
66 if INSTALLER_TYPE == 'apex':
70 if node.roles[0].lower() == "controller":
71 node_info = {'name': "node%s" % index, 'role': node.roles[0],
72 'ip': node.ip, 'user': user}
73 node_list.append(node_info)
76 node_info = {'name': node.name, 'role': 'unknown', 'ip': node.ip,
78 node_list.append(node_info)
81 if node.roles[0].lower() == "compute":
82 node_info = {'name': "node%s" % index, 'role': node.roles[0],
83 'ip': node.ip, 'user': user}
84 node_list.append(node_info)
87 node_info = {'name': node.name, 'role': 'unknown', 'ip': node.ip,
89 node_list.append(node_info)
90 if args.INSTALLER_TYPE == 'compass':
91 for item in node_list:
92 item['password'] = 'root'
94 for item in node_list:
95 item['key_filename'] = '/root/.ssh/id_rsa'
96 data = {'nodes': node_list}
97 with open(args.filepath, "w") as fw:
102 if not check_params():
105 handler = get_with_key()
107 handler = get_with_passwd()
109 print("Error: failed to get the node's handler.")
111 create_file(handler, args.INSTALLER_TYPE)
114 if __name__ == '__main__':