new handlers that take a uow [change_batch_new_handlers]

hjwp · hjwp · commit fd79a8e40433 · 2019-07-26T16:31:50.000+01:00
diff --git a/src/allocation/messagebus.py b/src/allocation/messagebus.py
@@ -1,5 +1,5 @@
 from typing import List, Dict, Callable, Type
-from allocation import email, events
+from allocation import email, events, services, redis_pubsub
 
 
 def handle(events_: List[events.Event]):
@@ -9,15 +9,31 @@ def handle(events_: List[events.Event]):
             handler(event)
 
 
-def send_out_of_stock_notification(event: events.OutOfStock):
+def send_out_of_stock_notification(
+        event: events.OutOfStock, uow: unit_of_work.AbstractUnitOfWork
+):
     email.send_mail(
         'stock@made.com',
         f'Out of stock for {event.sku}',
     )
 
 
+def reallocate(
+        event: events.Deallocated, uow: unit_of_work.AbstractUnitOfWork
+):
+    services.allocate(event.orderid, event.sku, event.qty, uow=uow)
+
+
+def publish_allocated_event(
+        event: events.Allocated, uow: unit_of_work.AbstractUnitOfWork,
+):
+    redis_pubsub.publish('line_allocated', event)
+
+
 HANDLERS = {
     events.OutOfStock: [send_out_of_stock_notification],
+    events.Allocated: [publish_allocated_event],
+    events.Deallocated: [reallocate],
 
 }  # type: Dict[Type[events.Event], List[Callable]]
 
