Skip to content

Commit 5cd8f39

Browse files
committed
finder method on repo [for_order_method]
1 parent 99b6bf2 commit 5cd8f39

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

src/allocation/adapters/repository.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -57,3 +57,11 @@ def _get_by_batchref(self, batchref):
5757
return self.session.query(model.Product).join(model.Batch).filter(
5858
orm.batches.c.reference == batchref,
5959
).first()
60+
61+
62+
def for_order(self, orderid):
63+
order_lines = self.session.query(model.OrderLine).filter_by(orderid=orderid)
64+
skus = {l.sku for l in order_lines}
65+
return self.session.query(model.Product).join(model.Batch).filter(
66+
model.Batch.sku.in_(skus)
67+
)

0 commit comments

Comments
 (0)