messagebus takes a uow [handle_takes_uow]

hjwp · hjwp · commit 541ab52185cb · 2024-03-01T15:05:12.000Z
diff --git a/src/allocation/service_layer/messagebus.py b/src/allocation/service_layer/messagebus.py
@@ -1,11 +1,18 @@
-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 = {
