freqtrade_origin/freqtrade/plugins/protectionmanager.py

60 lines
1.9 KiB
Python
Raw Normal View History

"""
Protection manager class
"""
import logging
2020-10-15 06:07:09 +00:00
from datetime import datetime, timezone
from typing import Dict, List
2020-10-15 06:07:09 +00:00
from freqtrade.persistence import PairLocks
from freqtrade.plugins.protections import IProtection
from freqtrade.resolvers import ProtectionResolver
2020-10-15 05:38:00 +00:00
logger = logging.getLogger(__name__)
class ProtectionManager():
2020-10-14 05:40:44 +00:00
def __init__(self, config: dict) -> None:
self._config = config
self._protection_handlers: List[IProtection] = []
2020-10-14 18:03:56 +00:00
for protection_handler_config in self._config.get('protections', []):
if 'method' not in protection_handler_config:
logger.warning(f"No method found in {protection_handler_config}, ignoring.")
continue
protection_handler = ProtectionResolver.load_protection(
protection_handler_config['method'],
config=config,
protection_config=protection_handler_config,
)
self._protection_handlers.append(protection_handler)
if not self._protection_handlers:
2020-10-14 18:03:56 +00:00
logger.info("No protection Handlers defined.")
@property
def name_list(self) -> List[str]:
"""
Get list of loaded Protection Handler names
"""
return [p.name for p in self._protection_handlers]
def short_desc(self) -> List[Dict]:
"""
List of short_desc for each Pairlist Handler
"""
2020-10-14 18:03:56 +00:00
return [{p.name: p.short_desc()} for p in self._protection_handlers]
def global_stop(self) -> bool:
2020-10-15 06:07:09 +00:00
now = datetime.now(timezone.utc)
2020-10-15 05:38:00 +00:00
2020-10-14 18:03:56 +00:00
for protection_handler in self._protection_handlers:
2020-10-15 06:07:09 +00:00
result, until, reason = protection_handler.global_stop(now)
2020-10-14 18:03:56 +00:00
2020-10-15 06:07:09 +00:00
# Early stopping - first positive result blocks further trades
if result and until:
PairLocks.lock_pair('*', until, reason)
2020-10-14 18:03:56 +00:00
return True
return False