freqtrade_origin/freqtrade/strategy/strategy_wrapper.py

47 lines
1.5 KiB
Python
Raw Normal View History

import logging
from copy import deepcopy
2022-04-16 04:53:38 +00:00
from functools import wraps
from typing import Any, Callable, TypeVar, cast
from freqtrade.exceptions import StrategyError
2020-09-28 17:39:41 +00:00
logger = logging.getLogger(__name__)
2022-04-16 04:53:38 +00:00
F = TypeVar('F', bound=Callable[..., Any])
def strategy_safe_wrapper(f: F, message: str = "", default_retval=None, supress_error=False) -> F:
2020-02-22 10:41:22 +00:00
"""
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.
"""
2022-04-16 04:53:38 +00:00
@wraps(f)
def wrapper(*args, **kwargs):
try:
if 'trade' in kwargs:
# Protect accidental modifications from within the strategy
kwargs['trade'] = deepcopy(kwargs['trade'])
return f(*args, **kwargs)
except ValueError as error:
logger.warning(
f"{message}"
f"Strategy caused the following exception: {error}"
f"{f}"
)
2020-06-14 05:15:24 +00:00
if default_retval is None and not supress_error:
raise StrategyError(str(error)) from error
return default_retval
except Exception as error:
logger.exception(
2020-02-22 10:52:39 +00:00
f"{message}"
f"Unexpected error {error} calling {f}"
)
2020-06-14 05:15:24 +00:00
if default_retval is None and not supress_error:
raise StrategyError(str(error)) from error
return default_retval
2022-04-16 04:53:38 +00:00
return cast(F, wrapper)