Initial commit: Ventor Odoo packages (4 packages)

odoo-bringout-ventor-outgoing_routing/outgoing_routing/models/stock_quant.py

@@ -0,0 +1,108 @@
+# Copyright 2020 VentorTech OU
+# License LGPL-3.0 or later (https://www.gnu.org/licenses/lgpl-3.0).
+
+from odoo import api, models, fields
+from odoo.tools.float_utils import float_compare
+
+import functools
+
+
+class StockQuant(models.Model):
+    _inherit = 'stock.quant'
+
+    removal_prio = fields.Integer(
+        related="location_id.removal_prio",
+        store=True,
+    )
+
+    @api.model
+    def _get_removal_strategy_order(self, removal_strategy):
+        # THIS IS A OVERRIDE STANDARD METHOD
+        strategy_order = self.env.user.company_id.outgoing_routing_order
+
+        if removal_strategy == 'location_priority':
+            return 'removal_prio %s, id' % (['ASC', 'DESC'][int(strategy_order)])
+        return super(StockQuant, self)._get_removal_strategy_order(removal_strategy)
+
+    @api.model
+    def _update_reserved_quantity(self, product_id, location_id, quantity, lot_id=None, package_id=None, owner_id=None, strict=False):
+        """ Updates reserved quantity in quants
+        """
+        self = self.with_context(reservation_strategy=self.env.user.company_id.stock_reservation_strategy, reservation_quantity=quantity)
+        return super(StockQuant, self)._update_reserved_quantity(product_id, location_id, quantity, lot_id, package_id, owner_id, strict)
+
+    def _gather(self, product_id, location_id, lot_id=None, package_id=None, owner_id=None, strict=False):
+        """ Gather (and reorder, if required) quants
+        """
+        context = dict(self.env.context)
+        quants = super(StockQuant, self)._gather(product_id, location_id, lot_id=lot_id, package_id=package_id, owner_id=owner_id, strict=strict)
+
+        if self._context.get('skip_ventor_reordering') or (product_id.categ_id and product_id.categ_id.removal_strategy_id)\
+                or location_id.removal_strategy_id:
+            return quants
+
+        strategy = self.env.user.company_id.stock_reservation_strategy
+
+        func_reorder = getattr(self, '_reorder_{}'.format(strategy))
+        if func_reorder:
+            quants = func_reorder(quants, product_id)
+
+        # mom taught us to clean after ourselves
+        if context.get('reservation_strategy'):
+            del context['reservation_strategy']
+        if context.get('reservation_quantity'):
+            del context['reservation_quantity']
+        self = self.with_context(context)
+
+        return quants
+
+    def _reorder_none(self, quants, product_id):
+        """ No reorder, i.e. use out of the box strategy
+        """
+        return quants
+
+    def _reorder_base(self, quants, product_id):
+        """ Reorders quants by location removal priority
+        """
+        def _r_getattr(obj, attr, *args):
+            return functools.reduce(getattr, [obj] + attr.split('.'))
+
+        route = self.env.user.company_id.outgoing_routing_strategy
+        order = self.env.user.company_id.outgoing_routing_order
+
+        return quants.sorted(
+            key=lambda op: _r_getattr(op, route, 'None'),
+            reverse=int(order)
+        )
+
+    def _reorder_quantity(self, quants, product_id):
+        """ Reorders quants by product quantity in locations and location priority
+        """
+        default_route = 'name'  # i.e. location_id.name
+
+        route = self.env.user.company_id.outgoing_routing_strategy
+        order = self.env.user.company_id.outgoing_routing_order
+
+        base, field = route.split('.', 1)
+        startegy = field if base in ('location_id') else default_route
+
+        required = self.env.context.get('reservation_quantity', 0)
+        rounding = product_id.uom_id.rounding
+
+        locations = {}
+        queues = [self.env['stock.quant'], self.env['stock.quant']]  # (lprio, hprio)
+
+        for quant in quants:
+            locations.setdefault(quant.location_id, []).append(quant)
+
+        for location in sorted(locations,
+            key=lambda location: getattr(location, startegy, 'None'),
+            reverse=int(order)
+        ):
+            location_quants = locations.get(location)
+            quantity = sum([qt.quantity - qt.reserved_quantity for qt in location_quants])
+            priority = float_compare(quantity, required, precision_rounding=rounding) >= 0
+            for location_quant in location_quants:
+                queues[priority] |= location_quant
+
+        return queues[True] + queues[False]