Adding upstream version 2.14.3.upstream/2.14.3upstream

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>

1 files changed, 18 insertions, 0 deletions

diff --git a/test/lib/ansible_test/_internal/cli/argparsing/actions.py b/test/lib/ansible_test/_internal/cli/argparsing/actions.py
new file mode 100644
index 0000000..2bcf982
--- /dev/null
+++ b/test/lib/ansible_test/_internal/cli/argparsing/actions.py
@@ -0,0 +1,18 @@
+"""Actions for argparse."""
+from __future__ import annotations
+
+import argparse
+import enum
+import typing as t
+
+
+class EnumAction(argparse.Action):
+    """Parse an enum using the lowercase enum names."""
+    def __init__(self, **kwargs: t.Any) -> None:
+        self.enum_type: t.Type[enum.Enum] = kwargs.pop('type', None)
+        kwargs.setdefault('choices', tuple(e.name.lower() for e in self.enum_type))
+        super().__init__(**kwargs)
+
+    def __call__(self, parser, namespace, values, option_string=None):
+        value = self.enum_type[values.upper()]
+        setattr(namespace, self.dest, value)