Merge pull request DIRACGrid#7272 from andresailer/8_createClient

[8.0] create client get functions from HandlerMixins
fstagni · Nov 6, 2023 · 6d71e45 · 6d71e45
2 parents 1af9355 + a3e360a
commit 6d71e45
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
@@ -156,8 +156,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