[v4] fix CLI for some mixin methods

1 files changed, 15 insertions, 10 deletions

diff --git a/gitlab/cli.py b/gitlab/cli.py
index f6b357b..be9b112 100644
--- a/gitlab/cli.py
+++ b/gitlab/cli.py
@@ -35,7 +35,7 @@ camel_re = re.compile('(.)([A-Z])')
 custom_actions = {}
 
 
-def register_custom_action(cls_name, mandatory=tuple(), optional=tuple()):
+def register_custom_action(cls_names, mandatory=tuple(), optional=tuple()):
     def wrap(f):
         @functools.wraps(f)
         def wrapped_f(*args, **kwargs):
@@ -43,15 +43,20 @@ def register_custom_action(cls_name, mandatory=tuple(), optional=tuple()):
 
         # in_obj defines whether the method belongs to the obj or the manager
         in_obj = True
-        final_name = cls_name
-        if cls_name.endswith('Manager'):
-            final_name = cls_name.replace('Manager', '')
-            in_obj = False
-        if final_name not in custom_actions:
-            custom_actions[final_name] = {}
-
-        action = f.__name__
-        custom_actions[final_name][action] = (mandatory, optional, in_obj)
+        classes = cls_names
+        if type(cls_names) != tuple:
+            classes = (cls_names, )
+
+        for cls_name in cls_names:
+            final_name = cls_name
+            if cls_name.endswith('Manager'):
+                final_name = cls_name.replace('Manager', '')
+                in_obj = False
+            if final_name not in custom_actions:
+                custom_actions[final_name] = {}
+
+            action = f.__name__.replace('_', '-')
+            custom_actions[final_name][action] = (mandatory, optional, in_obj)
 
         return wrapped_f
     return wrap