cluster maangement routines

src/cluster.py

@@ -0,0 +1,152 @@
+from pyVmomi import vim
+from tabulate import tabulate
+from network import EsxNetwork, net_print_details
+from datastore import EsxDataStore, ds_print_details
+from host import EsxHost, host_print_details
+from misc import esx_objects, esx_name, sizeof_fmt
+
+###########
+# HELPERS #
+###########
+
+def cluster_get(service, name):
+    cls = esx_objects(service, vim.ClusterComputeResource)
+    for c in cls:
+        # try to lookup by key first
+        ckey = esx_name(c)
+        if ckey == name:
+            return EsxCluster(service, c)
+
+        # fallback to name lookup
+        cname = cluster.name
+        if cname == name:
+            return EsxCluster(service, c)
+
+    return None
+
+def cluster_get_all(service):
+    l = []
+    cls = esx_objects(service, vim.ClusterComputeResource)
+    for cl in cls:
+        c = EsxCluster(service, cl)
+        l.append(c)
+    return l
+
+def cluster_net(s, opt):
+    cluster = cluster_get(s, opt['<name>'])
+    if not cluster:
+        return
+
+    networks = cluster.net()
+    net_print_details(networks)
+
+def cluster_ds(s, opt):
+    cluster = cluster_get(s, opt['<name>'])
+    if not cluster:
+        return
+
+    datastores = cluster.ds()
+    ds_print_details(datastores)
+
+def cluster_host(s, opt):
+    cluster = cluster_get(s, opt['<name>'])
+    if not cluster:
+        return
+
+    hosts = cluster.host()
+    host_print_details(hosts)
+
+def cluster_list(s, opt):
+    pool = EsxClusterPool(s)
+    headers = [ "Key", "Name", "Status", "Hosts", "Cores", "Threads", "Memory" ]
+    tabs = []
+    for cl in pool.clusters:
+        info = cl.info()
+        vals = [ cl.key, cl.name, cl.status,
+                 info.hosts, info.cores, info.threads, sizeof_fmt(info.mem) ]
+        tabs.append(vals)
+
+    print tabulate(tabs, headers)
+
+def cluster_parser(service, opt):
+    if   opt['list']  == True: cluster_list(service, opt)
+    elif opt['net']   == True: cluster_net(service, opt)
+    elif opt['ds']    == True: cluster_ds(service, opt)
+    elif opt['hosts']  == True: cluster_host(service, opt)
+
+###########
+# CLASSES #
+###########
+
+class EsxClusterInfo:
+    def __init__(self, c):
+        summary = c.summary
+        self.name = esx_name(c)
+        self.cores = summary.numCpuCores
+        self.threads = summary.numCpuThreads
+        self.mem = summary.totalMemory
+        self.hosts = summary.numHosts
+
+class EsxCluster:
+    def __init__(self, service, cluster):
+        self.service = service
+        self.cluster = cluster
+        self.key = esx_name(cluster)
+        self.name = cluster.name
+        self.status = cluster.overallStatus
+
+    def info(self):
+        return EsxClusterInfo(self.cluster)
+
+    # def pool(self):
+    #     l = []
+    #     pools = self.cluster.resourcePool
+    #     for p in pools:
+    #         pl = EsxResourcePool(p)
+    #         l.append(pl)
+    #     return l
+
+    def net(self):
+        l = []
+        networks = self.cluster.network
+        for net in networks:
+            n = EsxNetwork(net)
+            l.append(n)
+        return l
+
+    def ds(self):
+        l = []
+        for s in self.cluster.datastore:
+            ds = EsxDataStore(self.service, s)
+            l.append(ds)
+        return l
+
+    def host(self):
+        l = []
+        for h in self.cluster.host:
+            hs = EsxHost(self.service, h)
+            l.append(hs)
+        return l
+
+    def __str__(self):
+        return self.name
+
+class EsxClusterPool:
+    def __init__(self, service):
+        self.clusters = cluster_get_all(service)
+
+    def list(self):
+        return self.clusters
+
+    def get(self, name):
+        for c in self.clusters:
+            if c.name == name:
+                return c
+        return None
+
+    def __str__(self):
+        r  = "ESXi Clusters:\n"
+        for c in self.clusters:
+            r += str(c)
+        r += "\n"
+        return r

vsphere

@@ -4,6 +4,10 @@
 Interactive CLI for VMware vSphere orchestration.
 
 Usage:
+    vsphere cluster ds <name>
+    vsphere cluster hosts <name>
+    vsphere cluster list
+    vsphere cluster net <name>
     vsphere datastore browse <name> <path>
     vsphere datastore download <name> <path>
     vsphere datastore list
@@ -38,6 +42,7 @@ from pyVim import connect
 from pyVmomi import vmodl, vim
 import atexit
 from config import EsxConfig
+from cluster import cluster_parser
 from dc import dc_parser
 from host import host_parser
 from vm import vm_parser
@@ -51,6 +56,7 @@ def opt_parser(opt):
     elif opt['template']  == True: template_parser(service, opt)
     elif opt['host']      == True: host_parser(service, opt)
     elif opt['dc']        == True: dc_parser(service, opt)
+    elif opt['cluster']   == True: cluster_parser(service, opt)
     elif opt['datastore'] == True: datastore_parser(service, opt)
 
 ########