31 lines
901 B
Python
31 lines
901 B
Python
# The COPYRIGHT file at the top level of this repository contains the full
|
|
# copyright notices and license terms.
|
|
from trytond.pool import Pool, PoolMeta
|
|
from trytond.pyson import Eval
|
|
from trytond.transaction import Transaction
|
|
|
|
__all__ = ['Lot', 'Move']
|
|
__metaclass__ = PoolMeta
|
|
|
|
|
|
class Lot:
|
|
__name__ = 'stock.lot'
|
|
|
|
@classmethod
|
|
def get_quantity(cls, lots, name):
|
|
"Return null instead of 0.0 if no locations in context"
|
|
if not Transaction().context.get('locations'):
|
|
return {}.fromkeys([l.id for l in lots], None)
|
|
return super(Lot, cls).get_quantity(lots, name)
|
|
|
|
|
|
class Move():
|
|
__name__ = 'stock.move'
|
|
|
|
@classmethod
|
|
def __setup__(cls):
|
|
super(Move, cls).__setup__()
|
|
cls.lot.context['locations'] = [Eval('from_location')]
|
|
if 'from_location' not in cls.lot.depends:
|
|
cls.lot.depends.append('from_location')
|