Skip to content

Commit 46ee1ec

Browse files
techfreaqueGuillaumeDSM
authored andcommitted
[TradingMetadata] allow overriding with get_additional_metadata
1 parent 5a4ae08 commit 46ee1ec

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

octobot/storage/trading_metadata.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -55,17 +55,20 @@ async def store_backtesting_run_metadata(exchange_managers, start_time, user_inp
5555

5656
async def _get_trading_metadata(exchange_managers, run_start_time, user_inputs, run_dbs_identifier, is_backtesting) \
5757
-> dict:
58+
trading_mode = trading_api.get_trading_modes(exchange_managers[0])[0]
5859
multi_exchanges_data = await _get_multi_exchange_data(exchange_managers, is_backtesting)
5960
single_exchange_data = await _get_single_exchange_data(
6061
exchange_managers[0],
62+
trading_mode,
6163
run_start_time,
6264
user_inputs,
6365
run_dbs_identifier,
6466
is_backtesting
6567
)
6668
return {
6769
**single_exchange_data,
68-
**multi_exchanges_data
70+
**multi_exchanges_data,
71+
**(await trading_mode.get_additional_metadata(is_backtesting))
6972
}
7073

7174

@@ -213,9 +216,7 @@ async def _get_multi_exchange_data(exchange_managers, is_backtesting):
213216
}
214217

215218

216-
async def _get_single_exchange_data(exchange_manager, run_start_time, user_inputs, run_dbs_identifier, is_backtesting):
217-
trading_mode = trading_api.get_trading_modes(exchange_manager)[0]
218-
219+
async def _get_single_exchange_data(exchange_manager, trading_mode, run_start_time, user_inputs, run_dbs_identifier, is_backtesting):
219220
exchange_type = trading_enums.ExchangeTypes.FUTURE.value if exchange_manager.is_future \
220221
else trading_enums.ExchangeTypes.SPOT.value
221222
if user_inputs is None:
@@ -246,6 +247,5 @@ async def _get_single_exchange_data(exchange_manager, run_start_time, user_input
246247
common_enums.BacktestingMetadata.LEVERAGE.value: leverage,
247248
common_enums.DBRows.TRADING_TYPE.value: exchange_type,
248249
common_enums.DBRows.REFERENCE_MARKET.value: trading_api.get_reference_market(exchange_manager.config),
249-
},
250-
**(await trading_mode.get_additional_metadata(is_backtesting))
250+
}
251251
}

0 commit comments

Comments
 (0)