modify flask to use messagebus [flask_uses_messagebus]

hjwp · hjwp · commit aaa066f44773 · 2024-03-01T15:05:12.000Z
diff --git a/src/allocation/entrypoints/flask_app.py b/src/allocation/entrypoints/flask_app.py
@@ -1,9 +1,10 @@
 from datetime import datetime
 from flask import Flask, request
 
+from allocation.domain import events
 from allocation.adapters import orm
-from allocation.service_layer import services, unit_of_work
-from allocation.service_layer.services import InvalidSku
+from allocation.service_layer import messagebus, unit_of_work
+from allocation.service_layer.handlers import InvalidSku
 
 app = Flask(__name__)
 orm.start_mappers()
@@ -14,25 +15,21 @@ def add_batch():
     eta = request.json["eta"]
     if eta is not None:
         eta = datetime.fromisoformat(eta).date()
-    services.add_batch(
-        request.json["ref"],
-        request.json["sku"],
-        request.json["qty"],
-        eta,
-        unit_of_work.SqlAlchemyUnitOfWork(),
+    event = events.BatchCreated(
+        request.json["ref"], request.json["sku"], request.json["qty"], eta
     )
+    messagebus.handle(event, unit_of_work.SqlAlchemyUnitOfWork())
     return "OK", 201
 
 
 @app.route("/allocate", methods=["POST"])
 def allocate_endpoint():
     try:
-        batchref = services.allocate(
-            request.json["orderid"],
-            request.json["sku"],
-            request.json["qty"],
-            unit_of_work.SqlAlchemyUnitOfWork(),
+        event = events.AllocationRequired(
+            request.json["orderid"], request.json["sku"], request.json["qty"]
         )
+        results = messagebus.handle(event, unit_of_work.SqlAlchemyUnitOfWork())
+        batchref = results.pop(0)
     except InvalidSku as e:
         return {"message": str(e)}, 400
 
