first commit

modules/product_kit/account.py

@@ -0,0 +1,61 @@
+# 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.model import ModelView, Workflow
+from trytond.pool import Pool, PoolMeta
+from trytond.transaction import Transaction
+
+
+class Invoice(metaclass=PoolMeta):
+    __name__ = 'account.invoice'
+
+    @classmethod
+    @ModelView.button
+    @Workflow.transition('posted')
+    def post(cls, invoices):
+        pool = Pool()
+        Move = pool.get('stock.move')
+        transaction = Transaction()
+        context = transaction.context
+        super().post(invoices)
+        moves = sum(
+            (l.stock_component_moves for i in invoices for l in i.lines), [])
+        if moves:
+            with transaction.set_context(
+                    queue_batch=context.get('queue_batch', True)):
+                Move.__queue__.update_unit_price(moves)
+
+
+class InvoiceLine(metaclass=PoolMeta):
+    __name__ = 'account.invoice.line'
+
+    @property
+    def stock_component_moves(self):
+        return []
+
+
+class InvoiceLineSale(metaclass=PoolMeta):
+    __name__ = 'account.invoice.line'
+
+    @property
+    def stock_component_moves(self):
+        pool = Pool()
+        SaleLine = pool.get('sale.line')
+        moves = super().stock_component_moves
+        if isinstance(self.origin, SaleLine):
+            for component in self.origin.components:
+                moves.extend(component.moves)
+        return moves
+
+
+class InvoiceLinePurchase(metaclass=PoolMeta):
+    __name__ = 'account.invoice.line'
+
+    @property
+    def stock_component_moves(self):
+        pool = Pool()
+        PurchaseLine = pool.get('purchase.line')
+        moves = super().stock_component_moves
+        if isinstance(self.origin, PurchaseLine):
+            for component in self.origin.components:
+                moves.extend(component.moves)
+        return moves