Allow dispatch assets to receive liquidity

[perso/Immae/Projets/Cryptomonnaies/Cryptoportfolio/Trader.git] / store.py

diff --git a/store.py b/store.py
index 841a0fc9e4f581ad468a2c3e8664507b003b4232..abbe5eef556eb88455aec50f12d07b041c0f5c67 100644 (file)
--- a/store.py
+++ b/store.py
@@ -27,9 +27,9 @@ class BalanceStore:
                 cls.all[currency] = portfolio.Balance(currency, balance)
 
     @classmethod
-    def dispatch_assets(cls, amount, repartition=None):
+    def dispatch_assets(cls, amount, liquidity="medium", repartition=None):
         if repartition is None:
-            repartition = portfolio.Portfolio.repartition()
+            repartition = portfolio.Portfolio.repartition(liquidity=liquidity)
         sum_ratio = sum([v[0] for k, v in repartition.items()])
         amounts = {}
         for currency, (ptt, trade_type) in repartition.items():