From: Frank Brehm <frank.brehm@pixelpark.com>
Date: Tue, 25 Aug 2020 16:03:33 +0000 (+0200)
Subject: Start refactoring lib/cr_vmware_tpl/handler.py for using datastore clusters.
X-Git-Tag: 2.2.1~1^2~1^2~7^2~4
X-Git-Url: https://git.uhu-banane.org/?a=commitdiff_plain;h=3f3499dfe1146886d0d605a38e67c5dcead7b2c1;p=pixelpark%2Fcreate-vmware-tpl.git

Start refactoring lib/cr_vmware_tpl/handler.py for using datastore clusters.
---

diff --git a/lib/cr_vmware_tpl/handler.py b/lib/cr_vmware_tpl/handler.py
index e99e391..5997cdb 100644
--- a/lib/cr_vmware_tpl/handler.py
+++ b/lib/cr_vmware_tpl/handler.py
@@ -42,7 +42,7 @@ from .cobbler import Cobbler
 
 from .xlate import XLATOR
 
-__version__ = '1.7.6'
+__version__ = '1.8.0'
 
 LOG = logging.getLogger(__name__)
 TZ = pytz.timezone('Europe/Berlin')
@@ -67,6 +67,31 @@ class TempVmExistsError(ExpectedHandlerError):
         return msg
 
 
+# =============================================================================
+class NoDatastoreFoundError(ExpectedHandlerError):
+    """Special error class for the case, no appropriate datastore coud be found."""
+
+    # -------------------------------------------------------------------------
+    def __init__(self, data_size_gb, ds_cluster_name=None):
+
+        self.data_size_gb = data_size_gb
+        self.ds_cluster_name = ds_cluster_name
+
+    # -------------------------------------------------------------------------
+    def __str__(self):
+
+        if self.ds_cluster_name:
+            msg = _(
+                "Could not find a datastore of {size:0.1f} GiB size in "
+                "datastore cluster {c_name!r}.").format(
+                size=self.data_size_gb, c_name=self.ds_cluster_name)
+        else:
+            msg = _("Could not find a datastore of {:0.1f} GiB size.").format(
+                self.data_size_gb)
+
+        return msg
+
+
 # =============================================================================
 class CrTplHandler(BaseHandler):
     """
@@ -363,8 +388,29 @@ class CrTplHandler(BaseHandler):
             "Selecting a SAN based datastore with at least {:0.1f} GiB available "
             "space.").format(self.config.data_size_gb))
 
+        self.vsphere.get_ds_clusters()
         self.vsphere.get_datastores()
 
+        ds_to_use = None
+        if self.config.storage_cluster:
+            ds_to_use = self.select_data_store_from_cluster()
+        if not ds_to_use:
+            ds_to_use = self.select_simple_data_store()
+
+        if not ds_to_use:
+            c_name = None
+            if self.config.storage_cluster:
+                c_name = self.config.storage_cluster
+            raise NoDatastoreFoundError(self.config.data_size_gb, c_name)
+
+        self.tpl_data_store = ds_to_use
+        LOG.info(_("Using datastore {!r} for volume of temporary VM to create.").format(
+            ds_to_use.name))
+        return
+
+    # -------------------------------------------------------------------------
+    def select_simple_data_store(self):
+
         usable_ds = []
         for ds in self.vsphere.datastores.values():
             if not ds.accessible:
@@ -407,10 +453,9 @@ class CrTplHandler(BaseHandler):
             if not len(ds_list):
                 continue
 
-            self.tpl_data_store = random.choice(ds_list)
-            LOG.info(_("Using datastore {!r} for volume of temporary VM to create.").format(
-                self.tpl_data_store.name))
-            break
+            return random.choice(ds_list)
+
+        return None
 
     # -------------------------------------------------------------------------
     def create_vm(self):