docs: look for functions also in the HandlerMixin classes

andresailer · Nov 6, 2023 · 05458f0 · 05458f0
1 parent e5924b3
commit 05458f0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/DIRAC/Core/Base/Client.py b/src/DIRAC/Core/Base/Client.py
@@ -157,8 +157,8 @@ def addFunctions(clientCls):
 
             # loop over all the nodes (classes, functions, imports) in the handlerModule
             for node in ast.iter_child_nodes(handlerAst):
-                # find only a class with the name of the handlerClass
-                if not (isinstance(node, ast.ClassDef) and node.name == handlerClassName):
+                # find only a class that starts with the name of the handlerClass
+                if not (isinstance(node, ast.ClassDef) and node.name.startswith(handlerClassName)):
                     continue
                 for member in ast.iter_child_nodes(node):
                     # only look at functions