Files
tradon/modules/product_kit/sale.py
2026-03-14 09:42:12 +00:00

240 lines
8.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 trytond.i18n import gettext
from trytond.model import ModelSQL, ModelView, fields
from trytond.modules.product import round_price
from trytond.pool import Pool, PoolMeta
from trytond.pyson import Eval
from trytond.transaction import Transaction
from .common import (
AmendmentLineMixin, get_moves, get_shipments_returns,
handle_shipment_exception_mixin, order_line_component_mixin,
order_line_mixin, order_mixin, search_moves, search_shipments_returns)
from .product import ComponentMixin
class Sale(order_mixin('sale'), metaclass=PoolMeta):
__name__ = 'sale.sale'
def _get_shipment_moves(self, shipment_type):
moves = super()._get_shipment_moves(shipment_type)
for line in self.lines:
if line.components:
for component in line.components:
move = component.get_move(shipment_type)
if move:
moves[component] = move
return moves
@get_shipments_returns('stock.shipment.out')
def get_shipments(self, name):
return super().get_shipments(name)
@get_shipments_returns('stock.shipment.out.return')
def get_shipment_returns(self, name):
return super().get_shipment_returns(name)
@classmethod
@search_shipments_returns('stock.shipment.out')
def search_shipments(cls, name, clause):
return super().search_shipments(name, clause)
@classmethod
@search_shipments_returns('stock.shipment.out.return')
def search_shipment_returns(cls, name, clause):
return super().search_shipment_returns(name, clause)
@get_moves
def get_moves(self, name):
return super().get_moves(name)
@classmethod
@search_moves
def search_moves(cls, name, clause):
return super().search_moves(name, clause)
@property
def _invoice_grouping_origins(self):
return super()._shipment_grouping_origins + ['sale.line.component']
@property
def _shipment_grouping_origins(self):
return super()._shipment_grouping_origins + ['sale.line.component']
class Line(order_line_mixin('sale'), metaclass=PoolMeta):
__name__ = 'sale.line'
def get_component_order_line(self, component, **values):
values = values.copy()
values['sale'] = self.sale
return super().get_component_order_line(component, **values)
@classmethod
def movable_types(cls):
return super().movable_types() + ['kit']
class LineComponent(
order_line_component_mixin('sale'), ComponentMixin,
ModelSQL, ModelView):
__name__ = 'sale.line.component'
@classmethod
def __setup__(cls):
super().__setup__()
states = {
'readonly': Eval('line_state') != 'draft',
}
cls.product.states = states
cls.quantity.states = states
cls.unit.states = states
cls.fixed.states = states
@property
def warehouse(self):
return self.line.warehouse
def get_move(self, shipment_type):
from trytond.modules.sale.exceptions import PartyLocationError
pool = Pool()
Move = pool.get('stock.move')
Date = pool.get('ir.date')
if (shipment_type == 'out') != (self.quantity >= 0):
return
quantity = (
self._get_move_quantity(shipment_type)
- self._get_shipped_quantity(shipment_type))
quantity = self.unit.round(quantity)
if quantity <= 0:
return
if not self.line.sale.party.customer_location:
raise PartyLocationError(
gettext('sale.msg_sale_customer_location_required',
sale=self.sale.rec_name,
party=self.sale.party.rec_name))
with Transaction().set_context(company=self.line.sale.company.id):
today = Date.today()
move = Move()
move.quantity = quantity
move.unit = self.unit
move.product = self.product
move.from_location = self.line.from_location
move.to_location = self.line.to_location
move.state = 'draft'
move.company = self.line.sale.company
if move.on_change_with_unit_price_required():
move.unit_price = round_price(
self.line.unit_price * self.price_ratio)
move.currency = self.line.sale.currency
else:
move.unit_price = None
move.currency = None
move.planned_date = max(
self.line.planned_shipping_date or today, today)
move.origin = self
move.origin_planned_date = move.planned_date
return move
def _get_move_quantity(self, shipment_type):
'Return the quantity that should be shipped'
pool = Pool()
Uom = pool.get('product.uom')
if self.line.sale.shipment_method == 'order':
return abs(self.quantity)
elif self.line.sale.shipment_method == 'invoice':
quantity = 0.0
for invoice_line in self.line.invoice_lines:
if (invoice_line.invoice
and invoice_line.invoice.state == 'paid'):
quantity += Uom.compute_qty(invoice_line.unit,
invoice_line.quantity, self.line.unit)
return self.quantity * quantity / self.line.quantity
@property
def _move_remaining_quantity(self):
pool = Pool()
UoM = pool.get('product.uom')
quantity = super()._move_remaining_quantity
if self.line.sale.shipment_method == 'invoice':
invoices_ignored = set(self.line.sale.invoices_ignored)
ratio = self.get_moved_ratio()
if ratio:
for invoice_line in self.line.invoice_lines:
if invoice_line.invoice in invoices_ignored:
quantity -= UoM.compute_qty(
invoice_line.unit * self.line.unit,
invoice_line.quantity / ratio,
self.line.unit)
return quantity
def check_move_quantity(self):
from trytond.modules.sale.exceptions import SaleMoveQuantity
pool = Pool()
Lang = pool.get('ir.lang')
Warning = pool.get('res.user.warning')
lang = Lang.get()
move_type = 'in' if self.quantity >= 0 else 'return'
quantity = (
self._get_move_quantity(move_type)
- self._get_shipped_quantity(move_type))
if quantity < 0:
warning_name = Warning.format(
'check_move_quantity', [self])
if Warning.check(warning_name):
raise SaleMoveQuantity(warning_name, gettext(
'sale.msg_sale_line_move_quantity',
line=self.rec_name,
extra=lang.format_number_symbol(
-quantity, self.unit),
quantity=lang.format_number_symbol(
self.quantity, self.unit)))
class LineComponentIgnoredMove(ModelSQL):
__name__ = 'sale.line.component-ignored-stock.move'
component = fields.Many2One(
'sale.line.component', "Component",
ondelete='CASCADE', required=True)
move = fields.Many2One(
'stock.move', "Move", ondelete='RESTRICT', required=True)
class LineComponentRecreatedMove(ModelSQL):
__name__ = 'sale.line.component-recreated-stock.move'
component = fields.Many2One(
'sale.line.component', "Sale Line Component",
ondelete='CASCADE', required=True)
move = fields.Many2One(
'stock.move', "Move", ondelete='RESTRICT', required=True)
class HandleShipmentException(
handle_shipment_exception_mixin('sale'),
metaclass=PoolMeta):
__name__ = 'sale.handle.shipment.exception'
class Amendment(metaclass=PoolMeta):
__name__ = 'sale.amendment'
@classmethod
def _stock_moves(cls, line):
yield from super()._stock_moves(line)
for component in line.components:
for move in component.moves:
if move.state == 'draft':
yield move
class AmendmentLine(AmendmentLineMixin, metaclass=PoolMeta):
__name__ = 'sale.amendment.line'