diff options
| -rw-r--r-- | ledger_process.py | 6 |
1 files changed, 5 insertions, 1 deletions
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) |