messagebus takes a uow [handle_takes_uow]

hjwp · hjwp · commit 9716fe08d27c · 2020-02-21T12:05:48.000Z
diff --git a/src/allocation/service_layer/messagebus.py b/src/allocation/service_layer/messagebus.py
@@ -1,11 +1,14 @@
-from typing import List, Dict, Callable, Type
+from __future__ import annotations
+from typing import List, Dict, Callable, Type, TYPE_CHECKING
 from allocation.domain import events
 from . import handlers
+if TYPE_CHECKING:
+    from allocation.service_layer import unit_of_work
 
 
-def handle(event: events.Event):
+def handle(event: events.Event, uow: unit_of_work.AbstractUnitOfWork):
     for handler in HANDLERS[type(event)]:
-        handler(event)
+        handler(event, uow=uow)
 
 
 HANDLERS = {
