Add helper method to calculate protection until

This commit is contained in:
Matthias
2020-11-17 19:43:12 +01:00
parent 47cd856fea
commit 59091ef2b7
6 changed files with 27 additions and 8 deletions

View File

@@ -42,8 +42,8 @@ class CooldownPeriod(IProtection):
trade = Trade.get_trades(filters).first()
if trade:
self.log_on_refresh(logger.info, f"Cooldown for {pair} for {self._stop_duration}.")
until = trade.close_date.replace(
tzinfo=timezone.utc) + timedelta(minutes=self._stop_duration)
until = self.calculate_lock_end([trade], self._stop_duration)
return True, until, self._reason()
return False, None, None

View File

@@ -1,10 +1,11 @@
import logging
from abc import ABC, abstractmethod
from datetime import datetime
from typing import Any, Dict, Optional, Tuple
from datetime import datetime, timedelta, timezone
from typing import Any, Dict, List, Optional, Tuple
from freqtrade.mixins import LoggingMixin
from freqtrade.persistence import Trade
logger = logging.getLogger(__name__)
@@ -45,3 +46,17 @@ class IProtection(LoggingMixin, ABC):
:return: Tuple of [bool, until, reason].
If true, this pair will be locked with <reason> until <until>
"""
@staticmethod
def calculate_lock_end(trades: List[Trade], stop_minutes: int) -> datetime:
"""
Get lock end time
"""
max_date: datetime = max([trade.close_date for trade in trades])
# comming from Database, tzinfo is not set.
if max_date.tzinfo is None:
max_date = max_date.replace(tzinfo=timezone.utc)
until = max_date + timedelta(minutes=stop_minutes)
return until

View File

@@ -3,7 +3,6 @@ import logging
from datetime import datetime, timedelta
from typing import Any, Dict
from freqtrade.persistence import Trade
from freqtrade.plugins.protections import IProtection, ProtectionReturn
@@ -57,7 +56,8 @@ class LowProfitPairs(IProtection):
logger.info,
f"Trading for {pair} stopped due to {profit} < {self._required_profit} "
f"within {self._lookback_period} minutes.")
until = date_now + timedelta(minutes=self._stop_duration)
until = self.calculate_lock_end(trades, self._stop_duration)
return True, until, self._reason(profit)
return False, None, None

View File

@@ -55,7 +55,7 @@ class StoplossGuard(IProtection):
if len(trades) > self._trade_limit:
self.log_on_refresh(logger.info, f"Trading stopped due to {self._trade_limit} "
f"stoplosses within {self._lookback_period} minutes.")
until = date_now + timedelta(minutes=self._stop_duration)
until = self.calculate_lock_end(trades, self._stop_duration)
return True, until, self._reason()
return False, None, None