Files
tryton/bom.py
T
Cédric Krier f3dbaf9a7c Remove unique product constraint on BOM
The constraint is needed only for compute_factor which can be written to not
rely on it.

issue10363
review336091002
2021-05-18 22:19:09 +02:00

276 lines
9.2 KiB
Python

# 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, DeactivableMixin, fields
from trytond.wizard import Wizard, StateView, Button
from trytond.pyson import Eval
from trytond.pool import Pool
class BOM(DeactivableMixin, ModelSQL, ModelView):
"Bill of Material"
__name__ = 'production.bom'
name = fields.Char('Name', required=True, translate=True)
inputs = fields.One2Many('production.bom.input', 'bom', 'Inputs')
outputs = fields.One2Many('production.bom.output', 'bom', 'Outputs')
output_products = fields.Many2Many('production.bom.output',
'bom', 'product', 'Output Products')
def compute_factor(self, product, quantity, uom):
'''
Compute factor for an output product
'''
Uom = Pool().get('product.uom')
output_quantity = 0
for output in self.outputs:
if output.product == product:
output_quantity += Uom.compute_qty(
output.uom, output.quantity, uom, round=False)
if output_quantity:
return quantity / output_quantity
else:
return 0
@classmethod
def copy(cls, records, default=None):
if default is None:
default = {}
else:
default = default.copy()
default.setdefault('output_products', None)
return super(BOM, cls).copy(records, default=default)
class BOMInput(ModelSQL, ModelView):
"Bill of Material Input"
__name__ = 'production.bom.input'
bom = fields.Many2One('production.bom', 'BOM', required=True,
select=1, ondelete='CASCADE')
product = fields.Many2One('product.product', 'Product', required=True)
uom_category = fields.Function(fields.Many2One(
'product.uom.category', 'Uom Category'), 'on_change_with_uom_category')
uom = fields.Many2One('product.uom', 'Uom', required=True,
domain=[
('category', '=', Eval('uom_category')),
], depends=['uom_category'])
unit_digits = fields.Function(fields.Integer('Unit Digits'),
'on_change_with_unit_digits')
quantity = fields.Float('Quantity', required=True,
domain=['OR',
('quantity', '>=', 0),
('quantity', '=', None),
],
digits=(16, Eval('unit_digits', 2)),
depends=['unit_digits'])
@classmethod
def __setup__(cls):
super(BOMInput, cls).__setup__()
cls.product.domain = [('type', 'in', cls.get_product_types())]
cls.__access__.add('bom')
@classmethod
def __register__(cls, module):
super().__register__(module)
table_h = cls.__table_handler__(module)
# Migration from 6.0: remove unique constraint
table_h.drop_constraint('product_bom_uniq')
@classmethod
def get_product_types(cls):
return ['goods', 'assets']
@fields.depends('product', 'uom')
def on_change_product(self):
if self.product:
category = self.product.default_uom.category
if not self.uom or self.uom.category != category:
self.uom = self.product.default_uom
self.unit_digits = self.product.default_uom.digits
else:
self.uom = None
@fields.depends('product')
def on_change_with_uom_category(self, name=None):
if self.product:
return self.product.default_uom.category.id
@fields.depends('uom')
def on_change_with_unit_digits(self, name=None):
if self.uom:
return self.uom.digits
return 2
def get_rec_name(self, name):
return self.product.rec_name
@classmethod
def search_rec_name(cls, name, clause):
return [('product.rec_name',) + tuple(clause[1:])]
@classmethod
def validate(cls, boms):
super(BOMInput, cls).validate(boms)
for bom in boms:
bom.check_bom_recursion()
def check_bom_recursion(self):
'''
Check BOM recursion
'''
self.product.check_bom_recursion()
def compute_quantity(self, factor):
return self.uom.ceil(self.quantity * factor)
class BOMOutput(BOMInput):
"Bill of Material Output"
__name__ = 'production.bom.output'
_table = 'production_bom_output' # Needed to override BOMInput._table
def compute_quantity(self, factor):
return self.uom.floor(self.quantity * factor)
@classmethod
def delete(cls, outputs):
pool = Pool()
ProductBOM = pool.get('product.product-production.bom')
bom_products = [b for o in outputs for b in o.product.boms]
super(BOMOutput, cls).delete(outputs)
# Validate that output_products domain on bom is still valid
ProductBOM._validate(bom_products, ['bom'])
class BOMTree(ModelView):
'BOM Tree'
__name__ = 'production.bom.tree'
product = fields.Many2One('product.product', 'Product')
quantity = fields.Float('Quantity', digits=(16, Eval('unit_digits', 2)),
depends=['unit_digits'])
uom = fields.Many2One('product.uom', 'Uom')
unit_digits = fields.Integer('Unit Digits')
childs = fields.One2Many('production.bom.tree', None, 'Childs')
@classmethod
def tree(cls, product, quantity, uom, bom=None):
Input = Pool().get('production.bom.input')
result = []
if bom is None:
if not product.boms:
return result
bom = product.boms[0].bom
factor = bom.compute_factor(product, quantity, uom)
for input_ in bom.inputs:
quantity = Input.compute_quantity(input_, factor)
childs = cls.tree(input_.product, quantity, input_.uom)
values = {
'product': input_.product.id,
'product.rec_name': input_.product.rec_name,
'quantity': quantity,
'uom': input_.uom.id,
'uom.rec_name': input_.uom.rec_name,
'unit_digits': input_.uom.digits,
'childs': childs,
}
result.append(values)
return result
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)
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)
@fields.depends('uom')
def on_change_with_unit_digits(self):
if self.uom:
return self.uom.digits
return 2
class OpenBOMTreeTree(ModelView):
'Open BOM Tree'
__name__ = 'production.bom.tree.open.tree'
bom_tree = fields.One2Many('production.bom.tree', None, 'BOM Tree',
readonly=True)
@classmethod
def tree(cls, bom, product, quantity, uom):
pool = Pool()
Tree = pool.get('production.bom.tree')
childs = Tree.tree(product, quantity, uom, bom=bom)
bom_tree = [{
'product': product.id,
'product.rec_name': product.rec_name,
'quantity': quantity,
'uom': uom.id,
'uom.rec_name': uom.rec_name,
'unit_digits': uom.digits,
'childs': childs,
}]
return {
'bom_tree': bom_tree,
}
class OpenBOMTree(Wizard):
'Open BOM Tree'
__name__ = 'production.bom.tree.open'
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-back'),
Button('Close', 'end', 'tryton-close'),
])
def default_start(self, fields):
defaults = {}
product = self.record
defaults['category'] = product.default_uom.category.id
if getattr(self.start, 'uom', None):
defaults['uom'] = self.start.uom.id
defaults['unit_digits'] = self.start.unit_digits
else:
defaults['uom'] = product.default_uom.id
defaults['unit_digits'] = product.default_uom.digits
defaults['product'] = product.id
if getattr(self.start, 'bom', None):
defaults['bom'] = self.start.bom.id
elif product.boms:
defaults['bom'] = product.boms[0].id
defaults['quantity'] = getattr(self.start, 'quantity', None)
return defaults
def default_tree(self, fields):
pool = Pool()
BomTree = pool.get('production.bom.tree.open.tree')
return BomTree.tree(self.start.bom.bom, self.start.product,
self.start.quantity, self.start.uom)