destroy_vms.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

"""Destroy VMs and Folders in a vSphere environment.

Usage:
    destroy_vms.py
    destroy_vms.py -f FILE
    destroy_vms.py -f FILE --folder FOLDER
    destroy_vms.py --version
    destroy_vms.py (-h | --help)

Options:
    -h, --help          Prints this page
    --version           Prints current version
    -f, --file FILE     Name of JSON file with server connection and login information
    --folder FOLDER     Name of a folder to recursively destroy

"""

from docopt import docopt

from vsphere import vSphere
from vsphere_utils import *
from vm_utils import *
from utils import *

__version__ = "0.1.0"

args = docopt(__doc__, version=__version__, help=True)

# TODO: use six to allow python 2.7 to be used?
server = None  # Suppress warnings (sigh)
if args["--file"]:
    from json import load

    with open(args["--file"], "r") as login_file:
        info = load(fp=login_file)

    server = vSphere(datacenter=info["datacenter"], username=info["username"], password=info["password"],
                     hostname=info["hostname"], port=info["port"], datastore=info["datastore"])
else:
    from getpass import getpass
    print("Enter information to connect to vSphere environment")
    host = input("Hostname: ")
    port = int(input("Port: "))
    user = input("Username: ")
    pswd = getpass("Password: ")
    datacenter = input("vSphere Datacenter (should be one level below the vCenter server instance "
                       "in the VMs and Templates view): ")
    if prompt_y_n_question("Would you like to specify the datastore used "):
        datastore = input("vSphere Datastore: ")
        server = vSphere(datacenter=datacenter, username=user, password=pswd,
                         hostname=host, port=port, datastore=datastore)
    else:
        server = vSphere(datacenter=datacenter, username=user, password=pswd, hostname=host, port=port)

if prompt_y_n_question("Do you wish to destroy a folder and it's sub-trees? "):
    while True:
        folder_name = input("Name of folder to destroy: ")
        folder = server.get_folder(folder_name=folder_name)  # find folder
        if folder:
            break
        else:
            print("Couldn't find a folder with name {}. Perhaps try another? ".format(folder_name))

    print("Recursively destroying folder {}".format(folder.name))
    destroy_everything(folder)

elif prompt_y_n_question("Do you wish to destroy all VMs in a folder? "):
    while True:
        folder_name = input("Name of folder: ")
        folder = server.get_folder(folder_name=folder_name)  # find folder
        if folder:
            break
        else:
            print("Couldn't find a folder with name {}. Perhaps try another? ".format(folder_name))

    print("Destroying all VMs in folder {}".format(folder_name))
    for vm in folder.childEntity:
        destroy_vm(vm)

else:
    while True:
        vm_name = input("Name of VM to destroy: ")
        vm = server.get_vm(vm_name=vm_name)  # find folder
        if vm:
            break
        else:
            print("Couldn't find a VM with name {}. Perhaps try another? ".format(vm_name))

    print("Destroying VM with name {}".format(vm_name))
    destroy_vm(vm)