another try

Author: jan-janssen

python_workflow_definition/src/python_workflow_definition/cwl/__main__.py

@@ -4,13 +4,12 @@
 import importlib.util
 
 
-def load_function(funct):
-    p, m = funct.rsplit(".", 1)
-    spec = importlib.util.spec_from_file_location(p, p + ".py")
+def load_function(file_name, funct):
+    spec = importlib.util.spec_from_file_location("workflow", file_name)
     module = importlib.util.module_from_spec(spec)
-    sys.modules[p] = module
+    sys.modules["workflow"] = module
     spec.loader.exec_module(module)
-    return getattr(module, m)
+    return getattr(module, funct)
 
 
 def convert_argument(arg):
@@ -24,19 +23,17 @@ def convert_argument(arg):
 if __name__ == "__main__":
     # load input
     argument_lst = sys.argv[1:]
-    funct = [
-        load_function(funct=arg.split("=")[-1])
-        for arg in argument_lst
-        if "--function=" in arg
-    ][0]
+    funct = [arg.split("=")[-1] for arg in argument_lst if "--function=" in arg][0]
+    file = [arg.split("=")[-1] for arg in argument_lst if "--workflowfile=" in arg][0]
+    workflow_function = load_function(file_name=file, funct=funct)
     kwargs = {
         arg.split("=")[0][6:]: convert_argument(arg=arg.split("=")[-1])
         for arg in argument_lst
         if "--arg_" in arg
     }
 
     # evaluate function
-    result = funct(**kwargs)
+    result = workflow_function(**kwargs)
 
     # store output
     if isinstance(result, dict):