From 3fd5c7ce8ec71dd74fc3f4c72ab99463c6f911a7 Mon Sep 17 00:00:00 2001 From: uvok Date: Thu, 17 Apr 2025 18:43:38 +0200 Subject: sort-order ledger processing --- bla.py | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/bla.py b/bla.py index 4595bcb..a915102 100644 --- a/bla.py +++ b/bla.py @@ -90,10 +90,15 @@ def process_trade(refid: str, trades: List[LedgerAction]): def process_ledger(read_actions: List[LedgerAction], output_path: str): - # for refid, actions in groupby( - # sorted(read_actions, key=lambda a: a.refid), lambda a: a.refid - # ): - for refid, actions in groupby(read_actions, lambda a: a.refid): + # don't make any assumtptions about ledger sorting + # groupby requires sorted inputs + actions_sorted = sorted(read_actions, key=lambda a: (a.refid, a.timestamp)) + _grouped_actions = groupby(actions_sorted, lambda a: a.refid) + _list_tuple_actions = [(k, list(v)) for k, v in _grouped_actions] + # finally, sort groupy by first available timestamp + sorted_grouped_actions = sorted(_list_tuple_actions, key=lambda a: a[1][0].timestamp) + + for refid, actions in sorted_grouped_actions: actions = list(actions) if len(actions) == 0: -- cgit v1.2.3