Add first version of protection manager

This commit is contained in:
Matthias
2020-10-13 08:06:29 +02:00
parent b6b9c8e5cc
commit a0bd2ce837
6 changed files with 122 additions and 0 deletions

View File

View File

@@ -0,0 +1,51 @@
"""
Protection manager class
"""
import logging
from typing import Dict, List
from freqtrade.exceptions import OperationalException
from freqtrade.plugins.protections import IProtection
from freqtrade.resolvers import ProtectionResolver
logger = logging.getLogger(__name__)
class ProtectionManager():
def __init__(self, exchange, config: dict) -> None:
self._exchange = exchange
self._config = config
self._protection_handlers: List[IProtection] = []
self._tickers_needed = False
for protection_handler_config in self._config.get('protections', None):
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'],
exchange=exchange,
protectionmanager=self,
config=config,
protection_config=protection_handler_config,
)
self._tickers_needed |= protection_handler.needstickers
self._protection_handlers.append(protection_handler)
if not self._protection_handlers:
raise OperationalException("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
"""
return [{p.name: p.short_desc()} for p in self._pairlist_handlers]

View File

@@ -0,0 +1,2 @@
# flake8: noqa: F401
from freqtrade.plugins.protections.iprotection import IProtection

View File

@@ -0,0 +1,24 @@
import logging
from abc import ABC, abstractmethod
from typing import Any, Dict
logger = logging.getLogger(__name__)
class IProtection(ABC):
def __init__(self, config: Dict[str, Any]) -> None:
self._config = config
@property
def name(self) -> str:
return self.__class__.__name__
@abstractmethod
def short_desc(self) -> str:
"""
Short method description - used for startup-messages
-> Please overwrite in subclasses
"""