From 9b69786341d14fd4327b117a12437fd1650cd965 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Isma=C3=ABl=20Bouya?= Date: Thu, 3 May 2018 00:22:33 +0200 Subject: Include current portfolio currencies when printing balances --- market.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) (limited to 'market.py') diff --git a/market.py b/market.py index 6edd605..5876071 100644 --- a/market.py +++ b/market.py @@ -255,6 +255,7 @@ class Processor: "name": "print_balances", "number": 1, "fetch_balances": ["begin"], + "fetch_balances_args": { "add_portfolio": True }, "print_tickers": { "base_currency": "BTC" }, } ], @@ -386,15 +387,19 @@ class Processor: def process_step(self, scenario_name, step, kwargs): process_name = "process_{}__{}_{}".format(scenario_name, step["number"], step["name"]) self.market.report.log_stage("{}_begin".format(process_name)) + + fetch_args = step.get("fetch_balances_args", {}) if "begin" in step.get("fetch_balances", []): - self.market.balances.fetch_balances(tag="{}_begin".format(process_name), log_tickers=True) + self.market.balances.fetch_balances(tag="{}_begin".format(process_name), + log_tickers=True, **fetch_args) for action in self.ordered_actions: if action in step: self.run_action(action, step[action], kwargs) if "end" in step.get("fetch_balances", []): - self.market.balances.fetch_balances(tag="{}_end".format(process_name), log_tickers=True) + self.market.balances.fetch_balances(tag="{}_end".format(process_name), + log_tickers=True, **fetch_args) self.market.report.log_stage("{}_end".format(process_name)) def method_arguments(self, action): -- cgit v1.2.3