use repository and go via Product [view_using_repo]

hjwp · hjwp · commit ccbaa5a231de · 2020-03-05T19:57:00.000Z
diff --git a/src/allocation/views.py b/src/allocation/views.py
@@ -2,12 +2,10 @@
 
 def allocations(orderid: str, uow: unit_of_work.SqlAlchemyUnitOfWork):
     with uow:
-        results = list(uow.session.execute(
-            'SELECT ol.sku, b.reference'
-            ' FROM allocations AS a'
-            ' JOIN batches AS b ON a.batch_id = b.id'
-            ' JOIN order_lines AS ol ON a.orderline_id = ol.id'
-            ' WHERE ol.orderid = :orderid',
-            dict(orderid=orderid)
-        ))
-    return [{'sku': sku, 'batchref': batchref} for sku, batchref in results]
+        products = uow.products.for_order(orderid=orderid)
+        batches = [b for p in products for b in p.batches]
+        return [
+            {'sku': b.sku, 'batchref': b.reference}
+            for b in batches
+            if orderid in b.orderids
+        ]
