start on handler for change quantity [change_quantity_handler]

hjwp · hjwp · commit eb11e04e72c2 · 2019-11-20T21:12:36.000Z
diff --git a/src/allocation/handlers.py b/src/allocation/handlers.py
@@ -1,12 +1,11 @@
 from __future__ import annotations
-import typing
+from typing import TYPE_CHECKING
 from allocation import events, email, exceptions, model
 from allocation.model import OrderLine
-if typing.TYPE_CHECKING:
+if TYPE_CHECKING:
     from allocation import unit_of_work
 
 
-
 def add_batch(
         event: events.BatchCreated, uow: unit_of_work.AbstractUnitOfWork
 ):
@@ -34,6 +33,16 @@ def allocate(
         return batchref
 
 
+
+def change_batch_quantity(
+        event: events.BatchQuantityChanged, uow: unit_of_work.AbstractUnitOfWork
+):
+    with uow:
+        product = uow.products.get_by_batchref(batchref=event.ref)
+        product.change_batch_quantity(ref=event.ref, qty=event.qty)
+        uow.commit()
+
+
 # pylint: disable=unused-argument
 
 def send_out_of_stock_notification(
diff --git a/src/allocation/messagebus.py b/src/allocation/messagebus.py
@@ -16,6 +16,8 @@ def handle(event: events.Event, uow: unit_of_work.AbstractUnitOfWork):
 
 HANDLERS = {
     events.BatchCreated: [handlers.add_batch],
+    events.BatchQuantityChanged: [handlers.change_batch_quantity],
     events.AllocationRequired: [handlers.allocate],
     events.OutOfStock: [handlers.send_out_of_stock_notification],
+
 }  # type: Dict[Type[events.Event], List[Callable]]
