From 7ab23e291602b6ffd54622cffdc76ba0b4437a05 Mon Sep 17 00:00:00 2001 From: =?utf8?q?Isma=C3=ABl=20Bouya?= Date: Mon, 22 Jan 2018 01:04:30 +0100 Subject: [PATCH] Allow dispatch asset to have custom repartition --- portfolio.py | 17 +++++++++++++---- 1 file changed, 13 insertions(+), 4 deletions(-) diff --git a/portfolio.py b/portfolio.py index 946a96a..20f83e4 100644 --- a/portfolio.py +++ b/portfolio.py @@ -205,11 +205,12 @@ class Balance: return cls.known_balances @classmethod - def dispatch_assets(cls, amount): - repartition_pertenthousand = Portfolio.repartition_pertenthousand() - sum_pertenthousand = sum([v for k, v in repartition_pertenthousand.items()]) + def dispatch_assets(cls, amount, repartition=None): + if repartition is None: + repartition = Portfolio.repartition_pertenthousand() + sum_pertenthousand = sum([v for k, v in repartition.items()]) amounts = {} - for currency, ptt in repartition_pertenthousand.items(): + for currency, ptt in repartition.items(): amounts[currency] = ptt * amount / sum_pertenthousand if currency not in cls.known_balances: cls.known_balances[currency] = cls(currency, 0, 0, 0) @@ -233,6 +234,14 @@ class Balance: new_repartition = cls.dispatch_assets(total_base_value) Trade.compute_trades(values_in_base, new_repartition, only=only, market=market) + @classmethod + def prepare_trades_to_sell_all(cls, market, base_currency="BTC", compute_value="average"): + cls.fetch_balances(market) + values_in_base = cls.in_currency(base_currency, market, compute_value=compute_value) + total_base_value = sum(values_in_base.values()) + new_repartition = cls.dispatch_assets(total_base_value, repartition={ base_currency: 1 }) + Trade.compute_trades(values_in_base, new_repartition, market=market) + def __repr__(self): return "Balance({} [{}/{}/{}])".format(self.currency, str(self.free), str(self.used), str(self.total)) -- 2.41.0