From 8bc2c74b205fbceb90170b03c13ea3dc63d720b1 Mon Sep 17 00:00:00 2001
From: John Chilton <jmchilton@gmail.com>
Date: Tue, 6 May 2014 08:54:30 -0500
Subject: [PATCH] Refactor workflow module to allow multiple kinds of inputs.

---
 lib/galaxy/workflow/modules.py | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

diff --git a/lib/galaxy/workflow/modules.py b/lib/galaxy/workflow/modules.py
index 6a934ea3b14..f1d083db6c0 100644
--- a/lib/galaxy/workflow/modules.py
+++ b/lib/galaxy/workflow/modules.py
@@ -111,21 +111,19 @@ class WorkflowModule( object ):
         raise TypeError( "Abstract method" )
 
 
-class InputDataModule( WorkflowModule ):
-    type = "data_input"
-    name = "Input dataset"
+class InputModule( WorkflowModule ):
 
     @classmethod
     def new( Class, trans, tool_id=None ):
         module = Class( trans )
-        module.state = dict( name="Input Dataset" )
+        module.state = dict( name=Class.default_name )
         return module
 
     @classmethod
     def from_dict( Class, trans, d, secure=True ):
         module = Class( trans )
         state = from_json_string( d["tool_state"] )
-        module.state = dict( name=state.get( "name", "Input Dataset" ) )
+        module.state = dict( name=state.get( "name", Class.default_name ) )
         return module
 
     @classmethod
@@ -191,6 +189,12 @@ class InputDataModule( WorkflowModule ):
         return None, dict( output=state.inputs['input'])
 
 
+class InputDataModule( InputModule ):
+    type = "data_input"
+    name = "Input dataset"
+    default_name = "Input Dataset"
+
+
 class ToolModule( WorkflowModule ):
 
     type = "tool"