Added esxdatacenter.datacenter_configured state

salt/states/esxdatacenter.py

@@ -0,0 +1,134 @@
+# -*- coding: utf-8 -*-
+'''
+Salt states to create and manage VMware vSphere datacenters (datacenters).
+
+:codeauthor: :email:`Alexandru Bleotu <alexandru.bleotu@morganstaley.com>`
+
+Dependencies
+============
+
+- pyVmomi Python Module
+
+States
+======
+
+datacenter_configured
+---------------------
+
+Makes sure a datacenter exists and is correctly configured.
+
+If the state is run by an ``esxdatacenter`` minion, the name of the datacenter
+is retrieved from the proxy details, otherwise the datacenter has the same name
+as the state.
+
+Supported proxies: esxdatacenter
+
+
+Example:
+
+1. Make sure that a datacenter named ``target_dc`` exists on the vCenter, using a
+``esxdatacenter`` proxy:
+
+Proxy minion configuration (connects passthrough to the vCenter):
+
+.. code-block:: yaml
+    proxy:
+      proxytype: esxdatacenter
+      datacenter: target_dc
+      vcenter: vcenter.fake.com
+      mechanism: sspi
+      domain: fake.com
+      principal: host
+
+State configuration:
+
+.. code-block:: yaml
+
+    datacenter_state:
+      esxdatacenter.datacenter_configured
+'''
+
+# Import Python Libs
+from __future__ import absolute_import
+import logging
+
+# Import Salt Libs
+import salt.exceptions
+
+# Get Logging Started
+log = logging.getLogger(__name__)
+LOGIN_DETAILS = {}
+
+
+def __virtual__():
+    return 'esxdatacenter'
+
+
+def mod_init(low):
+    return True
+
+
+def datacenter_configured(name):
+    '''
+    Makes sure a datacenter exists.
+
+    If the state is run by an ``esxdatacenter`` minion, the name of the
+    datacenter is retrieved from the proxy details, otherwise the datacenter
+    has the same name as the state.
+
+    Supported proxies: esxdatacenter
+
+    name:
+        Datacenter name. Ignored if the proxytype is ``esxdatacenter``.
+    '''
+    proxy_type = __salt__['vsphere.get_proxy_type']()
+    if proxy_type == 'esxdatacenter':
+        dc_name = __salt__['esxdatacenter.get_details']()['datacenter']
+    else:
+        dc_name = name
+    log.info('Running datacenter_configured for datacenter \'{0}\''
+             ''.format(dc_name))
+    ret = {'name': name, 'changes': {}, 'pchanges': {},
+           'result': None, 'comment': 'Default'}
+    comments = []
+    changes = {}
+    pchanges = {}
+    si = None
+    try:
+        si = __salt__['vsphere.get_service_instance_via_proxy']()
+        dcs = __salt__['vsphere.list_datacenters_via_proxy'](
+            datacenter_names=[dc_name], service_instance=si)
+        if not dcs:
+            if __opts__['test']:
+                comments.append('State will create '
+                                'datacenter \'{0}\'.'.format(dc_name))
+                log.info(comments[-1])
+                pchanges.update({'new': {'name': dc_name}})
+            else:
+                log.debug ('Creating datacenter \'{0}\'. '.format(dc_name))
+                __salt__['vsphere.create_datacenter'](dc_name, si)
+                comments.append('Created datacenter \'{0}\'.'.format(dc_name))
+                log.info(comments[-1])
+                changes.update({'new': {'name': dc_name}})
+        else:
+            comments.append('Datacenter \'{0}\' already exists. Nothing to be '
+                            'done.'.format(dc_name))
+            log.info(comments[-1])
+        __salt__['vsphere.disconnect'](si)
+        if __opts__['test'] and pchanges:
+            ret_status = None
+        else:
+            ret_status = True
+        ret.update({'result': ret_status,
+                    'comment': '\n'.join(comments),
+                    'changes': changes,
+                    'pchanges': pchanges})
+        return ret
+    except salt.exceptions.CommandExecutionError as exc:
+        log.error('Error: {}'.format(exc))
+        if si:
+            __salt__['vsphere.disconnect'](si)
+        ret.update({
+            'result': False if not __opts__['test'] else None,
+            'comment': str(exc)})
+        return ret