simple input object generation

Author: mr-c

cwltool/main.py

@@ -3,6 +3,7 @@
 from __future__ import absolute_import
 
 import argparse
+import collections
 import functools
 import json
 import logging
@@ -220,6 +221,9 @@ def arg_parser():  # type: () -> argparse.ArgumentParser
     parser.add_argument("--relax-path-checks", action="store_true",
                         default=False, help="Relax requirements on path names to permit "
                         "spaces and hash characters.", dest="relax_path_checks")
+    exgroup.add_argument("--make-template", action="store_true",
+                         help="Generate a template input object")
+
 
     parser.add_argument("workflow", type=Text, nargs="?", default=None)
     parser.add_argument("job_order", nargs=argparse.REMAINDER)
@@ -439,6 +443,56 @@ def generate_parser(toolparser, tool, namemap, records):
 
     return toolparser
 
+def generate_example_input(inptype):
+    defaults = { 'null': 'null',
+                 'Any': 'null',
+                 'boolean': False,
+                 'int': 0,
+                 'long': 0,
+                 'float': 0.1,
+                 'double': 0.1,
+                 'string': 'default_string',
+                 'File': { 'class': 'File',
+                           'path': 'default/file/path' },
+                 'Directory': { 'class': 'Directory',
+                                'path': 'default/directory/path' } }
+    if (not isinstance(inptype, str) and
+        not isinstance(inptype, collections.Mapping)
+        and isinstance(inptype, collections.Iterable)):
+        if len(inptype) == 2 and 'null' in inptype:
+            inptype.remove('null')
+            return generate_example_input(inptype[0])
+            # TODO: indicate that this input is optional
+        else:
+            raise Exception("multi-types other than optional not yet supported"
+                            " for generating example input objects: %s"
+                            % inptype)
+    if 'type' in inptype:
+        if inptype['type'] == 'array':
+            return [ generate_example_input(inptype['items']) ]
+        elif inptype['type'] == 'enum':
+            return 'valid_enum_value'
+            # TODO: list valid values in a comment
+        elif inptype['type'] == 'record':
+            record = {}
+            for field in inptype['fields']:
+                record[shortname(field)] = generate_example_input(field)
+            return record
+    else:
+        return defaults.get(inptype, 'custom_type')
+        # TODO: support custom types, complex arrays
+
+
+def generate_input_template(tool):
+    # type: (Process) -> Dict[Text, Any]
+    template = {}
+    for inp in tool.tool["inputs"]:
+        name = shortname(inp["id"])
+        inptype = inp["type"]
+        template[name] = generate_example_input(inptype)
+    return template
+
+
 
 def load_job_order(args, t, stdin, print_input_deps=False, relative_deps=False,
                    stdout=sys.stdout, make_fs_access=None, fetcher_constructor=None):
@@ -669,7 +723,8 @@ def main(argsl=None,  # type: List[str]
                      'validate': False,
                      'enable_ga4gh_tool_registry': False,
                      'ga4gh_tool_registries': [],
-                     'find_default_container': None
+                     'find_default_container': None,
+                     'make_template': False
         }):
             if not hasattr(args, k):
                 setattr(args, k, v)
@@ -751,6 +806,11 @@ def main(argsl=None,  # type: List[str]
 
             tool = make_tool(document_loader, avsc_names, metadata, uri,
                              makeTool, make_tool_kwds)
+            if args.make_template:
+                yaml.safe_dump(generate_input_template(tool), sys.stdout,
+                               default_flow_style=False, indent=4,
+                               block_seq_indent=2)
+                return 0
 
             if args.validate:
                 return 0