73 lines
2.6 KiB
Python
73 lines
2.6 KiB
Python
"""
|
|
This module manages webhook communication
|
|
"""
|
|
import logging
|
|
from typing import Any, Dict
|
|
|
|
from requests import post, RequestException
|
|
|
|
from freqtrade.rpc import RPC, RPCMessageType
|
|
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
logger.debug('Included module rpc.webhook ...')
|
|
|
|
|
|
class Webhook(RPC):
|
|
""" This class handles all webhook communication """
|
|
|
|
def __init__(self, freqtrade) -> None:
|
|
"""
|
|
Init the Webhook class, and init the super class RPC
|
|
:param freqtrade: Instance of a freqtrade bot
|
|
:return: None
|
|
"""
|
|
super().__init__(freqtrade)
|
|
|
|
self._config = freqtrade.config
|
|
self._url = self._config['webhook']['url']
|
|
|
|
def cleanup(self) -> None:
|
|
"""
|
|
Cleanup pending module resources.
|
|
This will do nothing for webhooks, they will simply not be called anymore
|
|
"""
|
|
pass
|
|
|
|
def send_msg(self, msg: Dict[str, Any]) -> None:
|
|
""" Send a message to telegram channel """
|
|
try:
|
|
|
|
if msg['type'] == RPCMessageType.BUY_NOTIFICATION:
|
|
valuedict = self._config['webhook'].get('webhookbuy', None)
|
|
elif msg['type'] == RPCMessageType.BUY_CANCEL_NOTIFICATION:
|
|
valuedict = self._config['webhook'].get('webhookbuycancel', None)
|
|
elif msg['type'] == RPCMessageType.SELL_NOTIFICATION:
|
|
valuedict = self._config['webhook'].get('webhooksell', None)
|
|
elif msg['type'] == RPCMessageType.SELL_CANCEL_NOTIFICATION:
|
|
valuedict = self._config['webhook'].get('webhooksellcancel', None)
|
|
elif msg['type'] in (RPCMessageType.STATUS_NOTIFICATION,
|
|
RPCMessageType.CUSTOM_NOTIFICATION,
|
|
RPCMessageType.WARNING_NOTIFICATION):
|
|
valuedict = self._config['webhook'].get('webhookstatus', None)
|
|
else:
|
|
raise NotImplementedError('Unknown message type: {}'.format(msg['type']))
|
|
if not valuedict:
|
|
logger.info("Message type %s not configured for webhooks", msg['type'])
|
|
return
|
|
|
|
payload = {key: value.format(**msg) for (key, value) in valuedict.items()}
|
|
self._send_msg(payload)
|
|
except KeyError as exc:
|
|
logger.exception("Problem calling Webhook. Please check your webhook configuration. "
|
|
"Exception: %s", exc)
|
|
|
|
def _send_msg(self, payload: dict) -> None:
|
|
"""do the actual call to the webhook"""
|
|
|
|
try:
|
|
post(self._url, data=payload)
|
|
except RequestException as exc:
|
|
logger.warning("Could not call webhook url. Exception: %s", exc)
|