trytond-sale_unit_load/unit_load.py
2020-03-17 17:55:48 +00:00

56 lines
1.8 KiB
Python

# The COPYRIGHT file at the top level of this repository contains the full
# copyright notices and license terms.
from trytond.model import fields
from trytond.pool import PoolMeta
__all__ = ['UnitLoad']
class UnitLoad(metaclass=PoolMeta):
__name__ = 'stock.unit_load'
customer = fields.Function(fields.Many2One('party.party', 'Customer'),
'get_customer')
sale = fields.Function(fields.Many2One('sale.sale', 'Sale'), 'get_sale',
searcher='search_sale')
sale_line = fields.Function(
fields.Many2One('sale.line', 'Sale line'), 'get_sale_line',
searcher='search_sale_line')
def get_customer(self, name=None):
return self.shipment_out.customer.id if self.shipment_out else None
def get_sale(self, name=None):
if self.sale_line:
return self.sale_line.sale.id
return None
def get_sale_line(self, name=None):
moves = [m for m in self.moves
if m.origin and m.origin.__name__ == 'sale.line' and
m.state not in ('draft', 'cancel')]
return moves[0].origin.id if moves else None
@classmethod
def search_sale_line(cls, name, clause):
_field = 'moves.origin'
if '.' in clause[0]:
_field += '.%s' % clause[0][10:]
else:
if isinstance(clause[2], (int, list, tuple, set)):
_field += '.id'
else:
_field += '.rec_name'
return [(_field,) + tuple(clause[1:]) + ('sale.line',)]
@classmethod
def search_sale(cls, name, clause):
_field = 'sale_line.%s' % clause[0]
if isinstance(clause[2], (int, list, tuple, set)):
_field += '.id'
else:
_field += '.rec_name'
return [(_field, ) + tuple(clause[1:])]