diff --git a/freqtrade/persistence/trade_model.py b/freqtrade/persistence/trade_model.py index c3532c783..c1fbbc30c 100644 --- a/freqtrade/persistence/trade_model.py +++ b/freqtrade/persistence/trade_model.py @@ -1351,20 +1351,15 @@ class LocalTrade: """ CustomDataWrapper.set_custom_data(trade_id=self.id, key=key, value=value) - def get_custom_data(self, key: str, default: Any = None, retrieval_mode: str = "value") -> Any: + def get_custom_data(self, key: str, default: Any = None) -> Any: """ Get custom data for this trade. :param key: key of the custom data :param default: value to return if no data is found - :param retrieval_mode: 'value' (default) to return the custom data's value, - or 'object' to return the entire custom data object. """ data = CustomDataWrapper.get_custom_data(trade_id=self.id, key=key) if data: - if retrieval_mode == "object": - return data[0] - # default behavior: return only the value return data[0].value return default diff --git a/freqtrade/rpc/rpc.py b/freqtrade/rpc/rpc.py index e4ec1a6b5..3649adbf6 100644 --- a/freqtrade/rpc/rpc.py +++ b/freqtrade/rpc/rpc.py @@ -1145,7 +1145,7 @@ class RPC: for trade in trades: # Depending on whether a specific key is provided, retrieve custom data accordingly. if key: - data = trade.get_custom_data(key=key, retrieval_mode="object") + data = trade.get_custom_data_entry(key=key) # If data exists, wrap it in a list so the output remains consistent. custom_data = [data] if data else [] else: