new service to add a batch [add_batch_service]

hjwp · hjwp · commit 40bd56d3700e · 2020-01-08T20:41:36.000Z
diff --git a/services.py b/services.py
@@ -1,4 +1,6 @@
 from __future__ import annotations
+from typing import Optional
+from datetime import date
 
 import model
 from model import OrderLine
@@ -11,6 +13,15 @@ class InvalidSku(Exception):
 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,
+):
+    repo.add(model.Batch(ref, sku, qty, eta))
+    session.commit()
+
+
 def allocate(
         orderid: str, sku: str, qty: int, repo: AbstractRepository, session
 ) -> str:
