order is now a dict too

hjwp · hjwp · commit 019760316e24 · 2019-02-05T11:15:41.000Z
diff --git a/domain_model.py b/domain_model.py
@@ -1,5 +1,3 @@
-from dataclasses import dataclass
-
 class Allocation(dict):
 
     @property
@@ -10,7 +8,7 @@ def skus(self):
     def for_(order, source):
         return Allocation({
             sku: source
-            for sku, quantity in order.lines.items()
+            for sku, quantity in order.items()
             if source.can_allocate(sku, quantity)
         })
 
@@ -27,14 +25,11 @@ def fully_allocates(self, order):
 
 
 
-@dataclass
-class Order:
-    lines: dict
-    allocation: Allocation = None
+class Order(dict):
 
     @property
     def skus(self):
-        return self.lines.keys()
+        return self.keys()
 
     @property
     def fully_allocated(self):
