Adds support for workflow requirements

Author: dleehr

cwlgen/__init__.py

@@ -339,7 +339,7 @@ def __init__(self, expression_lib=None):
 
     def _to_dict(self):
         return {'expressionLib': [self.expressionLib]}
-    
+
 
 class DockerRequirement(Requirement):
     '''

cwlgen/workflow.py

@@ -15,6 +15,7 @@ def __init__(self):
         self.steps = []
         self.inputs = []
         self.outputs = []
+        self.requirements = []
         self._path = None
 
 
@@ -47,6 +48,14 @@ def export(self, outfile=None):
             for out in self.outputs:
                 cwl_workflow['outputs'][out.id] = out.get_dict()
 
+        # Add requirements.
+        requirements = {}
+        for requirement in self.requirements:
+            requirement.add(requirements)
+
+        if requirements:
+            cwl_workflow['requirements'] = requirements
+
         # Write CWL file in YAML
         if outfile is None:
             six.print_(CWL_SHEBANG, "\n", sep='')

test/test_unit_workflow.py

@@ -86,5 +86,20 @@ def test_generates_workflow_int_inputs(self):
         self.assertEqual(expected, generated)
 
 
+    def test_add_requirements(self):
+        w = cwlgen.Workflow()
+        req = cwlgen.InlineJavascriptReq(expression_lib='test_lib')
+        w.requirements.append(req)
+        generated = self.capture_tempfile(w.export)
+        expected = b"""#!/usr/bin/env cwl-runner
 
+class: Workflow
+cwlVersion: v1.0
+inputs: {}
+outputs: {}
+requirements:
+  InlineJavascriptRequirement:
+    expressionLib: [test_lib]
+"""
+        self.assertEqual(expected, generated)