Skip to content

Commit 6b404b5

Browse files
committed
add api endpoint for add_batch [api_for_add_batch]
1 parent 96301d2 commit 6b404b5

File tree

1 file changed

+19
-1
lines changed

1 file changed

+19
-1
lines changed

entrypoints/flask_app.py

Lines changed: 19 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
from datetime import datetime
12
from flask import Flask, request
23
from sqlalchemy import create_engine
34
from sqlalchemy.orm import sessionmaker
@@ -7,12 +8,29 @@
78
from adapters import orm, repository
89
from service_layer import services
910

10-
1111
orm.start_mappers()
1212
get_session = sessionmaker(bind=create_engine(config.get_postgres_uri()))
1313
app = Flask(__name__)
1414

1515

16+
@app.route("/add_batch", methods=["POST"])
17+
def add_batch():
18+
session = get_session()
19+
repo = repository.SqlAlchemyRepository(session)
20+
eta = request.json["eta"]
21+
if eta is not None:
22+
eta = datetime.fromisoformat(eta).date()
23+
services.add_batch(
24+
request.json["ref"],
25+
request.json["sku"],
26+
request.json["qty"],
27+
eta,
28+
repo,
29+
session,
30+
)
31+
return "OK", 201
32+
33+
1634
@app.route("/allocate", methods=["POST"])
1735
def allocate_endpoint():
1836
session = get_session()

0 commit comments

Comments
 (0)