CTYPTO/freqtrade
1
0
Fork 0
mirror of https://github.com/freqtrade/freqtrade.git synced 2025-12-02 01:53:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: add retrieval_mode in get_custom_data function to chose between value or full custom_data object

Browse Source
This commit is contained in:
Axel-CH
2025-03-11 23:36:24 -04:00
parent 743422ccf3
commit 4c7ff7ab0c
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
freqtrade/persistence/trade_model.py
View File

@@ -1351,13 +1351,20 @@ class LocalTrade:
"""
CustomDataWrapper.set_custom_data(trade_id=self.id, key=key, value=value)
def get_custom_data(self, key: str, default: Any = None) -> Any:
def get_custom_data(self, key: str, default: Any = None, retrieval_mode: str = "value") -> Any:
"""
Get custom data for this trade
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