sort out short names in csv files

[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 164bd0a57791dceef340f612acc732af2a0a454c..c9319f9f6ea405c51daeef6fdc1a093bdeae45ce 100644 (file)
--- a/src/budget_sync/budget_graph.py
+++ b/src/budget_sync/budget_graph.py
@@ -808,7 +808,9 @@ class BudgetGraph:
             retval[milestone] = OrderedSet()
         for milestone, payments in self.milestone_payments.items():
             for payment in payments:
-                retval[milestone].add(str(payment.payee.identifier))
+                short_name = str(payment.payee.output_markdown_file)
+                short_name = short_name.replace(".mdwn", "")
+                retval[milestone].add(short_name)
         return retval
 
     def __repr__(self):