Start moving to handlers [services_to_handlers]

hjwp · hjwp · commit 1fb57915a20a · 2024-03-01T15:05:12.000Z
diff --git a/src/allocation/adapters/email.py b/src/allocation/adapters/email.py
@@ -1,2 +1,2 @@
-def send_mail(*args):
+def send(*args):
     print("SENDING EMAIL:", *args)
diff --git a/src/allocation/service_layer/handlers.py b/src/allocation/service_layer/handlers.py
@@ -1,8 +1,6 @@
 from __future__ import annotations
-from typing import Optional, TYPE_CHECKING
-from datetime import date
-
-from allocation.domain import model
+from typing import TYPE_CHECKING
+from allocation.domain import events, model
 from allocation.domain.model import OrderLine
 
 if TYPE_CHECKING:
@@ -14,27 +12,42 @@ class InvalidSku(Exception):
 
 
 def add_batch(
-    ref: str, sku: str, qty: int, eta: Optional[date],
+    event: events.BatchCreated,
     uow: unit_of_work.AbstractUnitOfWork,
 ):
     with uow:
-        product = uow.products.get(sku=sku)
+        product = uow.products.get(sku=event.sku)
         if product is None:
-            product = model.Product(sku, batches=[])
+            product = model.Product(event.sku, batches=[])
             uow.products.add(product)
-        product.batches.append(model.Batch(ref, sku, qty, eta))
+        product.batches.append(
+            model.Batch(event.ref, event.sku, event.qty, event.eta)
+        )
         uow.commit()
 
 
 def allocate(
-    orderid: str, sku: str, qty: int,
+    event: events.AllocationRequired,
     uow: unit_of_work.AbstractUnitOfWork,
 ) -> str:
-    line = OrderLine(orderid, sku, qty)
+    line = OrderLine(event.orderid, event.sku, event.qty)
     with uow:
         product = uow.products.get(sku=line.sku)
         if product is None:
             raise InvalidSku(f"Invalid sku {line.sku}")
         batchref = product.allocate(line)
         uow.commit()
         return batchref
+
+
+# pylint: disable=unused-argument
+
+
+def send_out_of_stock_notification(
+    event: events.OutOfStock,
+    uow: unit_of_work.AbstractUnitOfWork,
+):
+    email.send(
+        "stock@made.com",
+        f"Out of stock for {event.sku}",
+    )
diff --git a/src/allocation/service_layer/messagebus.py b/src/allocation/service_layer/messagebus.py
@@ -1,20 +1,15 @@
 from typing import List, Dict, Callable, Type
-from allocation.adapters import email
 from allocation.domain import events
+from . import handlers
 
 
 def handle(event: events.Event):
     for handler in HANDLERS[type(event)]:
         handler(event)
 
 
-def send_out_of_stock_notification(event: events.OutOfStock):
-    email.send_mail(
-        "stock@made.com",
-        f"Out of stock for {event.sku}",
-    )
-
-
 HANDLERS = {
-    events.OutOfStock: [send_out_of_stock_notification],
+    events.BatchCreated: [handlers.add_batch],
+    events.AllocationRequired: [handlers.allocate],
+    events.OutOfStock: [handlers.send_out_of_stock_notification],
 }  # type: Dict[Type[events.Event], List[Callable]]

Original file line number	Diff line number	Diff line change
`@@ -1,2 +1,2 @@`
`1`		`-def send_mail(*args):`
	`1`	`+def send(*args):`
`2`	`2`	`print("SENDING EMAIL:", *args)`