def postgres_session_factory(postgres_db):
start_mappers()
yield sessionmaker(bind=postgres_db)
clear_mappers()
from datetime import datetime
from flask import Flask, jsonify, request
from allocation import (
commands,
exceptions,
messagebus,
notifications,
orm,
redis_pubsub,
unit_of_work,
views,
)
app = Flask(__name__)
orm.start_mappers()
bus = messagebus.MessageBus(uow=unit_of_work.SqlAlchemyUnitOfWork(),
notifications=notifications.EmailNotifications(),
publish=redis_pubsub.publish)
@app.route("/add_batch", methods=['POST'])
def add_batch():
eta = request.json['eta']
if eta is not None:
eta = datetime.fromisoformat(eta).date()
command = commands.CreateBatch(
request.json['ref'],
request.json['sku'],
request.json['qty'],
eta,
)
def sqlite_session_factory(in_memory_sqlite_db):
start_mappers()
yield sessionmaker(bind=in_memory_sqlite_db)
clear_mappers()
