Cédric Krier
426854340e
We allow to filter the stock moves updated by landed cost. issue9560 review302071002
39 lines
1.2 KiB
Python
39 lines
1.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 decimal import Decimal
|
|
|
|
from trytond.pool import PoolMeta
|
|
|
|
|
|
class LandedCost(metaclass=PoolMeta):
|
|
__name__ = 'account.landed_cost'
|
|
|
|
@classmethod
|
|
def __setup__(cls):
|
|
super(LandedCost, cls).__setup__()
|
|
cls.allocation_method.selection.append(('weight', 'By Weight'))
|
|
|
|
def allocate_cost_by_weight(self):
|
|
self._allocate_cost(self._get_weight_factors())
|
|
|
|
def unallocate_cost_by_weight(self):
|
|
self._allocate_cost(self._get_weight_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[move.id] = 1 / length
|
|
else:
|
|
factors[move.id] = weights[move.id] / sum_weight
|
|
return factors
|