first commit
This commit is contained in:
root
398
modules/stock_consignment/stock.py
Normal file
398
modules/stock_consignment/stock.py
Normal file
@@ -0,0 +1,398 @@
|
||||
# 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.pool import Pool, PoolMeta
|
||||
from trytond.pyson import Eval
|
||||
from trytond.transaction import Transaction, inactive_records
|
||||
|
||||
|
||||
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()._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().__setup__()
|
||||
cls.warehouse_to.domain = ['OR',
|
||||
cls.warehouse_to.domain,
|
||||
('type', '=', 'storage'),
|
||||
]
|
||||
|
||||
|
||||
def set_origin_consignment(state):
|
||||
def decorator(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
|
||||
original_state, move.state = move.state, state
|
||||
if (move.on_change_with_unit_price_required()
|
||||
and move.unit_price is None):
|
||||
move.unit_price = line.unit_price
|
||||
move.currency = line.currency
|
||||
else:
|
||||
move.unit_price = None
|
||||
move.currency = None
|
||||
move.state = original_state
|
||||
cls.save(list(move2line.keys()))
|
||||
return func(cls, moves)
|
||||
return wrapper
|
||||
return decorator
|
||||
|
||||
|
||||
def unset_origin_consignment(state):
|
||||
def decorator(func):
|
||||
@wraps(func)
|
||||
def wrapper(cls, moves):
|
||||
pool = Pool()
|
||||
InvoiceLine = pool.get('account.invoice.line')
|
||||
lines, to_save = [], set()
|
||||
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.add(move)
|
||||
if (not move.on_change_with_unit_price_required()
|
||||
and (move.unit_price or move.currency)):
|
||||
move.unit_price = None
|
||||
move.currency = None
|
||||
to_save.add(move)
|
||||
if lines:
|
||||
InvoiceLine.delete(lines)
|
||||
if to_save:
|
||||
cls.save(list(to_save))
|
||||
return func(cls, moves)
|
||||
return wrapper
|
||||
return decorator
|
||||
|
||||
|
||||
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', 'unit_price', 'currency')
|
||||
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):
|
||||
# Keep the requirement to allow origin consignment decorators to
|
||||
# set the unit price before changing the move state
|
||||
required = (self.unit_price is not None) and self.currency
|
||||
return required
|
||||
|
||||
@classmethod
|
||||
def _get_origin(cls):
|
||||
return super()._get_origin() + ['account.invoice.line']
|
||||
|
||||
@fields.depends('from_location')
|
||||
def on_change_with_assignation_required(self, name=None):
|
||||
required = super().on_change_with_assignation_required(
|
||||
name=name)
|
||||
if self.from_location:
|
||||
if (self.quantity
|
||||
and self.from_location.type == 'supplier'
|
||||
and self.from_location.warehouse):
|
||||
required = True
|
||||
return required
|
||||
|
||||
@classmethod
|
||||
def search_assignation_required(cls, name, clause):
|
||||
pool = Pool()
|
||||
Location = pool.get('stock.location')
|
||||
_, operator, operand = clause
|
||||
domain = super().search_assignation_required(name, clause)
|
||||
reverse = {
|
||||
'=': '!=',
|
||||
'!=': '=',
|
||||
}
|
||||
if operator in reverse:
|
||||
with inactive_records():
|
||||
warehouses = Location.search([
|
||||
('type', '=', 'warehouse'),
|
||||
])
|
||||
warehouse_ids = list(map(int, warehouses))
|
||||
if operator == '!=':
|
||||
bool_op = 'AND'
|
||||
warehouse_op = 'not child_of'
|
||||
else:
|
||||
bool_op = 'OR'
|
||||
warehouse_op = 'child_of'
|
||||
if not operand:
|
||||
operator = reverse[operator]
|
||||
domain = [bool_op,
|
||||
domain,
|
||||
[('quantity', '!=', 0),
|
||||
('from_location.type', operator, 'supplier'),
|
||||
('from_location', warehouse_op, warehouse_ids, 'parent'),
|
||||
],
|
||||
]
|
||||
return domain
|
||||
|
||||
@property
|
||||
def is_supplier_consignment(self):
|
||||
return (self.from_location.type == 'supplier'
|
||||
and self.to_location.type in {'storage', 'production', 'customer'}
|
||||
and self.from_location.consignment_party)
|
||||
|
||||
@property
|
||||
def is_customer_consignment(self):
|
||||
return (
|
||||
self.from_location.type in {'storage', 'production', 'supplier'}
|
||||
and self.to_location.type == 'customer'
|
||||
and self.from_location.consignment_party)
|
||||
|
||||
def get_invoice_lines_consignment(self):
|
||||
lines = []
|
||||
if self.is_supplier_consignment:
|
||||
lines.append(self._get_supplier_invoice_line_consignment())
|
||||
if self.is_customer_consignment:
|
||||
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.unit
|
||||
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.id,
|
||||
taxes=[t.id for t in line.taxes]):
|
||||
line.unit_price = Product.get_purchase_price(
|
||||
[line.product], line.quantity)[line.product.id]
|
||||
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.unit
|
||||
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.id,
|
||||
taxes=[t.id for t in line.taxes]):
|
||||
line.unit_price = Product.get_sale_price(
|
||||
[line.product], line.quantity)[line.product.id]
|
||||
return line
|
||||
|
||||
@classmethod
|
||||
@ModelView.button
|
||||
@Workflow.transition('draft')
|
||||
@unset_origin_consignment('draft')
|
||||
def draft(cls, moves):
|
||||
super().draft(moves)
|
||||
|
||||
@classmethod
|
||||
@ModelView.button
|
||||
@Workflow.transition('assigned')
|
||||
@set_origin_consignment('assigned')
|
||||
def assign(cls, moves):
|
||||
super().assign(moves)
|
||||
|
||||
@classmethod
|
||||
@ModelView.button
|
||||
@Workflow.transition('done')
|
||||
@set_origin_consignment('done')
|
||||
def do(cls, moves):
|
||||
super().do(moves)
|
||||
|
||||
@classmethod
|
||||
@ModelView.button
|
||||
@Workflow.transition('cancelled')
|
||||
@unset_origin_consignment('cancelled')
|
||||
def cancel(cls, moves):
|
||||
super().cancel(moves)
|
||||
|
||||
@classmethod
|
||||
def copy(cls, moves, default=None):
|
||||
pool = Pool()
|
||||
InvoiceLine = pool.get('account.invoice.line')
|
||||
|
||||
default = default.copy() if default is not None else {}
|
||||
consigment_moves = {
|
||||
m.id for m in moves if isinstance(m.origin, InvoiceLine)}
|
||||
|
||||
def consigment_moves_cleared(name, default):
|
||||
default = default.copy()
|
||||
|
||||
def default_value(data):
|
||||
if data['id'] in consigment_moves:
|
||||
return None
|
||||
elif name in default:
|
||||
if callable(default[name]):
|
||||
return default[name](data)
|
||||
else:
|
||||
return default[name]
|
||||
else:
|
||||
return data.get(name)
|
||||
return default_value
|
||||
|
||||
default['origin'] = consigment_moves_cleared('origin', default)
|
||||
default['unit_price'] = consigment_moves_cleared('unit_price', default)
|
||||
default['currency'] = consigment_moves_cleared('currency', default)
|
||||
|
||||
moves = super().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().__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().on_change_with_planned_start_date(
|
||||
pattern=pattern)
|
||||
|
||||
|
||||
class Inventory(metaclass=PoolMeta):
|
||||
__name__ = 'stock.inventory'
|
||||
|
||||
@classmethod
|
||||
def __setup__(cls):
|
||||
super().__setup__()
|
||||
cls.location.domain = ['OR',
|
||||
cls.location.domain,
|
||||
('type', '=', 'supplier'),
|
||||
]
|
||||
|
||||
|
||||
class OrderPoint(metaclass=PoolMeta):
|
||||
__name__ = 'stock.order_point'
|
||||
|
||||
@classmethod
|
||||
def __setup__(cls):
|
||||
super().__setup__()
|
||||
cls.provisioning_location.domain = ['OR',
|
||||
cls.provisioning_location.domain,
|
||||
('type', '=', 'supplier'),
|
||||
]
|
||||
Reference in New Issue
Block a user