finder method on repo [for_order_method]

hjwp · hjwp · commit 38f89db43944 · 2021-02-24T20:08:31.000Z
diff --git a/src/allocation/adapters/repository.py b/src/allocation/adapters/repository.py
@@ -52,6 +52,17 @@ def _get_by_batchref(self, batchref):
         return (
             self.session.query(model.Product)
             .join(model.Batch)
-            .filter(orm.batches.c.reference == batchref)
+            .filter(
+                orm.batches.c.reference == batchref,
+            )
             .first()
         )
+
+    def for_order(self, orderid):
+        order_lines = self.session.query(model.OrderLine).filter_by(orderid=orderid)
+        skus = {l.sku for l in order_lines}
+        return (
+            self.session.query(model.Product)
+            .join(model.Batch)
+            .filter(model.Batch.sku.in_(skus))
+        )
