Files
stock-logistics-warehouse/stock_mts_mto_rule/models/stock_rule.py
2020-12-18 11:58:34 +01:00

84 lines
3.5 KiB
Python

# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
from odoo import _, api, fields, models
from odoo.exceptions import ValidationError
from odoo.tools import float_compare, float_is_zero
class StockRule(models.Model):
_inherit = "stock.rule"
action = fields.Selection(
selection_add=[("split_procurement", "Choose between MTS and MTO")]
)
mts_rule_id = fields.Many2one("stock.rule", string="MTS Rule")
mto_rule_id = fields.Many2one("stock.rule", string="MTO Rule")
@api.constrains("action", "mts_rule_id", "mto_rule_id")
def _check_mts_mto_rule(self):
for rule in self:
if rule.action == "split_procurement":
if not rule.mts_rule_id or not rule.mto_rule_id:
msg = _(
"No MTS or MTO rule configured on procurement " "rule: %s!"
) % (rule.name,)
raise ValidationError(msg)
if (
rule.mts_rule_id.location_src_id.id
!= rule.mto_rule_id.location_src_id.id
):
msg = _(
"Inconsistency between the source locations of "
"the mts and mto rules linked to the procurement "
"rule: %s! It should be the same."
) % (rule.name,)
raise ValidationError(msg)
@api.multi
def get_mto_qty_to_order(self, product, product_qty, product_uom, values):
self.ensure_one()
precision = self.env["decimal.precision"].precision_get(
"Product Unit of Measure"
)
src_location_id = self.mts_rule_id.location_src_id.id
product_location = product.with_context(location=src_location_id)
virtual_available = product_location.virtual_available
qty_available = product.uom_id._compute_quantity(virtual_available, product_uom)
if float_compare(qty_available, 0.0, precision_digits=precision) > 0:
if (
float_compare(qty_available, product_qty, precision_digits=precision)
>= 0
):
return 0.0
else:
return product_qty - qty_available
return product_qty
def _run_split_procurement(
self, product_id, product_qty, product_uom, location_id, name, origin, values
):
precision = self.env["decimal.precision"].precision_get(
"Product Unit of Measure"
)
needed_qty = self.get_mto_qty_to_order(
product_id, product_qty, product_uom, values
)
if float_is_zero(needed_qty, precision_digits=precision):
getattr(self.mts_rule_id, "_run_%s" % self.mts_rule_id.action)(
product_id, product_qty, product_uom, location_id, name, origin, values
)
elif float_compare(needed_qty, product_qty, precision_digits=precision) == 0.0:
getattr(self.mto_rule_id, "_run_%s" % self.mto_rule_id.action)(
product_id, product_qty, product_uom, location_id, name, origin, values
)
else:
mts_qty = product_qty - needed_qty
getattr(self.mts_rule_id, "_run_%s" % self.mts_rule_id.action)(
product_id, mts_qty, product_uom, location_id, name, origin, values
)
getattr(self.mto_rule_id, "_run_%s" % self.mto_rule_id.action)(
product_id, needed_qty, product_uom, location_id, name, origin, values
)
return True