--- a/app/soc/views/models/timeline.py	Sat Aug 22 04:05:48 2009 +0530
+++ b/app/soc/views/models/timeline.py	Mon Aug 24 04:31:23 2009 +0530
@@ -58,7 +58,7 @@
     new_params['name'] = "Timeline"
 
     patterns = [(r'^%(url_name)s/(?P<access_type>edit)/%(key_fields)s$',
-                  '%(module_package)s.%(module_name)s.edit', 
+                  'soc.views.models.%(module_name)s.edit', 
                   "Edit %(name_short)s")]
 
     new_params['create_extra_dynaproperties'] = {
@@ -72,15 +72,16 @@
 
     super(View, self).__init__(params=params)
 
-    create_form = params_helper.getCreateForm(self._params, 
-        program_logic.logic.timeline_logic.getModel())
-    edit_form = dynaform.extendDynaForm(
-      dynaform = create_form,
-      dynainclude = self._params['edit_dynainclude'],
-      dynaexclude = self._params['edit_dynaexclude'],
-      )
+    for name, logic_value in program_logic.logic.TIMELINE_LOGIC.iteritems():
+      create_form = params_helper.getCreateForm(self._params, 
+          logic_value.getModel())
+      edit_form = dynaform.extendDynaForm(
+        dynaform = create_form,
+        dynainclude = self._params['edit_dynainclude'],
+        dynaexclude = self._params['edit_dynaexclude'],
+        )
 
-    self._params['edit_form'] = edit_form
+      self._params['edit_form_%s' % name] = edit_form
 
   def edit(self, request, access_type,
            page_name=None, params=None, seed=None, **kwargs):
@@ -96,7 +97,9 @@
 
     program = program_logic.logic.getFromKeyFields(key_fields)
     if program:
-      params['logic'] = program_logic.logic.timeline_logic
+      workflow = program.workflow
+      params['edit_form'] = params["edit_form_%s" % workflow]
+      params['logic'] = program_logic.logic.TIMELINE_LOGIC[workflow]
 
     return super(View, self).edit(request, access_type, page_name=page_name,
                                   params=params, seed=seed, **kwargs)