make Batches sortable [dunder_gt]

hjwp · hjwp · commit e2e7bc573f35 · 2019-07-18T22:07:56.000+01:00
diff --git a/model.py b/model.py
@@ -29,6 +29,9 @@ def __init__(
         self._purchased_quantity = qty
         self._allocations = set()  # type: Set[OrderLine]
 
+    def __repr__(self):
+        return f'<Batch {self.reference}>'
+
     def __eq__(self, other):
         if not isinstance(other, Batch):
             return False
@@ -37,6 +40,13 @@ def __eq__(self, other):
     def __hash__(self):
         return hash(self.reference)
 
+    def __gt__(self, other):
+        if self.eta is None:
+            return False
+        if other.eta is None:
+            return True
+        return self.eta > other.eta
+
     def allocate(self, line: OrderLine):
         if self.can_allocate(line):
             self._allocations.add(line)
