vmtree.py

#!/bin/usr/python
"""

Prints VMWare Datacenter as tree
author: samuelsh (c)
date: 2017
"""
import argparse
import atexit
import getpass
import traceback

import sys
from pyVim.connect import SmartConnect, Disconnect

from logger import Logger
from vmutils import VmUtils


def get_args():
    """
    Supports the command-line arguments listed below.
    """

    parser = argparse.ArgumentParser(
        description='Process args for retrieving all the Virtual Machines')
    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', '--view', choices=['vms', 'hosts'],
                        help="Preferred view: VMs and Templates/Hosts and Clusters", required=True)
    parser.add_argument('-d', '--folders_only', help="Print folders only", action="store_true")
    args = parser.parse_args()
    return args


def main():
    """
   """
    args = get_args()
    logger = Logger().logger
    logger.debug("Logger Initialized %s" % logger)

    if args.password:
        password = args.password
    else:
        password = getpass.getpass(prompt='Enter password for host %s and '
                                          'user %s: ' % (args.host, args.user))

    si = SmartConnect(host=args.host,
                      user=args.user,
                      pwd=password)
    if not si:
        logger.error("Could not connect to the specified host using specified "
                     "username and password")
        return -1

    atexit.register(Disconnect, si)
    VmUtils.print_vm_tree_vms(args, si)

    return 0


# Start program
if __name__ == "__main__":
    try:
        main()
    except Exception as e:
        traceback.print_exc()
        sys.exit(1)