add list of not-submitted bugs

[utils.git] / src / budget_sync / budget_graph.py

diff --git a/src/budget_sync/budget_graph.py b/src/budget_sync/budget_graph.py
index 483f36ec2db89bc9e6fa3e0b9a02668f374b17e6..de1bed8c7f9d88f7e70fb97c5bb0cd0eb2ae89ee 100644 (file)
--- a/src/budget_sync/budget_graph.py
+++ b/src/budget_sync/budget_graph.py
@@ -235,6 +235,7 @@ class PaymentSummary:
         self.total_submitted = Money(0)
         self.submitted_date = None
         self.paid_date = None
+        self.not_submitted = []
         summary_state = None
         for payment in self.payments:
             if summary_state is None:
@@ -255,11 +256,15 @@ class PaymentSummary:
                 self.total_paid += payment.amount
             else:
                 assert payment.state is PayeeState.NotYetSubmitted
+                self.not_submitted.append(payment.node.bug.id)
         if summary_state is None:
             self.state = PaymentSummaryState.NotYetSubmitted
         else:
             self.state = summary_state
 
+    def get_not_submitted(self):
+        return self.not_submitted
+
     def __repr__(self) -> str:
         return (f"PaymentSummary(total={self.total}, "
                 f"total_paid={self.total_paid}, "