From 1a7f0ead2b6c8f830c16a6786f4c0ff5af4d5142 Mon Sep 17 00:00:00 2001
From: Mustufa Rangwala <mra@mra-laptop>
Date: Wed, 1 Sep 2010 13:13:35 +0530
Subject: [PATCH] [IMP] Budget: workflow => cancel to draft

bzr revid: mra@mra-laptop-20100901074335-fw3fndwyyq4uzcl9
---
 addons/account_budget/crossovered_budget.py           |  6 ++++++
 addons/account_budget/crossovered_budget_view.xml     |  1 +
 addons/account_budget/crossovered_budget_workflow.xml | 11 +++++++++--
 3 files changed, 16 insertions(+), 2 deletions(-)

diff --git a/addons/account_budget/crossovered_budget.py b/addons/account_budget/crossovered_budget.py
index a0973cde6a1..9252d2cf210 100644
--- a/addons/account_budget/crossovered_budget.py
+++ b/addons/account_budget/crossovered_budget.py
@@ -126,6 +126,12 @@ class crossovered_budget(osv.osv):
         })
         return True
 
+    def budget_draft(self, cr, uid, ids, *args):
+        self.write(cr, uid, ids, {
+            'state':'draft'
+        })
+        return True
+
     def budget_validate(self, cr, uid, ids, *args):
         self.write(cr, uid, ids, {
             'state':'validate',
diff --git a/addons/account_budget/crossovered_budget_view.xml b/addons/account_budget/crossovered_budget_view.xml
index 45b9d726597..b863979ad92 100644
--- a/addons/account_budget/crossovered_budget_view.xml
+++ b/addons/account_budget/crossovered_budget_view.xml
@@ -149,6 +149,7 @@
                                 <button string="Approve" name="validate" states="confirm" type="workflow" icon="gtk-ok"/>
                                 <button string="Done" name="done" states="validate" type="workflow"   icon="gtk-jump-to" />
                                 <button string="Cancel" name="cancel" states="confirm,validate" type="workflow" icon="gtk-cancel"/>
+                                <button name="draft" states="cancel" string="Reset to Draft" type="workflow" icon="terp-stock_effects-object-colorize"/>
                             </group>
 
                 </form>
diff --git a/addons/account_budget/crossovered_budget_workflow.xml b/addons/account_budget/crossovered_budget_workflow.xml
index cae760d1507..f304775bcca 100644
--- a/addons/account_budget/crossovered_budget_workflow.xml
+++ b/addons/account_budget/crossovered_budget_workflow.xml
@@ -19,7 +19,8 @@
 		<field name="wkf_id" ref="wkf_crossovered_budget" />
 		<field name="flow_start">True</field>
 		<field name="name">draft</field>
-
+		<field name="action">write({'state':'draft'})</field>
+        <field name="kind">function</field>
 	</record>
 
 	<record model="workflow.activity" id="act_confirm">
@@ -42,8 +43,8 @@
 		<field name="name">cancel</field>
 		<field name="kind">function</field>
 		<field name="action">budget_cancel()</field>
-
 	</record>
+
 	<record model="workflow.activity" id="act_done">
 		<field name="wkf_id" ref="wkf_crossovered_budget" />
 		<field name="name">done</field>
@@ -88,5 +89,11 @@
 		<field name="role_id" ref="crossovered_budget_role"/>
 	</record>
 
+	<record model="workflow.transition" id="t6">
+		<field name="act_from" ref="act_cancel" />
+		<field name="act_to" ref="act_draft" />
+		<field name="signal">draft</field>
+	</record>
+
 </data>
 </openerp>