diff --git a/sample/getallvms.py b/sample/getallvms.py index 21afcba..5a15a4c 100644 --- a/sample/getallvms.py +++ b/sample/getallvms.py @@ -24,6 +24,7 @@ from pyVmomi import vmodl import argparse import atexit +import getpass import sys @@ -35,7 +36,7 @@ def GetArgs(): parser.add_argument('-s', '--host', required=True, action='store', help='Remote host to connect to') parser.add_argument('-o', '--port', type=int, default=443, action='store', help='Port to connect on') parser.add_argument('-u', '--user', required=True, action='store', help='User name to use when connecting to host') - parser.add_argument('-p', '--password', required=True, action='store', help='Password to use when connecting to host') + parser.add_argument('-p', '--password', required=False, action='store', help='Password to use when connecting to host') args = parser.parse_args() return args @@ -77,12 +78,17 @@ def main(): """ args = GetArgs() + if args.password: + password = args.password + else: + password = getpass.getpass(prompt='Enter password for host %s and user %s: ' % (args.host,args.user)) + try: si = None try: si = SmartConnect(host=args.host, user=args.user, - pwd=args.password, + pwd=password, port=int(args.port)) except IOError, e: pass diff --git a/sample/poweronvm.py b/sample/poweronvm.py index 84de7c7..4962a98 100755 --- a/sample/poweronvm.py +++ b/sample/poweronvm.py @@ -24,6 +24,7 @@ from pyVmomi import vim, vmodl import argparse import atexit +import getpass import sys def GetArgs(): @@ -35,7 +36,7 @@ def GetArgs(): parser.add_argument('-s', '--host', required=True, action='store', help='Remote host to connect to') parser.add_argument('-o', '--port', type=int, default=443, action='store', help='Port to connect on') parser.add_argument('-u', '--user', required=True, action='store', help='User name to use when connecting to host') - parser.add_argument('-p', '--password', required=True, action='store', help='Password to use when connecting to host') + parser.add_argument('-p', '--password', required=False, action='store', help='Password to use when connecting to host') parser.add_argument('-v', '--vmname', required=True, action='append', help='Names of the Virtual Machines to power on') args = parser.parse_args() return args @@ -98,6 +99,11 @@ def main(): """ args = GetArgs() + if args.password: + password = args.password + else: + password = getpass.getpass(prompt='Enter password for host %s and user %s: ' % (args.host,args.user)) + try: vmnames = args.vmname if not len(vmnames): @@ -108,7 +114,7 @@ def main(): try: si = SmartConnect(host=args.host, user=args.user, - pwd=args.password, + pwd=password, port=int(args.port)) except IOError, e: pass