use bootstrap in flask [flask_calls_bootstrap]

hjwp · hjwp · commit d0c9333be0a0 · 2020-03-05T19:59:44.000Z
diff --git a/src/allocation/entrypoints/flask_app.py b/src/allocation/entrypoints/flask_app.py
@@ -1,14 +1,11 @@
 from datetime import datetime
 from flask import Flask, jsonify, request
-
 from allocation.domain import commands
-from allocation.adapters import orm
-from allocation.service_layer import messagebus, unit_of_work
 from allocation.service_layer.handlers import InvalidSku
-from allocation import views
+from allocation import bootstrap, views
 
 app = Flask(__name__)
-orm.start_mappers()
+bus = bootstrap.bootstrap()
 
 
 @app.route("/add_batch", methods=['POST'])
@@ -19,8 +16,7 @@ def add_batch():
     cmd = commands.CreateBatch(
         request.json['ref'], request.json['sku'], request.json['qty'], eta,
     )
-    uow = unit_of_work.SqlAlchemyUnitOfWork()
-    messagebus.handle(cmd, uow)
+    bus.handle(cmd)
     return 'OK', 201
 
 
@@ -30,8 +26,7 @@ def allocate_endpoint():
         cmd = commands.Allocate(
             request.json['orderid'], request.json['sku'], request.json['qty'],
         )
-        uow = unit_of_work.SqlAlchemyUnitOfWork()
-        messagebus.handle(cmd, uow)
+        bus.handle(cmd)
     except InvalidSku as e:
         return jsonify({'message': str(e)}), 400
 
@@ -40,8 +35,7 @@ def allocate_endpoint():
 
 @app.route("/allocations/<orderid>", methods=['GET'])
 def allocations_view_endpoint(orderid):
-    uow = unit_of_work.SqlAlchemyUnitOfWork()
-    result = views.allocations(orderid, uow)
+    result = views.allocations(orderid, bus.uow)
     if not result:
         return 'not found', 404
     return jsonify(result), 200
