From c11edf3d9a14e2b420787ae96797afd801b562b9 Mon Sep 17 00:00:00 2001 From: uvok Date: Fri, 18 Apr 2025 12:09:55 +0200 Subject: Add assertion, reformat --- ledger_process.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) (limited to 'ledger_process.py') diff --git a/ledger_process.py b/ledger_process.py index 51d4fa6..24a6aca 100644 --- a/ledger_process.py +++ b/ledger_process.py @@ -59,12 +59,16 @@ class LedgerProcess: elif action.type == "withdrawal" and action.asset != "EUR": assert len(actions) == 1 + assert action.amount < 0 + assert action.fee >= 0 currency = action.asset self.external_wallet.setdefault(currency, FIFOQueue()) self.fifo_queues.setdefault(currency, FIFOQueue()) - t = self.fifo_queues[currency].remove(lambda t: t.amount == -action.amount + action.fee) + t = self.fifo_queues[currency].remove( + lambda t: t.amount == -action.amount + action.fee + ) t.remove_coins(action.fee) self.external_wallet[currency].add_trade(t) -- cgit v1.2.3