bring static handlers dicts across from messagebus

hjwp · hjwp · commit 36c411069ba4 · 2020-03-05T19:58:45.000Z
diff --git a/src/allocation/service_layer/handlers.py b/src/allocation/service_layer/handlers.py
@@ -1,7 +1,7 @@
 #pylint: disable=unused-argument
 from __future__ import annotations
 from dataclasses import asdict
-from typing import Callable, TYPE_CHECKING
+from typing import List, Dict, Callable, Type, TYPE_CHECKING
 from allocation.domain import commands, events, model
 from allocation.domain.model import OrderLine
 if TYPE_CHECKING:
@@ -92,3 +92,16 @@ def remove_allocation_from_read_model(
             dict(orderid=event.orderid, sku=event.sku)
         )
         uow.commit()
+
+
+EVENT_HANDLERS = {
+    events.Allocated: [publish_allocated_event, add_allocation_to_read_model],
+    events.Deallocated: [remove_allocation_from_read_model, reallocate],
+    events.OutOfStock: [send_out_of_stock_notification],
+}  # type: Dict[Type[events.Event], List[Callable]]
+
+COMMAND_HANDLERS = {
+    commands.Allocate: allocate,
+    commands.CreateBatch: add_batch,
+    commands.ChangeBatchQuantity: change_batch_quantity,
+}  # type: Dict[Type[commands.Command], Callable]
diff --git a/src/allocation/service_layer/messagebus.py b/src/allocation/service_layer/messagebus.py
@@ -1,7 +1,7 @@
 # pylint: disable=broad-except
 from __future__ import annotations
 import logging
-from typing import List, Dict, Callable, Type, Union, TYPE_CHECKING
+from typing import Union, TYPE_CHECKING
 from allocation.domain import commands, events
 from . import handlers
 
@@ -53,22 +53,3 @@ def handle_command(
     except Exception:
         logger.exception('Exception handling command %s', command)
         raise
-
-
-EVENT_HANDLERS = {
-    events.Allocated: [
-        handlers.publish_allocated_event,
-        handlers.add_allocation_to_read_model
-    ],
-    events.Deallocated: [
-        handlers.remove_allocation_from_read_model,
-        handlers.reallocate,
-    ],
-    events.OutOfStock: [handlers.send_out_of_stock_notification],
-}  # type: Dict[Type[events.Event], List[Callable]]
-
-COMMAND_HANDLERS = {
-    commands.Allocate: handlers.allocate,
-    commands.CreateBatch: handlers.add_batch,
-    commands.ChangeBatchQuantity: handlers.change_batch_quantity,
-}  # type: Dict[Type[commands.Command], Callable]
