diff options
| -rw-r--r-- | ledger_process.py | 13 |
1 files changed, 10 insertions, 3 deletions
diff --git a/ledger_process.py b/ledger_process.py index 8f766be..d5199a6 100644 --- a/ledger_process.py +++ b/ledger_process.py @@ -48,7 +48,7 @@ class LedgerProcess: assert len(actions) == 1 self._process_withdrawal(action) - def _process_deposit(self, action): + def _process_deposit(self, action: LedgerAction): assert action.amount > 0 assert action.fee >= 0 @@ -56,11 +56,18 @@ class LedgerProcess: self.external_wallet.setdefault(currency, FIFOQueue()) self.fifo_queues.setdefault(currency, FIFOQueue()) - self.external_wallet[currency].remove(lambda t: t.amount == action.amount) + t = self.external_wallet[currency].remove(lambda t: t.amount == action.amount) + + # TODO: Fee handling + assert action.fee == 0 + wallet = self.fifo_queues[currency] + wallet.add_trade(t) + + # TODO: Sort logger.error("Don't know how do handle deposits yet.") - def _process_withdrawal(self, action): + def _process_withdrawal(self, action: LedgerAction): assert action.amount < 0 assert action.fee >= 0 |