Last active
August 3, 2016 13:03
-
-
Save gdelgado/54cd8f5462554a254665ab80ffae663f to your computer and use it in GitHub Desktop.
pyvmomi script graceful OS shutdown.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
#!/usr/bin/env python | |
""" | |
vSphere Python SDK program for shutting down VMs | |
""" | |
from __future__ import print_function | |
from pyVim.connect import SmartConnect, Disconnect | |
from pyVmomi import vim, vmodl | |
import argparse | |
import atexit | |
import getpass | |
import sys | |
import ssl | |
def GetArgs(): | |
""" | |
Supports the command-line arguments listed below. | |
""" | |
parser = argparse.ArgumentParser(description='Process args for shutting down a Virtual Machine') | |
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=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 shutdown') | |
args = parser.parse_args() | |
return args | |
def main(): | |
""" | |
Simple command-line program for shutting down virtual machines on a system. | |
""" | |
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)) | |
service_instance = None | |
context = ssl.SSLContext(ssl.PROTOCOL_TLSv1) | |
context.verify_mode = ssl.CERT_NONE | |
try: | |
service_instance = SmartConnect(host=args.host, | |
user=args.user, | |
pwd=args.password, | |
port=int(args.port), | |
sslContext=context) | |
if not service_instance: | |
print("Could not connect to the specified host using specified " | |
"username and password") | |
return -1 | |
atexit.register(Disconnect, service_instance) | |
content = service_instance.RetrieveContent() | |
# Search for all VMs | |
objview = content.viewManager.CreateContainerView(content.rootFolder, | |
[vim.VirtualMachine], | |
True) | |
vmList = objview.view | |
objview.Destroy() | |
for vm in vmList: | |
print("Shutting down VM: %s" % vm.name) | |
vm.ShutdownGuest() | |
except vmodl.MethodFault as error: | |
print("Caught vmodl fault : " + e.msg) | |
return -1 | |
return 0 | |
# Start program | |
if __name__ == "__main__": | |
main() |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment