From: Ismaël Bouya Date: Mon, 22 Jan 2018 00:04:30 +0000 (+0100) Subject: Allow dispatch asset to have custom repartition X-Git-Tag: v0.1~32 X-Git-Url: https://git.immae.eu/?p=perso%2FImmae%2FProjets%2FCryptomonnaies%2FCryptoportfolio%2FTrader.git;a=commitdiff_plain;h=7ab23e291602b6ffd54622cffdc76ba0b4437a05 Allow dispatch asset to have custom repartition --- 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))