[stable-2.14] ansible-test - Add support for argcomplete 3 (#80482) (#80485)

(cherry picked from commit 0371ea08d6)

changelogs/fragments/ansible-test-argcomplete-3.yml

@@ -0,0 +1,2 @@
+bugfixes:
+  - ansible-test - Add support for ``argcomplete`` version 3.

test/lib/ansible_test/_internal/cli/argparsing/argcompletion.py

@@ -17,6 +17,15 @@ class Substitute:
 try:
     import argcomplete
 
+    try:
+        # argcomplete 3+
+        # see: https://github.com/kislyuk/argcomplete/commit/bd781cb08512b94966312377186ebc5550f46ae0
+        from argcomplete.finders import (
+            CompletionFinder,
+            default_validator,
+        )
+    except ImportError:
+        # argcomplete <3
         from argcomplete import (
             CompletionFinder,
             default_validator,
@@ -72,6 +81,12 @@ class CompType(enum.Enum):
 def register_safe_action(action_type: t.Type[argparse.Action]) -> None:
     """Register the given action as a safe action for argcomplete to use during completion if it is not already registered."""
     if argcomplete and action_type not in argcomplete.safe_actions:
+        if isinstance(argcomplete.safe_actions, set):
+            # argcomplete 3+
+            # see: https://github.com/kislyuk/argcomplete/commit/bd781cb08512b94966312377186ebc5550f46ae0
+            argcomplete.safe_actions.add(action_type)
+        else:
+            # argcomplete <3
             argcomplete.safe_actions += (action_type,)
 
 

test/lib/ansible_test/_util/controller/sanity/mypy/ansible-test.ini

@@ -14,6 +14,9 @@ disable_error_code = misc
 [mypy-argcomplete]
 ignore_missing_imports = True
 
+[mypy-argcomplete.finders]
+ignore_missing_imports = True
+
 [mypy-coverage]
 ignore_missing_imports = True