first commit

modules/stock/ir.py

@@ -0,0 +1,57 @@
+# 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 itertools import groupby
+from operator import attrgetter
+
+from trytond.pool import Pool, PoolMeta
+from trytond.transaction import Transaction
+
+from .shipment import ShipmentAssignMixin
+
+
+class Cron(metaclass=PoolMeta):
+    __name__ = 'ir.cron'
+
+    @classmethod
+    def __setup__(cls):
+        super().__setup__()
+        cls.method.selection.extend([
+                ('product.product|recompute_cost_price_from_moves',
+                    "Recompute Cost Price from Moves"),
+                ('stock.shipment.out|reschedule',
+                    "Reschedule Customer Shipments"),
+                ('stock.shipment.in.return|reschedule',
+                    "Reschedule Supplier Return Shipments"),
+                ('stock.shipment.internal|reschedule',
+                    "Reschedule Internal Shipments"),
+                ('ir.cron|stock_shipment_assign_try', "Assign Shipments"),
+                ])
+
+    @classmethod
+    def __register__(cls, module):
+        table = cls.__table__()
+        cursor = Transaction().connection.cursor()
+
+        super().__register__(module)
+
+        # Migration from 7.0: replace assign_cron
+        cursor.execute(*table.update(
+                [table.method], ['ir.cron|stock_shipment_assign_try'],
+                where=table.method.in_([
+                        'stock.shipment.out|assign_cron',
+                        'stock.shipment.internal|assign_cron',
+                        ])))
+
+    @classmethod
+    def stock_shipment_assign_try(cls):
+        pool = Pool()
+        records = []
+        for _, kls in pool.iterobject():
+            if issubclass(kls, ShipmentAssignMixin):
+                records.extend(kls.to_assign())
+
+        records.sort(key=attrgetter('assign_order_key'))
+
+        for kls, records in groupby(records, key=attrgetter('__class__')):
+            kls.assign_try(list(records))