Implement latest flag system for OSMorphing tools

Enables marking tools as `latest` version to skip compatibility checks
while validating uniqueness per module to ensure proper tool selection.

Signed-off-by: Mihaela Balutoiu <[email protected]>

Author: mihaelabalutoiu

coriolis/osmorphing/base.py

@@ -127,6 +127,8 @@ class BaseLinuxOSMorphingTools(BaseOSMorphingTools):
 
     _packages = {}
 
+    latest = False
+
     def __init__(self, conn, os_root_dir, os_root_dev, hypervisor,
                  event_manager, detected_os_info, osmorphing_parameters,
                  operation_timeout=None):
@@ -156,6 +158,12 @@ def _version_supported_util(cls, version, minimum, maximum=None):
                 "Non-string version provided: %s (type %s)" % (
                     version, type(version)))
 
+        if cls.latest:
+            LOG.debug(
+                "Skipping all version checks for latest osmorphing tool "
+                "class: %s", cls.__name__)
+            return True
+
         float_regex = "([0-9]+(\\.[0-9]+)?)"
         match = re.match(float_regex, version)
         if not match:

coriolis/osmorphing/manager.py

@@ -85,6 +85,27 @@ def get_osmorphing_tools_class_for_provider(
         "and 'osmorphing_info' %s: %s",
         type(provider), os_type, osmorphing_info, available_tools_cls)
 
+    module_classes = {}
+    for toolscls in available_tools_cls:
+        module_name = toolscls.__module__
+        if module_name not in module_classes:
+            module_classes[module_name] = []
+        module_classes[module_name].append(toolscls)
+
+    for module_name, classes in module_classes.items():
+        latest_flags = [getattr(cls, 'latest', False) for cls in classes]
+        latest_count = sum(latest_flags)
+
+        if latest_count > 1:
+            latest_classes = [
+                cls.__name__ for cls in classes if getattr(
+                    cls, 'latest', False)]
+            raise exception.InvalidOSMorphingTools(
+                "Provider class '%s' returned multiple 'latest' OSMorphing "
+                "tools from module '%s': %s. Only one class per module "
+                "can be marked as 'latest'." % (
+                    type(provider), module_name, latest_classes))
+
     osmorphing_base_class = base_osmorphing.BaseOSMorphingTools
     for toolscls in available_tools_cls:
         if not issubclass(toolscls, osmorphing_base_class):