vsan-find-thick-vmswap.py

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

"""
vSAN 6.6 Management SDK sample to find
VM with "Thick" provisioned swap files

Usage:
    python vsan-find-thick-vmswap.py -s 192.168.1.100 \
        -u 'administrator@vsphere.local' -p 'VMware1!' \
        -c VSAN-Cluster
"""

__author__ = 'VMware, Inc'
from pyVmomi import vim
from pyVim.connect import SmartConnect, Disconnect
import atexit
import argparse
import getpass
import json
import random
import sys
import ssl
# import the VSAN API python bindings
import vsanmgmtObjects
import vsanapiutils


def GetArgs():

    """
    Supports the command-line arguments listed below.
    """

    parser = argparse.ArgumentParser(
        description='Process args for VSAN SDK sample application')
    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('-c', '--cluster', dest='clusterName', required=True,
                        action='store')
    args = parser.parse_args()
    return args


def get_obj(content, vim_type, name=None):
    obj = None
    container = content.viewManager.CreateContainerView(
        content.rootFolder, vim_type, True)
    if name:
        for c in container.view:
            if c.name == name:
                obj = c
                return obj
    else:
        return container.view


# Start program
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))

    # For python 2.7.9 and later, the defaul SSL conext has more strict
    # connection handshaking rule. We may need turn of the hostname checking
    # and client side cert verification
    context = None
    if sys.version_info[:3] > (2, 7, 8):
        context = ssl.create_default_context()
        context.check_hostname = False
        context.verify_mode = ssl.CERT_NONE

    si = SmartConnect(host=args.host,
                      user=args.user,
                      pwd=password,
                      port=int(args.port),
                      sslContext=context)

    atexit.register(Disconnect, si)

    # for detecting whether the host is VC or ESXi
    aboutInfo = si.content.about

    if aboutInfo.apiType == 'VirtualCenter':
        majorApiVersion = aboutInfo.apiVersion.split('.')[0]
        if int(majorApiVersion) < 6:
            print('The Virtual Center with version %s (lower than 6.0) \
                   is not supported.' % aboutInfo.apiVersion)
            return -1

        vcMos = vsanapiutils.GetVsanVcMos(si._stub, context=context)
        vcos = vcMos['vsan-cluster-object-system']

        cluster = get_obj(si.content,
                          [vim.ClusterComputeResource], args.clusterName)

        # Retreive random ESXi host within vSAN Cluster
        # which will be used to get details about the vswap object
        randomHost = random.choice(cluster.host)
        vsanIntSys = randomHost.configManager.vsanInternalSystem

        # Retrieve all VMs within the given vSAN Cluster
        vms = cluster.resourcePool.vm

        # Create a map of VM Name to MoRef ID which is returned by vSAN API
        vmMoRefIdMapping = {}
        for vm in vms:
            vmMoRefIdMapping[vm] = vm.name

        if cluster is None:
            print("Cluster %s is not found for %s" % (args.clusterName,
                  args.host))
            return -1

        results = vcos.VsanQueryObjectIdentities(cluster=cluster,
                                                 objUuids=None,
                                                 objTypes=["vmswap"],
                                                 includeHealth=False,
                                                 includeObjIdentity=True,
                                                 includeSpaceSummary=False)
        for result in results.identities:
            vsanuuid = result.uuid
            vmMoref = result.vm
            vmName = vmMoRefIdMapping[vmMoref]
            jsonResult = json.loads(vsanIntSys.GetVsanObjExtAttrs([vsanuuid]))
            for j in jsonResult:
                allocation_type = jsonResult[j]['Allocation type']
                if allocation_type == "Zeroed thick":
                    print(vmName)


# Start program
if __name__ == "__main__":
    main()