move repository and uow out into a file [appendix_csvs_ends]

hjwp · hjwp · commit 5988c6e45864 · 2021-02-24T21:32:40.000Z
diff --git a/src/allocation/service_layer/csv_uow.py b/src/allocation/service_layer/csv_uow.py
@@ -0,0 +1,66 @@
+# pylint: disable=too-few-public-methods, protected-access
+import csv
+from datetime import datetime
+from pathlib import Path
+from typing import Dict
+
+from allocation.domain import model
+from allocation.adapters import repository
+from allocation.service_layer import unit_of_work
+
+
+class CsvRepository(repository.AbstractRepository):
+    def __init__(self, folder):
+        self._batches_path = Path(folder) / "batches.csv"
+        self._allocations_path = Path(folder) / "allocations.csv"
+        self._batches = {}  # type: Dict[str, model.Batch]
+        self._load()
+
+    def get(self, reference):
+        return self._batches.get(reference)
+
+    def add(self, batch):
+        self._batches[batch.reference] = batch
+
+    def _load(self):
+        with self._batches_path.open() as f:
+            reader = csv.DictReader(f)
+            for row in reader:
+                ref, sku = row["ref"], row["sku"]
+                qty = int(row["qty"])
+                if row["eta"]:
+                    eta = datetime.strptime(row["eta"], "%Y-%m-%d").date()
+                else:
+                    eta = None
+                self._batches[ref] = model.Batch(ref=ref, sku=sku, qty=qty, eta=eta)
+        if self._allocations_path.exists() is False:
+            return
+        with self._allocations_path.open() as f:
+            reader = csv.DictReader(f)
+            for row in reader:
+                batchref, orderid, sku = row["batchref"], row["orderid"], row["sku"]
+                qty = int(row["qty"])
+                line = model.OrderLine(orderid, sku, qty)
+                batch = self._batches[batchref]
+                batch._allocations.add(line)
+
+    def list(self):
+        return list(self._batches.values())
+
+
+class CsvUnitOfWork(unit_of_work.AbstractUnitOfWork):
+    def __init__(self, folder):
+        self.batches = CsvRepository(folder)
+
+    def commit(self):
+        with self.batches._allocations_path.open("w") as f:
+            writer = csv.writer(f)
+            writer.writerow(["orderid", "sku", "qty", "batchref"])
+            for batch in self.batches.list():
+                for line in batch._allocations:
+                    writer.writerow(
+                        [line.orderid, line.sku, line.qty, batch.reference]
+                    )
+
+    def rollback(self):
+        pass
diff --git a/src/bin/allocate-from-csv b/src/bin/allocate-from-csv
@@ -1,76 +1,14 @@
 #!/usr/bin/env python
-from __future__ import annotations
-from typing import Dict
 import csv
 import sys
-from datetime import datetime
 from pathlib import Path
 
-from allocation.domain import model
-from allocation.service_layer import services, unit_of_work
-from allocation.adapters import repository
-
-
-class CsvRepository(repository.AbstractRepository):
-    def __init__(self, folder):
-        self._batches_path = Path(folder) / "batches.csv"
-        self._allocations_path = Path(folder) / "allocations.csv"
-        self._batches = {}  # type: Dict[str, model.Batch]
-        self._load()
-
-    def get(self, reference):
-        return self._batches.get(reference)
-
-    def add(self, batch):
-        self._batches[batch.reference] = batch
-
-    def _load(self):
-        with self._batches_path.open() as f:
-            reader = csv.DictReader(f)
-            for row in reader:
-                ref, sku = row["ref"], row["sku"]
-                qty = int(row["qty"])
-                if row["eta"]:
-                    eta = datetime.strptime(row["eta"], "%Y-%m-%d").date()
-                else:
-                    eta = None
-                self._batches[ref] = model.Batch(ref=ref, sku=sku, qty=qty, eta=eta)
-        if self._allocations_path.exists() is False:
-            return
-        with self._allocations_path.open() as f:
-            reader = csv.DictReader(f)
-            for row in reader:
-                batchref, orderid, sku = row["batchref"], row["orderid"], row["sku"]
-                qty = int(row["qty"])
-                line = model.OrderLine(orderid, sku, qty)
-                batch = self._batches[batchref]
-                batch._allocations.add(line)
-
-    def list(self):
-        return list(self._batches.values())
-
-
-class CsvUnitOfWork(unit_of_work.AbstractUnitOfWork):
-    def __init__(self, folder):
-        self.batches = CsvRepository(folder)
-
-    def commit(self):
-        with self.batches._allocations_path.open("w") as f:
-            writer = csv.writer(f)
-            writer.writerow(["orderid", "sku", "qty", "batchref"])
-            for batch in self.batches.list():
-                for line in batch._allocations:
-                    writer.writerow(
-                        [line.orderid, line.sku, line.qty, batch.reference]
-                    )
-
-    def rollback(self):
-        pass
+from allocation.service_layer import csv_uow, services
 
 
 def main(folder):
     orders_path = Path(folder) / "orders.csv"
-    uow = CsvUnitOfWork(folder)
+    uow = csv_uow.CsvUnitOfWork(folder)
     with orders_path.open() as f:
         reader = csv.DictReader(f)
         for row in reader:
