New production module

bom.py

 #This file is part of Tryton.  The COPYRIGHT file at the top level of
 #this repository contains the full copyright notices and license terms.
 from trytond.model import ModelView, ModelSQL, fields
-from trytond.wizard import Wizard
+from trytond.wizard import Wizard, StateView, Button
+from trytond.transaction import Transaction
 from trytond.pyson import Eval
 from trytond.pool import Pool
 
 
 class BOM(ModelSQL, ModelView):
     "Bill of Material"
     _name = 'production.bom'
     _description = __doc__
 
     name = fields.Char('Name', required=True, translate=True)
@@ skipping 57 matching lines @@
         super(BOMInput, self).__init__()
         self._sql_constraints = [
             ('product_bom_uniq', 'UNIQUE(product, bom)',
                 'product_bom_uniq'),
         ]
         self._constraints += [
             ('check_bom_recursion', 'recursive_bom'),
         ]
         self._error_messages.update({
             'product_bom_uniq': 'Product must be unique per BOM!',
-            'recursive_bom': 'You can not create recursive boms!',
+            'recursive_bom': 'You can not create recursive BOMs!',
         })
 
     def on_change_product(self, vals):
         product_obj = Pool().get('product.product')
 
         res = {}
         if vals.get('product'):
             product = product_obj.browse(vals['product'])
             uom_ids = [x.id for x in product.default_uom.category.uoms]
             if (not vals.get('uom')
@@ skipping 44 matching lines @@
         return product_obj.check_bom_recursion(product_ids)
 
     def compute_quantity(self, line, factor):
         uom_obj = Pool().get('product.uom')
         return uom_obj.round(line.quantity * factor, line.uom.rounding)
 
 BOMInput()
 
 
 class BOMOutput(BOMInput):
-    "Bill of Material OutPut"
+    "Bill of Material Output"
     _name = 'production.bom.output'
     _description = __doc__
 
 BOMOutput()
 
 
 class BOMTree(ModelView):
     'BOM Tree'
     _name = 'production.bom.tree'
     _description = __doc__
@@ skipping 25 matching lines @@
                 'uom': input.uom.id,
                 'unit_digits': input.uom.digits,
                 'childs': childs,
             }
             result.append(values)
         return result
 
 BOMTree()
 
 
-class OpenBOMTreeInit(ModelView):
-    'Open BOM Tree Init'
-    _name = 'production.bom.tree.open.init'
+class OpenBOMTreeStart(ModelView):
+    'Open BOM Tree'
+    _name = 'production.bom.tree.open.start'
 
     quantity = fields.Float('Quantity', required=True,
         digits=(16, Eval('unit_digits', 2)), depends=['unit_digits'])
     uom = fields.Many2One('product.uom', 'Unit', required=True,
         domain=[
             ('category', '=', Eval('category')),
         ], depends=['category'])
     unit_digits = fields.Integer('Unit Digits', readonly=True,
         on_change_with=['uom'])
     category = fields.Many2One('product.uom.category', 'Category',
         readonly=True)
     bom = fields.Many2One('product.product-production.bom',
         'BOM', required=True, domain=[
             ('product', '=', Eval('product')),
         ], depends=['product'])
     product = fields.Many2One('product.product', 'Product', readonly=True)
 
     def on_change_with_unit_digits(self, values):
         uom_obj = Pool().get('product.uom')
         if values.get('uom'):
             uom = uom_obj.browse(values['uom'])
             return uom.digits
         return 2
 
-OpenBOMTreeInit()
+OpenBOMTreeStart()
 
 
 class OpenBOMTreeTree(ModelView):
-    'Open BOM Tree Tree'
+    'Open BOM Tree'
     _name = 'production.bom.tree.open.tree'
 
     bom_tree = fields.One2Many('production.bom.tree', None, 'BOM Tree')
 
-    def tree(self, bom_id, product_id, quantity, uom_id):
+    def tree(self, bom, product, quantity, uom):
         pool = Pool()
         tree_obj = pool.get('production.bom.tree')
-        bom_obj = pool.get('production.bom')
-        product_obj = pool.get('product.product')
-        uom_obj = pool.get('product.uom')
-
-        bom = bom_obj.browse(bom_id)
-        product = product_obj.browse(product_id)
-        uom = uom_obj.browse(uom_id)
+
         childs = tree_obj.tree(product, quantity, uom, bom=bom)
         bom_tree = [{
-            'product': product_id,
+            'product': product.id,
             'quantity': quantity,
-            'uom': uom_id,
+            'uom': uom.id,
             'unit_digits': uom.digits,
             'childs': childs,
         }]
         return {
             'bom_tree': bom_tree,
         }
 
 OpenBOMTreeTree()
 
 
 class OpenBOMTree(Wizard):
     'Open BOM Tree'
     _name = 'production.bom.tree.open'
 
-    states = {
-        'init': {
-            'actions': ['_init'],
-            'result': {
-                'type': 'form',
-                'object': 'production.bom.tree.open.init',
-                'state': [
-                    ('end', 'Cancel', 'tryton-cancel'),
-                    ('tree', 'Ok', 'tryton-ok', True),
-                ],
-            },
-        },
-        'tree': {
-            'actions': ['_tree'],
-            'result': {
-                'type': 'form',
-                'object': 'production.bom.tree.open.tree',
-                'state': [
-                    ('init', 'Change', 'tryton-go-previous'),
-                    ('end', 'Close', 'tryton-close', True),
-                ],
-            },
-        },
-    }
-
-    def _init(self, data):
-        product_obj = Pool().get('product.product')
-        result = {}
-        product = product_obj.browse(data['id'])
-        result['category'] = product.default_uom.category.id
-        result['uom'] = product.default_uom.id
-        result['unit_digits'] = product.default_uom.digits
-        result['product'] = product.id
-        if product.boms:
-            result['bom'] = product.boms[0].id
-        return result
-
-    def _tree(self, data):
-        bom_tree_obj = Pool().get('production.bom.tree.open.tree')
-        return bom_tree_obj.tree(data['form']['bom'], data['id'],
-            data['form']['quantity'], data['form']['uom'])
+    start = StateView('production.bom.tree.open.start',
+        'production.bom_tree_open_start_view_form', [
+            Button('Cancel', 'end', 'tryton-cancel'),
+            Button('Ok', 'tree', 'tryton-ok', True),
+            ])
+    tree = StateView('production.bom.tree.open.tree',
+        'production.bom_tree_open_tree_view_form', [
+            Button('Change', 'start', 'tryton-go-previous'),
+            Button('Close', 'end', 'tryton-close'),
+            ])
+
+    def default_start(self, session, fields):
+        product_obj = Pool().get('product.product')
+        defaults = {}
+        product = product_obj.browse(Transaction().context['active_id'])
+        defaults['category'] = product.default_uom.category.id
+        if session.start.uom:
+            defaults['uom'] = session.start.uom.id
+            defaults['unit_digits'] = session.start.unit_digits
+        else:
+            defaults['uom'] = product.default_uom.id
+            defaults['unit_digits'] = product.default_uom.digits
+        defaults['product'] = product.id
+        if session.start.bom:
+            defaults['bom'] = session.start.bom.id
+        elif product.boms:
+            defaults['bom'] = product.boms[0].id
+        defaults['quantity'] = session.start.quantity
+        return defaults
+
+    def default_tree(self, session, fields):
+        pool = Pool()
+        bom_tree_obj = pool.get('production.bom.tree.open.tree')
+        return bom_tree_obj.tree(session.start.bom.bom, session.start.product,
+            session.start.quantity, session.start.uom)
 
 OpenBOMTree()