319 lines
10 KiB
Python
319 lines
10 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 functools import wraps
|
|
|
|
from trytond.model import ModelView, Workflow, fields
|
|
from trytond.modules.product import round_price
|
|
from trytond.pool import Pool, PoolMeta
|
|
from trytond.pyson import Eval
|
|
from trytond.transaction import Transaction
|
|
|
|
|
|
class Location(metaclass=PoolMeta):
|
|
__name__ = 'stock.location'
|
|
|
|
consignment_party = fields.Many2One(
|
|
'party.party', "Consignment Party",
|
|
states={
|
|
'invisible': ~Eval('type').in_(['supplier', 'storage']),
|
|
},
|
|
help="The party invoiced when consignment stock is used.")
|
|
|
|
@classmethod
|
|
def __setup__(cls):
|
|
super().__setup__()
|
|
cls.lost_found_location.states['invisible'] &= (
|
|
~Eval('type').in_(['supplier', 'customer']))
|
|
|
|
@classmethod
|
|
def _parent_domain(cls):
|
|
domain = super(Location, cls)._parent_domain()
|
|
domain['supplier'].append('storage')
|
|
domain['storage'].append('customer')
|
|
return domain
|
|
|
|
@property
|
|
def lost_found_used(self):
|
|
lost_found = super().lost_found_used
|
|
if not lost_found and not self.warehouse and self.type == 'storage':
|
|
location = self.parent
|
|
while location:
|
|
if location.type in {'supplier', 'storage'}:
|
|
lost_found = location.lost_found_location
|
|
break
|
|
location = location.parent
|
|
return lost_found
|
|
|
|
|
|
class LocationLeadTime(metaclass=PoolMeta):
|
|
__name__ = 'stock.location.lead_time'
|
|
|
|
@classmethod
|
|
def __setup__(cls):
|
|
super(LocationLeadTime, cls).__setup__()
|
|
cls.warehouse_to.domain = ['OR',
|
|
cls.warehouse_to.domain,
|
|
('type', '=', 'storage'),
|
|
]
|
|
|
|
|
|
def set_origin_consignment(func):
|
|
@wraps(func)
|
|
def wrapper(cls, moves):
|
|
pool = Pool()
|
|
InvoiceLine = pool.get('account.invoice.line')
|
|
to_save = []
|
|
move2line = {}
|
|
for move in moves:
|
|
if not move.consignment_invoice_lines:
|
|
lines = move.get_invoice_lines_consignment()
|
|
if lines:
|
|
to_save.extend(lines)
|
|
move2line[move] = lines[0]
|
|
if to_save:
|
|
InvoiceLine.save(to_save)
|
|
for move, line in move2line.items():
|
|
if not move.origin:
|
|
move.origin = line
|
|
if move.unit_price is None:
|
|
move.unit_price = line.unit_price
|
|
move.currency = line.currency
|
|
cls.save(list(move2line.keys()))
|
|
return func(cls, moves)
|
|
return wrapper
|
|
|
|
|
|
def unset_origin_consignment(func):
|
|
@wraps(func)
|
|
def wrapper(cls, moves):
|
|
pool = Pool()
|
|
InvoiceLine = pool.get('account.invoice.line')
|
|
lines, to_save = [], []
|
|
for move in moves:
|
|
for invoice_line in move.consignment_invoice_lines:
|
|
lines.append(invoice_line)
|
|
if move.origin == move:
|
|
move.origin = None
|
|
to_save.append(move)
|
|
if lines:
|
|
InvoiceLine.delete(lines)
|
|
cls.save(to_save)
|
|
return func(cls, moves)
|
|
return wrapper
|
|
|
|
|
|
class Move(metaclass=PoolMeta):
|
|
__name__ = 'stock.move'
|
|
|
|
consignment_invoice_lines = fields.One2Many(
|
|
'account.invoice.line', 'origin', "Consignment Invoice Lines",
|
|
readonly=True,
|
|
states={
|
|
'invisible': ~Eval('consignment_invoice_lines'),
|
|
})
|
|
|
|
@fields.depends('state', 'from_location', 'to_location')
|
|
def on_change_with_unit_price_required(self, name=None):
|
|
required = super().on_change_with_unit_price_required(name)
|
|
if (required
|
|
and self.state in {'staging', 'draft'}
|
|
and self.from_location
|
|
and self.to_location
|
|
and ((
|
|
self.from_location.type == 'supplier'
|
|
and self.to_location.type in {
|
|
'storage', 'production', 'customer'})
|
|
or (self.from_location.type in {
|
|
'storage', 'production', 'supplier'}
|
|
and self.to_location.type == 'customer'))
|
|
and self.from_location.consignment_party):
|
|
required = False
|
|
return required
|
|
|
|
@classmethod
|
|
def _get_origin(cls):
|
|
return super(Move, cls)._get_origin() + ['account.invoice.line']
|
|
|
|
@fields.depends('from_location')
|
|
def on_change_with_assignation_required(self, name=None):
|
|
required = super(Move, self).on_change_with_assignation_required(
|
|
name=name)
|
|
if self.from_location:
|
|
if (self.from_location.type == 'supplier'
|
|
and self.from_location.warehouse):
|
|
required = True
|
|
return required
|
|
|
|
def get_invoice_lines_consignment(self):
|
|
lines = []
|
|
if (self.from_location.type == 'supplier'
|
|
and self.to_location.type in {
|
|
'storage', 'production', 'customer'}
|
|
and self.from_location.consignment_party):
|
|
lines.append(self._get_supplier_invoice_line_consignment())
|
|
if (self.from_location.type in {'storage', 'production', 'supplier'}
|
|
and self.to_location.type == 'customer'
|
|
and self.from_location.consignment_party):
|
|
lines.append(self._get_customer_invoice_line_consignment())
|
|
return lines
|
|
|
|
def _get_supplier_invoice_line_consignment(self):
|
|
pool = Pool()
|
|
InvoiceLine = pool.get('account.invoice.line')
|
|
Product = pool.get('product.product')
|
|
ProductSupplier = pool.get('purchase.product_supplier')
|
|
|
|
with Transaction().set_context(
|
|
supplier=self.from_location.consignment_party.id):
|
|
pattern = ProductSupplier.get_pattern()
|
|
for product_supplier in self.product.product_suppliers_used(**pattern):
|
|
currency = product_supplier.currency
|
|
break
|
|
else:
|
|
currency = self.company.currency
|
|
|
|
line = InvoiceLine()
|
|
line.invoice_type = 'in'
|
|
line.type = 'line'
|
|
line.company = self.company
|
|
line.party = self.from_location.consignment_party
|
|
line.currency = currency
|
|
line.product = self.product
|
|
line.quantity = self.quantity
|
|
line.unit = self.uom
|
|
line.stock_moves = [self]
|
|
line.origin = self
|
|
line.on_change_product()
|
|
|
|
with Transaction().set_context(
|
|
currency=line.currency.id,
|
|
supplier=line.party.id,
|
|
uom=line.unit,
|
|
taxes=[t.id for t in line.taxes]):
|
|
line.unit_price = Product.get_purchase_price(
|
|
[line.product], line.quantity)[line.product.id]
|
|
if line.unit_price is not None:
|
|
line.unit_price = round_price(line.unit_price)
|
|
return line
|
|
|
|
def _get_customer_invoice_line_consignment(self):
|
|
pool = Pool()
|
|
InvoiceLine = pool.get('account.invoice.line')
|
|
Product = pool.get('product.product')
|
|
|
|
line = InvoiceLine()
|
|
line.invoice_type = 'out'
|
|
line.type = 'line'
|
|
line.company = self.company
|
|
line.party = self.from_location.consignment_party
|
|
line.currency = self.company.currency
|
|
line.product = self.product
|
|
line.quantity = self.quantity
|
|
line.unit = self.uom
|
|
line.stock_moves = [self]
|
|
line.origin = self
|
|
line.on_change_product()
|
|
|
|
with Transaction().set_context(
|
|
currency=line.currency.id,
|
|
customer=line.party.id,
|
|
uom=line.unit,
|
|
taxes=[t.id for t in line.taxes]):
|
|
line.unit_price = Product.get_sale_price(
|
|
[line.product], line.quantity)[line.product.id]
|
|
if line.unit_price is not None:
|
|
line.unit_price = round_price(line.unit_price)
|
|
return line
|
|
|
|
@classmethod
|
|
@ModelView.button
|
|
@Workflow.transition('draft')
|
|
@unset_origin_consignment
|
|
def draft(cls, moves):
|
|
super(Move, cls).draft(moves)
|
|
|
|
@classmethod
|
|
@ModelView.button
|
|
@Workflow.transition('assigned')
|
|
@set_origin_consignment
|
|
def assign(cls, moves):
|
|
super(Move, cls).assign(moves)
|
|
|
|
@classmethod
|
|
@ModelView.button
|
|
@Workflow.transition('done')
|
|
@set_origin_consignment
|
|
def do(cls, moves):
|
|
super(Move, cls).do(moves)
|
|
|
|
@classmethod
|
|
@ModelView.button
|
|
@Workflow.transition('cancelled')
|
|
@unset_origin_consignment
|
|
def cancel(cls, moves):
|
|
super(Move, cls).cancel(moves)
|
|
|
|
@classmethod
|
|
def copy(cls, moves, default=None):
|
|
pool = Pool()
|
|
InvoiceLine = pool.get('account.invoice.line')
|
|
moves = super(Move, cls).copy(moves, default=default)
|
|
if not Transaction().context.get('_stock_move_split'):
|
|
to_save = []
|
|
for move in moves:
|
|
if isinstance(move.origin, InvoiceLine):
|
|
move.origin = None
|
|
to_save.append(move)
|
|
if to_save:
|
|
cls.save(to_save)
|
|
return moves
|
|
|
|
|
|
class ShipmentInternal(metaclass=PoolMeta):
|
|
__name__ = 'stock.shipment.internal'
|
|
|
|
@classmethod
|
|
def __setup__(cls):
|
|
super(ShipmentInternal, cls).__setup__()
|
|
cls.from_location.domain = ['OR',
|
|
cls.from_location.domain,
|
|
('type', '=', 'supplier'),
|
|
]
|
|
cls.to_location.domain = ['OR',
|
|
cls.to_location.domain,
|
|
('type', 'in', ['supplier', 'customer']),
|
|
]
|
|
|
|
@fields.depends('to_location')
|
|
def on_change_with_planned_start_date(self, pattern=None):
|
|
if pattern is None:
|
|
pattern = {}
|
|
if self.to_location and not self.to_location.warehouse:
|
|
pattern.setdefault('location_to', self.to_location.id)
|
|
return super(ShipmentInternal, self).on_change_with_planned_start_date(
|
|
pattern=pattern)
|
|
|
|
|
|
class Inventory(metaclass=PoolMeta):
|
|
__name__ = 'stock.inventory'
|
|
|
|
@classmethod
|
|
def __setup__(cls):
|
|
super(Inventory, cls).__setup__()
|
|
cls.location.domain = ['OR',
|
|
cls.location.domain,
|
|
('type', '=', 'supplier'),
|
|
]
|
|
|
|
|
|
class OrderPoint(metaclass=PoolMeta):
|
|
__name__ = 'stock.order_point'
|
|
|
|
@classmethod
|
|
def __setup__(cls):
|
|
super(OrderPoint, cls).__setup__()
|
|
cls.provisioning_location.domain = ['OR',
|
|
cls.provisioning_location.domain,
|
|
('type', '=', 'supplier'),
|
|
]
|