Get-VC-Solutions-Functions.py

#!/usr/bin/python

import sys

# Site Package Location in VXRM before import module section
sys.path.append("/usr/lib/vmware-marvin/marvind/webapps/ROOT/WEB-INF/classes/scripts/lib/python2.7/site-packages")

import ssl
from pyVmomi import vim, vmodl
from pyVim import connect
from datetime import datetime
import argparse
import getpass


def GetArgs():

   parser = argparse.ArgumentParser(description='Process args for connecting to vCenter')
   parser.add_argument('-v', '--vc', required=True, action='store', help='vCenter')
   parser.add_argument('-u', '--user', required=True, action='store', help='vCenter Administrator')
   parser.add_argument('-p', '--password', required=False, action='store', help='Password')
   args = parser.parse_args()

   return args


# Function to create the View for the extensions

def get_obj(si):

    extmanager = si.content.extensionManager
    extview = si.content.viewManager.CreateListView([extmanager])
    view = extview.view
    return view


#Function to create the Filter Spec

def create_filter_spec(si,pc):
    objSpecs = []
    extmanager = si.content.extensionManager
    extview = si.content.viewManager.CreateListView([extmanager])
    extension = extview
    print(extview)

    for ext in extview.view:
        print("ext.extensionList")
        print(ext.extensionList)
        objSpec = vmodl.query.PropertyCollector.ObjectSpec(obj=ext)
        objSpecs.append(objSpec)
        print("LALALA")
        print(objSpecs)
        print("LALALA")
    print(objSpecs)
    filterSpec = vmodl.query.PropertyCollector.FilterSpec()
    filterSpec.objectSet = objSpecs
    propSet = vmodl.query.PropertyCollector.PropertySpec(all=False)
    propSet.type = vim.Extension
    propSet.pathSet = "company"
    filterSpec.propSet = [propSet]
    return filterSpec


def filter_results(result):
    vms = []
    for o in result.objects:
        if o.propSet[0].val == "company":
            vms.append(o.obj)
    return vms


def main():

# global variables of main

    args = GetArgs()

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

    try:

        s = ssl.SSLContext(ssl.PROTOCOL_TLSv1)
        s.verify_mode = ssl.CERT_NONE
        # connection string
        si = connect.SmartConnect(host=args.vc,
                                  user=args.user,
                                  pwd=password,
                                  sslContext=s)

        pc = si.content.propertyCollector
        filter_spec = create_filter_spec(si, pc)
        options = vmodl.query.PropertyCollector.RetrieveOptions()
        result = pc.RetrievePropertiesEx([filter_spec], options)
        vms = filter_results(result)
        print(vms)
        print("VMs")
        for vm in vms:
            print(vm)

#       testing VC connection
        vctime = si.CurrentTime()
        print("vCenter Time: ", vctime.strftime("%Y-%m-%d %H:%M"))


        extensions = get_obj(si).view
        print(extensions)
        # 'vim.view.ListView:session[52d4057e-9e24-8619-7ec6-df8fff7948ed]5297b178-4d73-0423-f82c-504fe257616b'

        # this prints me all the extensionList attributes of the view
        for i in extensions:
            print(i.extensionList)

        print("testing filter")
        test = create_filter_spec(si,pc)
        print(test)




    except vmodl.MethodFault as error:
        print("Caught vmodl fault : " + error.msg)
        return -1

    return 0


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