fb60f684f7
Stop and start calls added Along with refactoring to base line and use decorators. Also modules loaded optionally if enabled in config or not binds to ip / port set from config.json with warning if not localhost TODO: - use argparse in client, and generally clean client up - create unit test - documentation - extend to other RCP commands, after feedback
51 lines
1.7 KiB
Python
51 lines
1.7 KiB
Python
"""
|
|
This module contains class to manage RPC communications (Telegram, Slack, Rest ...)
|
|
"""
|
|
import logging
|
|
from typing import List
|
|
|
|
from freqtrade.rpc.rpc import RPC
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class RPCManager(object):
|
|
"""
|
|
Class to manage RPC objects (Telegram, Slack, ...)
|
|
"""
|
|
def __init__(self, freqtrade) -> None:
|
|
""" Initializes all enabled rpc modules """
|
|
self.registered_modules: List[RPC] = []
|
|
|
|
# Enable telegram
|
|
if freqtrade.config['telegram'].get('enabled', False):
|
|
logger.info('Enabling rpc.telegram ...')
|
|
from freqtrade.rpc.telegram import Telegram
|
|
self.registered_modules.append(Telegram(freqtrade))
|
|
|
|
# Enable local rest server for cmd line control
|
|
if freqtrade.config['rest_cmd_line'].get('enabled', False):
|
|
logger.info('Enabling rpc.local_rest_server ...')
|
|
from freqtrade.rpc.local_rest_server import LocalRestSuperWrap
|
|
self.registered_modules.append(LocalRestSuperWrap(freqtrade))
|
|
|
|
def cleanup(self) -> None:
|
|
""" Stops all enabled rpc modules """
|
|
logger.info('Cleaning up rpc modules ...')
|
|
while self.registered_modules:
|
|
mod = self.registered_modules.pop()
|
|
logger.debug('Cleaning up rpc.%s ...', mod.name)
|
|
mod.cleanup()
|
|
del mod
|
|
|
|
def send_msg(self, msg: str) -> None:
|
|
"""
|
|
Send given markdown message to all registered rpc modules
|
|
:param msg: message
|
|
:return: None
|
|
"""
|
|
logger.info('Sending rpc message: %s', msg)
|
|
for mod in self.registered_modules:
|
|
logger.debug('Forwarding message to rpc.%s', mod.name)
|
|
mod.send_msg(msg)
|