chore(cli): allow overriding current function name for CLI action

nejch · nejch · commit dba828b9cfca · 2021-02-26T20:26:48.000+01:00
diff --git a/gitlab/cli.py b/gitlab/cli.py
@@ -21,7 +21,7 @@
 import functools
 import re
 import sys
-from typing import Any, Callable, Dict, Tuple
+from typing import Any, Callable, Dict, Optional, Tuple
 
 import gitlab.config
 
@@ -36,9 +36,12 @@
 
 
 def register_custom_action(
-    cls_names, mandatory: Tuple[Any, ...] = tuple(), optional: Tuple[Any, ...] = tuple()
+    cls_names,
+    mandatory: Tuple[Any, ...] = tuple(),
+    optional: Tuple[Any, ...] = tuple(),
+    custom_action: Optional[str] = None,
 ) -> Callable:
-    def wrap(f) -> Callable:
+    def wrap(f, custom_action: Optional[str] = custom_action) -> Callable:
         @functools.wraps(f)
         def wrapped_f(*args, **kwargs):
             return f(*args, **kwargs)
@@ -57,7 +60,7 @@ def wrapped_f(*args, **kwargs):
             if final_name not in custom_actions:
                 custom_actions[final_name] = {}
 
-            action = f.__name__.replace("_", "-")
+            action = custom_action or f.__name__.replace("_", "-")
             custom_actions[final_name][action] = (mandatory, optional, in_obj)
 
         return wrapped_f
diff --git a/gitlab/v4/objects/pipelines.py b/gitlab/v4/objects/pipelines.py
@@ -48,7 +48,7 @@ class MergeRequestPipelineManager(CreateMixin, ListMixin, RESTManager):
     # mr.pipelines(), execute the deprecated method for now.
     # TODO: in python-gitlab 3.0.0, remove this method entirely.
 
-    @cli.register_custom_action("ProjectMergeRequest")
+    @cli.register_custom_action("ProjectMergeRequest", custom_action="pipelines")
     @exc.on_http_error(exc.GitlabListError)
     def __call__(self, **kwargs):
         """List the merge request pipelines.
