first commit

modules/incoterm/stock.py

@@ -0,0 +1,131 @@
+# 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 ModelView, Workflow, fields
+from trytond.pool import Pool, PoolMeta
+from trytond.pyson import Eval
+
+from .common import IncotermMixin
+from .exceptions import DifferentIncotermWarning
+
+
+class ShipmentMixin(IncotermMixin):
+
+    @property
+    @fields.depends('incoterm', 'incoterm_location')
+    def shipping_to(self):
+        party = super().shipping_to
+        if self.incoterm and self.incoterm_location:
+            party = self.incoterm_location.party
+        return party
+
+    @property
+    @fields.depends('incoterm', 'incoterm_location')
+    def shipping_to_address(self):
+        address = super().shipping_to_address
+        if self.incoterm and self.incoterm_location:
+            address = self.incoterm_location
+        return address
+
+
+class ShipmentIn(ShipmentMixin, metaclass=PoolMeta):
+    __name__ = 'stock.shipment.in'
+
+    @classmethod
+    def _incoterm_related_party(cls):
+        return Eval('supplier'), {'supplier'}
+
+
+class ShipmentIn_Purchase(metaclass=PoolMeta):
+    __name__ = 'stock.shipment.in'
+
+    @classmethod
+    @ModelView.button
+    @Workflow.transition('received')
+    def receive(cls, shipments):
+        pool = Pool()
+        Warning = pool.get('res.user.warning')
+        PurchaseLine = pool.get('purchase.line')
+        for shipment in shipments:
+            if shipment.incoterm:
+                incoterms = {
+                    move.origin.purchase.incoterm for move in shipment.moves
+                    if isinstance(move.origin, PurchaseLine)
+                    and move.state != 'cancelled'}
+                if {shipment.incoterm} != incoterms:
+                    incoterms.discard(shipment.incoterm)
+                    origin_incoterms = ', '.join(
+                        i.rec_name if i else '' for i in incoterms)
+                    warning_key = Warning.format(
+                            'different_incoterm', [shipment])
+                    if Warning.check(warning_key):
+                        raise DifferentIncotermWarning(
+                            warning_key,
+                            gettext('incoterm'
+                                '.msg_shipment_different_incoterm',
+                                shipment_incoterm=shipment.incoterm.rec_name,
+                                shipment=shipment.rec_name,
+                                origin_incoterms=origin_incoterms))
+        super().receive(shipments)
+
+
+class ShipmentInReturn(ShipmentMixin, metaclass=PoolMeta):
+    __name__ = 'stock.shipment.in.return'
+
+    @classmethod
+    def _incoterm_related_party(cls):
+        return Eval('supplier'), {'supplier'}
+
+
+class ShipmentOut(ShipmentMixin, metaclass=PoolMeta):
+    __name__ = 'stock.shipment.out'
+
+    @classmethod
+    def _incoterm_readonly_state(cls):
+        return Eval('state').in_(['cancelled', 'shipped,' 'done'])
+
+    @classmethod
+    def _incoterm_related_party(cls):
+        return Eval('customer'), {'customer'}
+
+
+class ShipmentOut_Sale(metaclass=PoolMeta):
+    __name__ = 'stock.shipment.out'
+
+    @classmethod
+    @ModelView.button
+    @Workflow.transition('waiting')
+    def wait(cls, shipments, moves=None):
+        pool = Pool()
+        Warning = pool.get('res.user.warning')
+        SaleLine = pool.get('sale.line')
+        for shipment in shipments:
+            if shipment.incoterm:
+                incoterms = {
+                    move.origin.sale.incoterm for move in shipment.moves
+                    if isinstance(move.origin, SaleLine)
+                    and move.state != 'cancelled'}
+                if {shipment.incoterm} != incoterms:
+                    incoterms.discard(shipment.incoterm)
+                    origin_incoterms = ', '.join(
+                        i.rec_name if i else '' for i in incoterms)
+                    warning_key = Warning.format(
+                            'different_incoterm', [shipment])
+                    if Warning.check(warning_key):
+                        raise DifferentIncotermWarning(
+                            warning_key,
+                            gettext('incoterm'
+                                '.msg_shipment_different_incoterm',
+                                shipment_incoterm=shipment.incoterm.rec_name,
+                                shipment=shipment.rec_name,
+                                origin_incoterms=origin_incoterms))
+        super().wait(shipments, moves)
+
+
+class ShipmentOutReturn(ShipmentMixin, metaclass=PoolMeta):
+    __name__ = 'stock.shipment.out.return'
+
+    @classmethod
+    def _incoterm_related_party(cls):
+        return Eval('customer'), {'customer'}