Continue implement deposit

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