1 files changed, 57 insertions, 0 deletions
diff --git a/test/lib/ansible_test/_internal/cli/parsers/helpers.py b/test/lib/ansible_test/_internal/cli/parsers/helpers.py
new file mode 100644
index 0000000..836a893
--- /dev/null
+++ b/test/lib/ansible_test/_internal/cli/parsers/helpers.py
@@ -0,0 +1,57 @@
+"""Helper functions for composite parsers."""
+from __future__ import annotations
+
+from ...constants import (
+    CONTROLLER_PYTHON_VERSIONS,
+    SUPPORTED_PYTHON_VERSIONS,
+)
+
+from ...completion import (
+    docker_completion,
+    remote_completion,
+    filter_completion,
+)
+
+from ...host_configs import (
+    DockerConfig,
+    HostConfig,
+    PosixRemoteConfig,
+)
+
+
+def get_docker_pythons(name: str, controller: bool, strict: bool) -> list[str]:
+    """Return a list of docker instance Python versions supported by the specified host config."""
+    image_config = filter_completion(docker_completion()).get(name)
+    available_pythons = CONTROLLER_PYTHON_VERSIONS if controller else SUPPORTED_PYTHON_VERSIONS
+
+    if not image_config:
+        return [] if strict else list(available_pythons)
+
+    supported_pythons = [python for python in image_config.supported_pythons if python in available_pythons]
+
+    return supported_pythons
+
+
+def get_remote_pythons(name: str, controller: bool, strict: bool) -> list[str]:
+    """Return a list of remote instance Python versions supported by the specified host config."""
+    platform_config = filter_completion(remote_completion()).get(name)
+    available_pythons = CONTROLLER_PYTHON_VERSIONS if controller else SUPPORTED_PYTHON_VERSIONS
+
+    if not platform_config:
+        return [] if strict else list(available_pythons)
+
+    supported_pythons = [python for python in platform_config.supported_pythons if python in available_pythons]
+
+    return supported_pythons
+
+
+def get_controller_pythons(controller_config: HostConfig, strict: bool) -> list[str]:
+    """Return a list of controller Python versions supported by the specified host config."""
+    if isinstance(controller_config, DockerConfig):
+        pythons = get_docker_pythons(controller_config.name, False, strict)
+    elif isinstance(controller_config, PosixRemoteConfig):
+        pythons = get_remote_pythons(controller_config.name, False, strict)
+    else:
+        pythons = list(SUPPORTED_PYTHON_VERSIONS)
+
+    return pythons