adelorenzo/stable
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add docstring to strategy wrapper

Browse Source
This commit is contained in:
Matthias 2020-02-22 11:41:22 +01:00
parent 4a188525ec
commit 365fdf4c37
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
freqtrade/strategy/strategy_wrapper.py
View File

@ -6,6 +6,11 @@ logger = logging.getLogger(__name__)
def strategy_safe_wrapper(f, message: str = "", default_retval=None): def strategy_safe_wrapper(f, message: str = "", default_retval=None):
"""
Wrapper around user-provided methods and functions.
Caches all exceptions and returns either the default_retval (if it's not None) or raises
a StrategyError exception, which then needs to be handled by the calling method.
"""
def wrapper(*args, **kwargs): def wrapper(*args, **kwargs):
try: try:
return f(*args, **kwargs) return f(*args, **kwargs)
@ -15,14 +20,14 @@ def strategy_safe_wrapper(f, message: str = "", default_retval=None):
f"Strategy caused the following exception: {error}" f"Strategy caused the following exception: {error}"
f"{f}" f"{f}"
) )
if not default_retval: if default_retval is None:
raise StrategyError(str(error)) from error raise StrategyError(str(error)) from error
return default_retval return default_retval
except Exception as error: except Exception as error:
logger.exception( logger.exception(
f"Unexpected error {error} calling {f}" f"Unexpected error {error} calling {f}"
) )
if not default_retval: if default_retval is None:
raise StrategyError(str(error)) from error raise StrategyError(str(error)) from error
return default_retval return default_retval