new service to add a batch [add_batch_service]

hjwp · hjwp · commit 5c953dada5e1 · 2021-02-24T10:04:07.000Z
diff --git a/service_layer/services.py b/service_layer/services.py
@@ -1,4 +1,6 @@
 from __future__ import annotations
+from typing import Optional
+from datetime import date
 
 from domain import model
 from domain.model import OrderLine
@@ -13,6 +15,14 @@ def is_valid_sku(sku, batches):
     return sku in {b.sku for b in batches}
 
 
+def add_batch(
+    ref: str, sku: str, qty: int, eta: Optional[date],
+    repo: AbstractRepository, session,
+) -> None:
+    repo.add(model.Batch(ref, sku, qty, eta))
+    session.commit()
+
+
 def allocate(
     orderid: str, sku: str, qty: int,
     repo: AbstractRepository, session
