first commit

modules/account_stock_landed_cost_weight/account.py

@@ -0,0 +1,40 @@
+# 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.pool import PoolMeta
+
+
+class LandedCost(metaclass=PoolMeta):
+    __name__ = 'account.landed_cost'
+
+    @classmethod
+    def __setup__(cls):
+        super().__setup__()
+        cls.allocation_method.selection.append(('weight', 'By Weight'))
+
+    def allocate_cost_by_weight(self):
+        self.factors = self._get_factors('weight')
+        self._allocate_cost(self.factors)
+
+    def unallocate_cost_by_weight(self):
+        factors = self.factors or self._get_factors('weight')
+        self._allocate_cost(factors, sign=-1)
+
+    def _get_weight_factors(self):
+        "Return the factor for each move based on weight"
+        moves = self.stock_moves()
+        sum_weight = Decimal(0)
+        weights = {}
+        for move in moves:
+            weight = Decimal(str(move.internal_weight or 0))
+            weights[move.id] = weight
+            sum_weight += weight
+        factors = {}
+        length = Decimal(len(moves))
+        for move in moves:
+            if not sum_weight:
+                factors[str(move.id)] = 1 / length
+            else:
+                factors[str(move.id)] = weights[move.id] / sum_weight
+        return factors