Classg multi kind

class_generator/README.md

@@ -10,6 +10,20 @@
 
 ###### Install poetry environment
 
+- Using [pipx](https://github.com/pypa/pipx) (recommended)
+
+```bash
+ pipx install openshift-python-wrapper
+```
+
+- Using `pip`
+
+```bash
+python3 -m pip install openshift-python-wrapper
+```
+
+- Using [poetry](https://python-poetry.org/) (For development)
+
 ```bash
 poetry install
 ```
@@ -22,7 +36,14 @@ if type class-generator > /dev/null; then eval "$(_CLASS_GENERATOR_COMPLETE=zsh_
 
 ###### Call the script
 
+- All available options:
+
+```bash
+class-generator --help
+```
+
 - Running in normal mode with `--kind` flags:
+  - `--kind` can process multiple kinds at the same command, pass `--kind <kind1>,<kind2>,<kind3>`
 
 ```bash
 class-generator --kind <kind>

class_generator/class_generator.py

@@ -9,6 +9,7 @@
 import click
 import re
 
+from concurrent.futures import Future, ThreadPoolExecutor, as_completed
 import cloup
 from cloup.constraints import If, IsSet, accept_none, require_one
 from pyhelper_utils.shell import run_command
@@ -697,7 +698,12 @@ def generate_class_generator_tests() -> None:
     "-k",
     "--kind",
     type=click.STRING,
-    help="The Kind to generate the class for, Needs working cluster with admin privileges",
+    help="""
+    \b
+    The Kind to generate the class for, Needs working cluster with admin privileges.
+    multiple kinds can be sent separated by comma (without psaces)
+    Example: -k Deployment,Pod,ConfigMap
+""",
 )
 @cloup.option(
     "-o",
@@ -755,18 +761,40 @@ def main(
     pdb: bool,
     add_tests: bool,
 ):
-    _ = pdb
+    _ = pdb  # Used by `function_runner_with_pdb`
+
+    _kwargs: Dict[str, Any] = {
+        "overwrite": overwrite,
+        "interactive": interactive,
+        "dry_run": dry_run,
+        "debug": debug,
+        "process_debug_file": debug_file,
+        "output_file": output_file,
+        "add_tests": add_tests,
+    }
 
-    class_generator(
-        kind=kind,
-        overwrite=overwrite,
-        interactive=interactive,
-        dry_run=dry_run,
-        debug=debug,
-        process_debug_file=debug_file,
-        output_file=output_file,
-        add_tests=add_tests,
-    )
+    if interactive or debug_file:
+        class_generator(**_kwargs)
+
+    else:
+        kinds: List[str] = kind.split(",")
+        futures: List[Future] = []
+
+        with ThreadPoolExecutor() as executor:
+            for _kind in kinds:
+                _kwargs["kind"] = _kind
+                if pdb or len(kinds) == 1:
+                    class_generator(**_kwargs)
+
+                else:
+                    executor.submit(
+                        class_generator,
+                        **_kwargs,
+                    )
+
+        for _ in as_completed(futures):
+            # wait for all tasks to complete
+            pass
 
     if add_tests:
         generate_class_generator_tests()

class_generator/tests/test_class_generator.py

@@ -31,6 +31,11 @@
             os.path.join(TESTS_MANIFESTS_DIR, "deployment", "deployment_debug.json"),
             os.path.join(TESTS_MANIFESTS_DIR, "deployment", "deployment_res.py"),
         ),
+        (
+            "image_content_source_policy",
+            os.path.join(TESTS_MANIFESTS_DIR, "image_content_source_policy", "image_content_source_policy_debug.json"),
+            os.path.join(TESTS_MANIFESTS_DIR, "image_content_source_policy", "image_content_source_policy_res.py"),
+        ),
         (
             "pod",
             os.path.join(TESTS_MANIFESTS_DIR, "pod", "pod_debug.json"),
@@ -41,11 +46,6 @@
             os.path.join(TESTS_MANIFESTS_DIR, "secret", "secret_debug.json"),
             os.path.join(TESTS_MANIFESTS_DIR, "secret", "secret_res.py"),
         ),
-        (
-            "image_content_source_policy",
-            os.path.join(TESTS_MANIFESTS_DIR, "image_content_source_policy", "image_content_source_policy_debug.json"),
-            os.path.join(TESTS_MANIFESTS_DIR, "image_content_source_policy", "image_content_source_policy_res.py"),
-        ),
     ),
 )
 def test_parse_explain(tmpdir_factory, kind, debug_file, result_file):