modify flask to add new view endpoint and return 202s

hjwp · hjwp · commit 7ac1cf496548 · 2020-02-21T12:15:28.000Z
diff --git a/src/allocation/entrypoints/flask_app.py b/src/allocation/entrypoints/flask_app.py
@@ -5,6 +5,7 @@
 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
 
 app = Flask(__name__)
 orm.start_mappers()
@@ -35,4 +36,13 @@ def allocate_endpoint():
     except InvalidSku as e:
         return jsonify({'message': str(e)}), 400
 
-    return jsonify({'batchref': batchref}), 201
+    return 'OK', 202
+
+
+@app.route("/allocations/<orderid>", methods=['GET'])
+def allocations_view_endpoint(orderid):
+    uow = unit_of_work.SqlAlchemyUnitOfWork()
+    result = views.allocations(orderid, uow)
+    if not result:
+        return 'not found', 404
+    return jsonify(result), 200
