Merge pull request #7273 from DIRACGridBot/cherry-pick-2-6d71e45f6-in…

…tegration [sweep:integration] create client get functions from HandlerMixins
DIRACGrid · Nov 6, 2023 · 38cde53 · 38cde53
2 parents 0b55787 + 44f8df8
commit 38cde53
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
@@ -159,8 +159,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