mirror of
https://gitlab.com/datalifeit/trytond-sale_secondary_unit
synced 2023-12-14 05:52:53 +01:00
208 lines
8.4 KiB
Python
208 lines
8.4 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 decimal import Decimal
|
|
|
|
from trytond.model import fields
|
|
from trytond.pool import PoolMeta, Pool
|
|
from trytond.pyson import Eval, If
|
|
|
|
from trytond.modules.product import price_digits
|
|
|
|
|
|
class Line(metaclass=PoolMeta):
|
|
__name__ = 'sale.line'
|
|
|
|
secondary_quantity = fields.Function(
|
|
fields.Float("Secondary Quantity",
|
|
digits=(16, Eval('secondary_unit_digits', 2)),
|
|
states={
|
|
'invisible': ((Eval('type') != 'line')
|
|
| ~Eval('secondary_unit')),
|
|
'readonly': Eval('sale_state') != 'draft',
|
|
},
|
|
depends=[
|
|
'secondary_unit_digits', 'type', 'secondary_unit',
|
|
'sale_state']),
|
|
'on_change_with_secondary_quantity', setter='set_secondary')
|
|
secondary_unit = fields.Many2One(
|
|
'product.uom', "Secondary Unit", ondelete='RESTRICT',
|
|
domain=[
|
|
If(Eval('sale_state') == 'draft',
|
|
('category', '=', Eval('product_secondary_uom_category')),
|
|
()),
|
|
],
|
|
states={
|
|
'invisible': ((Eval('type') != 'line')
|
|
| (~Eval('secondary_uom_factor')
|
|
& ~Eval('secondary_uom_rate'))),
|
|
'readonly': Eval('sale_state') != 'draft',
|
|
},
|
|
depends=['product_secondary_uom_category', 'type',
|
|
'secondary_uom_factor', 'secondary_uom_rate', 'sale_state'])
|
|
secondary_unit_price = fields.Function(
|
|
fields.Numeric(
|
|
"Secondary Unit Price", digits=price_digits,
|
|
states={
|
|
'invisible': ((Eval('type') != 'line')
|
|
| ~Eval('secondary_unit')),
|
|
'readonly': Eval('sale_state') != 'draft',
|
|
},
|
|
depends=['type', 'secondary_unit', 'sale_state']),
|
|
'on_change_with_secondary_unit_price', setter='set_secondary')
|
|
|
|
secondary_uom_factor = fields.Float("Secondary UOM Factor")
|
|
secondary_uom_rate = fields.Float("Secondary UOM Rate")
|
|
|
|
secondary_unit_digits = fields.Function(
|
|
fields.Integer("Secondary Unit Digits"),
|
|
'on_change_with_secondary_unit_digits')
|
|
product_secondary_uom_category = fields.Function(
|
|
fields.Many2One(
|
|
'product.uom.category', "Product Secondary UOM Category"),
|
|
'on_change_with_product_secondary_uom_category')
|
|
|
|
@fields.depends('product')
|
|
def _secondary_record(self):
|
|
if self.product and self.product.sale_secondary_uom:
|
|
return self.product
|
|
|
|
@fields.depends('quantity', 'unit', 'secondary_unit',
|
|
'secondary_uom_factor', 'secondary_uom_rate')
|
|
def on_change_with_secondary_quantity(self, name=None):
|
|
pool = Pool()
|
|
Uom = pool.get('product.uom')
|
|
if (self.quantity and self.unit and self.secondary_unit
|
|
and (self.secondary_uom_factor or self.secondary_uom_rate)):
|
|
return Uom.compute_qty(
|
|
self.unit, self.quantity,
|
|
self.secondary_unit, round=True,
|
|
factor=self.secondary_uom_factor, rate=self.secondary_uom_rate)
|
|
else:
|
|
return None
|
|
|
|
@fields.depends('secondary_quantity', 'secondary_unit', 'unit',
|
|
'secondary_uom_factor', 'secondary_uom_rate',
|
|
methods=['on_change_quantity', 'on_change_with_amount'])
|
|
def on_change_secondary_quantity(self):
|
|
pool = Pool()
|
|
Uom = pool.get('product.uom')
|
|
if (self.secondary_quantity and self.secondary_unit and self.unit
|
|
and (self.secondary_uom_factor or self.secondary_uom_rate)):
|
|
self.quantity = Uom.compute_qty(
|
|
self.secondary_unit, self.secondary_quantity,
|
|
self.unit, round=True,
|
|
factor=self.secondary_uom_rate, rate=self.secondary_uom_factor)
|
|
self.on_change_quantity()
|
|
self.amount = self.on_change_with_amount()
|
|
|
|
@fields.depends('unit_price', 'unit', 'secondary_unit',
|
|
'secondary_uom_factor', 'secondary_uom_rate')
|
|
def on_change_with_secondary_unit_price(self, name=None):
|
|
pool = Pool()
|
|
Uom = pool.get('product.uom')
|
|
if (self.unit_price is not None and self.unit and self.secondary_unit
|
|
and (self.secondary_uom_factor or self.secondary_uom_rate)):
|
|
return Uom.compute_price(
|
|
self.unit, self.unit_price, self.secondary_unit,
|
|
factor=self.secondary_uom_factor, rate=self.secondary_uom_rate
|
|
).quantize(Decimal(1) / 10 ** price_digits[1])
|
|
else:
|
|
return None
|
|
|
|
@fields.depends('secondary_unit_price', 'secondary_unit', 'unit',
|
|
'secondary_uom_factor', 'secondary_uom_rate',
|
|
methods=['on_change_with_amount'])
|
|
def on_change_secondary_unit_price(self, name=None):
|
|
pool = Pool()
|
|
Uom = pool.get('product.uom')
|
|
if (self.secondary_unit_price is not None
|
|
and self.secondary_unit and self.unit
|
|
and (self.secondary_uom_factor or self.secondary_uom_rate)):
|
|
self.unit_price = Uom.compute_price(
|
|
self.secondary_unit, self.secondary_unit_price, self.unit,
|
|
factor=self.secondary_uom_rate, rate=self.secondary_uom_factor
|
|
).quantize(Decimal(1) / 10 ** price_digits[1])
|
|
self.amount = self.on_change_with_amount()
|
|
|
|
@fields.depends(methods=[
|
|
'on_change_secondary_quantity', 'on_change_secondary_unit_price'])
|
|
def on_change_secondary_unit(self):
|
|
self.on_change_secondary_quantity()
|
|
self.on_change_secondary_unit_price()
|
|
|
|
@fields.depends('secondary_unit')
|
|
def on_change_with_secondary_unit_digits(self, name=None):
|
|
if self.secondary_unit:
|
|
return self.secondary_unit.digits
|
|
|
|
@fields.depends(methods=['_secondary_record'])
|
|
def on_change_with_product_secondary_uom_category(self, name=None):
|
|
secondary_record = self._secondary_record()
|
|
if secondary_record:
|
|
return secondary_record.sale_secondary_uom.category.id
|
|
|
|
@classmethod
|
|
def set_secondary(cls, lines, name, value):
|
|
pass
|
|
|
|
@fields.depends('secondary_unit',
|
|
methods=['on_change_with_secondary_quantity', '_secondary_record'])
|
|
def on_change_product(self):
|
|
super().on_change_product()
|
|
secondary_record = self._secondary_record()
|
|
if secondary_record and self.secondary_unit:
|
|
secondary_uom = secondary_record.sale_secondary_uom
|
|
if secondary_uom:
|
|
if self.secondary_unit.category != secondary_uom.category:
|
|
self.secondary_unit = None
|
|
|
|
if secondary_record:
|
|
self.secondary_uom_factor = (
|
|
secondary_record.sale_secondary_uom_normal_factor)
|
|
self.secondary_uom_rate = (
|
|
secondary_record.sale_secondary_uom_normal_rate)
|
|
else:
|
|
self.secondary_unit = None
|
|
self.secondary_uom_factor = None
|
|
self.secondary_uom_rate = None
|
|
self.secondary_quantity = self.on_change_with_secondary_quantity()
|
|
|
|
def get_invoice_line(self):
|
|
pool = Pool()
|
|
InvoiceLine = pool.get('account.invoice.line')
|
|
lines = super().get_invoice_line()
|
|
if hasattr(InvoiceLine, 'secondary_unit'):
|
|
for line in lines:
|
|
if line.unit == self.unit:
|
|
line.secondary_unit = self.secondary_unit
|
|
return lines
|
|
|
|
def get_move(self, shipment_type):
|
|
move = super().get_move(shipment_type)
|
|
if move and hasattr(move.__class__, 'secondary_unit'):
|
|
if move.uom == self.unit:
|
|
move.secondary_unit = self.secondary_unit
|
|
return move
|
|
|
|
|
|
class Line_ProductCustomer(metaclass=PoolMeta):
|
|
__name__ = 'sale.line'
|
|
|
|
@fields.depends('product_customer')
|
|
def _secondary_record(self):
|
|
if self.product_customer and self.product_customer.sale_secondary_uom:
|
|
return self.product_customer
|
|
return super()._secondary_record()
|
|
|
|
|
|
class OpportunityLine(metaclass=PoolMeta):
|
|
__name__ = 'sale.opportunity.line'
|
|
|
|
def _set_sale_line_quantity(self, sale_line):
|
|
super()._set_sale_line_quantity(sale_line)
|
|
if self.unit.category != self.product.sale_uom.category:
|
|
sale_line.unit = self.product.sale_uom
|
|
sale_line.secondary_quantity = self.quantity
|
|
sale_line.secondary_unit = self.unit
|
|
sale_line.on_change_secondary_quantity()
|