From fd646da4ffa22da3941332dc3e5bb2b67d8a95b9 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?C=C3=A9dric=20Krier?= <ced@b2ck.com>
Date: Mon, 2 Jan 2012 18:06:39 +0100
Subject: [PATCH] New wizard design

---
 opportunity.py  | 55 +++++++++++++++++--------------------------------
 opportunity.xml | 12 +++++------
 2 files changed, 25 insertions(+), 42 deletions(-)

diff --git a/opportunity.py b/opportunity.py
index fd176b92bd..7ede3df3b3 100644
--- a/opportunity.py
+++ b/opportunity.py
@@ -4,7 +4,7 @@
 import datetime
 import time
 from trytond.model import ModelView, ModelSQL, ModelWorkflow, fields
-from trytond.wizard import Wizard
+from trytond.wizard import Wizard, StateView, StateAction, Button
 from trytond.backend import FIELDS
 from trytond.pyson import Equal, Eval, Not, In, If, Get, PYSONEncoder
 from trytond.transaction import Transaction
@@ -577,50 +577,33 @@ class SaleOpportunityEmployee(ModelSQL, ModelView):
 SaleOpportunityEmployee()
 
 
-class OpenSaleOpportunityEmployeeInit(ModelView):
-    'Open Sale Opportunity per Employee Init'
-    _name = 'sale.open_opportunity_employee.init'
+class OpenSaleOpportunityEmployeeStart(ModelView):
+    'Open Sale Opportunity per Employee'
+    _name = 'sale.opportunity_employee.open.start'
     _description = __doc__
     start_date = fields.Date('Start Date')
     end_date = fields.Date('End Date')
 
-OpenSaleOpportunityEmployeeInit()
+OpenSaleOpportunityEmployeeStart()
 
 
 class OpenSaleOpportunityEmployee(Wizard):
     'Open Sale Opportunity per Employee'
-    _name = 'sale.open_opportunity_employee'
-    states = {
-        'init': {
-            'result': {
-                'type': 'form',
-                'object': 'sale.open_opportunity_employee.init',
-                'state': [
-                    ('end', 'Cancel', 'tryton-cancel'),
-                    ('open', 'Open', 'tryton-ok', True),
-                ],
-            },
-        },
-        'open': {
-            'result': {
-                'type': 'action',
-                'action': '_action_open',
-                'state': 'end',
-            },
-        },
-    }
+    _name = 'sale.opportunity_employee.open'
 
-    def _action_open(self, data):
-        model_data_obj = Pool().get('ir.model.data')
-        act_window_obj = Pool().get('ir.action.act_window')
-        act_window_id = model_data_obj.get_id('sale_opportunity',
-                'act_opportunity_employee_form')
-        res = act_window_obj.read(act_window_id)
-        res['pyson_context'] = PYSONEncoder().encode({
-            'start_date': data['form']['start_date'],
-            'end_date': data['form']['end_date'],
-            })
-        return res
+    start = StateView('sale.opportunity_employee.open.start',
+        'sale_opportunity.opportunity_employee_open_start_view_form', [
+            Button('Cancel', 'end', 'tryton-cancel'),
+            Button('Open', 'open_', 'tryton-ok', default=True),
+            ])
+    open_ = StateAction('sale_opportunity.act_opportunity_employee_form')
+
+    def do_open_(self, session, action):
+        action['pyson_context'] = PYSONEncoder().encode({
+                'start_date': session.start.start_date,
+                'end_date': session.start.end_date,
+                })
+        return action, {}
 
 OpenSaleOpportunityEmployee()
 
diff --git a/opportunity.xml b/opportunity.xml
index f91f1e36bc..f19ab1ea1b 100644
--- a/opportunity.xml
+++ b/opportunity.xml
@@ -589,8 +589,8 @@
             <field name="rule_group" ref="rule_group_opportunity_employee"/>
         </record>
 
-        <record model="ir.ui.view" id="open_sale_opportunity_employee_init_view_form">
-            <field name="model">sale.open_opportunity_employee.init</field>
+        <record model="ir.ui.view" id="opportunity_employee_open_start_view_form">
+            <field name="model">sale.opportunity_employee.open.start</field>
             <field name="type">form</field>
             <field name="arch" type="xml">
                 <![CDATA[
@@ -604,12 +604,12 @@
             </field>
         </record>
 
-        <record model="ir.action.wizard" id="act_open_sale_opportunity_employee">
+        <record model="ir.action.wizard" id="act_opportunity_employee_open">
             <field name="name">Opportunities per Employee</field>
-            <field name="wiz_name">sale.open_opportunity_employee</field>
+            <field name="wiz_name">sale.opportunity_employee.open</field>
         </record>
-        <menuitem parent="menu_reporting" action="act_open_sale_opportunity_employee"
-            icon="tryton-list" id="menu_open_sale_opportunity_employee"/>
+        <menuitem parent="menu_reporting" action="act_opportunity_employee_open"
+            icon="tryton-list" id="menu_opportunity_employee_open"/>
 
         <record model="ir.ui.view" id="opportunity_monthly_view_tree">
             <field name="model">sale.opportunity_monthly</field>