from datetime import datetime

from flask import Flask, request

from sqlalchemy import create_engine

from sqlalchemy.orm import sessionmaker

import config

from domain import model

from adapters import orm, repository

from service_layer import services

orm.start_mappers()

get_session = sessionmaker(bind=create_engine(config.get_postgres_uri()))

app = Flask(__name__)

@app.route("/add_batch", methods=["POST"])

def add_batch():

session = get_session()

repo = repository.SqlAlchemyRepository(session)

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,

repo,

session,

)

return "OK", 201

@app.route("/allocate", methods=["POST"])

def allocate_endpoint():

session = get_session()

repo = repository.SqlAlchemyRepository(session)

try:

batchref = services.allocate(

request.json["orderid"],

request.json["sku"],

request.json["qty"],

repo,

session,

)

except (model.OutOfStock, services.InvalidSku) as e:

return {"message": str(e)}, 400

return {"batchref": batchref}, 201