ec252ff774
Added only the "Daily" call so far, submitting for early review/feedback
Called as example "./rest_client.py daily 3"
This depends on listed as requirements.
Flask==1.0.2
flask-jsonpify==1.5.0 (will do later)
flask-restful==0.3.6
TODO: make loading optional, cleanly unload on close
unit tests, take feedback, tidy output,
add other Telegram functions, onwards
local rest server is enabled/disabled from within config.json. E.g
"localrest": {
"enabled": true
},
The server is enabled from within existing rpc manager
and makes use of the existing superclass (RPC)
Through making use of the existing hard work done in rpc.py
It *should be easy to add the other Telegram calls into local_rpc_server.py
The server is wrapped in a thread to be non-blocking
The server and client accept serialised calls or not, used in daily to return json
The client can be used from command line or in a python client script
As example, from cmdline for last 3 days Daily
DannyMBP:rpc creslin$ ./rest_client.py daily 3
[
[
"2018-06-13",
"0.00000000 USDT",
"0.000 USD",
"0 trade"
],
[
"2018-06-12",
"0.00000000 USDT",
"0.000 USD",
"0 trade"
],
[
"2018-06-11",
"0.00000000 USDT",
"0.000 USD",
"0 trade"
]
]
54 lines
1.8 KiB
Python
54 lines
1.8 KiB
Python
"""
|
|
This module contains class to manage RPC communications (Telegram, Slack, Rest ....)
|
|
"""
|
|
import logging
|
|
from typing import List, Dict
|
|
|
|
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.get('telegram', {}).get('enabled', False):
|
|
logger.info('Enabling rpc.telegram ...')
|
|
from freqtrade.rpc.telegram import Telegram
|
|
self.registered_modules.append(Telegram(freqtrade))
|
|
|
|
# Enable local rest api server for cmd line control
|
|
if freqtrade.config.get('api_server', {}).get('enabled', False):
|
|
logger.info('Enabling rpc.api_server')
|
|
from freqtrade.rpc.api_server import ApiServer
|
|
self.registered_modules.append(ApiServer(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: Dict[str, str]) -> None:
|
|
"""
|
|
Send given message to all registered rpc modules.
|
|
A message consists of one or more key value pairs of strings.
|
|
e.g.:
|
|
{
|
|
'status': 'stopping bot'
|
|
}
|
|
"""
|
|
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)
|