diff --git a/freqtrade/rpc/telegram.py b/freqtrade/rpc/telegram.py index fe4929780..a5e8265da 100644 --- a/freqtrade/rpc/telegram.py +++ b/freqtrade/rpc/telegram.py @@ -31,7 +31,7 @@ def authorized_only(command_handler: Callable[..., None]) -> Callable[..., Any]: """ def wrapper(self, *args, **kwargs): """ Decorator logic """ - update = kwargs.get('update') or args[1] + update = kwargs.get('update') or args[0] # Reject unauthorized messages chat_id = int(self._config['telegram']['chat_id']) @@ -79,7 +79,8 @@ class Telegram(RPC): registers all known command handlers and starts polling for message updates """ - self._updater = Updater(token=self._config['telegram']['token'], workers=0) + self._updater = Updater(token=self._config['telegram']['token'], workers=0, + use_context=True) # Register command handler and start telegram message polling handles = [ @@ -175,7 +176,7 @@ class Telegram(RPC): self._send_msg(message) @authorized_only - def _status(self, bot: Bot, update: Update) -> None: + def _status(self, update: Update, context: CallbackContext) -> None: """ Handler for /status. Returns the current TradeThread status @@ -221,13 +222,13 @@ class Telegram(RPC): messages.append("\n".join([l for l in lines if l]).format(**r)) for msg in messages: - self._send_msg(msg, bot=bot) + self._send_msg(msg) except RPCException as e: - self._send_msg(str(e), bot=bot) + self._send_msg(str(e)) @authorized_only - def _status_table(self, bot: Bot, update: Update) -> None: + def _status_table(self, update: Update, context: CallbackContext) -> None: """ Handler for /status table. Returns the current TradeThread status in table format @@ -240,10 +241,10 @@ class Telegram(RPC): message = tabulate(df_statuses, headers='keys', tablefmt='simple') self._send_msg(f"
{message}", parse_mode=ParseMode.HTML)
except RPCException as e:
- self._send_msg(str(e), bot=bot)
+ self._send_msg(str(e))
@authorized_only
- def _daily(self, bot: Bot, update: Update) -> None:
+ def _daily(self, update: Update, context: CallbackContext) -> None:
"""
Handler for /daily {stats_tab}'
- self._send_msg(message, bot=bot, parse_mode=ParseMode.HTML)
+ self._send_msg(message, parse_mode=ParseMode.HTML)
except RPCException as e:
- self._send_msg(str(e), bot=bot)
+ self._send_msg(str(e))
@authorized_only
- def _profit(self, bot: Bot, update: Update) -> None:
+ def _profit(self, update: Update, context: CallbackContext) -> None:
"""
Handler for /profit.
Returns a cumulative profit statistics.
@@ -317,12 +318,12 @@ class Telegram(RPC):
f"*Latest Trade opened:* `{latest_trade_date}`\n" \
f"*Avg. Duration:* `{avg_duration}`\n" \
f"*Best Performing:* `{best_pair}: {best_rate:.2f}%`"
- self._send_msg(markdown_msg, bot=bot)
+ self._send_msg(markdown_msg)
except RPCException as e:
- self._send_msg(str(e), bot=bot)
+ self._send_msg(str(e))
@authorized_only
- def _balance(self, bot: Bot, update: Update) -> None:
+ def _balance(self, update: Update, context: CallbackContext) -> None:
""" Handler for /balance """
try:
result = self._rpc_balance(self._config.get('fiat_display_currency', ''))
@@ -339,7 +340,7 @@ class Telegram(RPC):
# Handle overflowing messsage length
if len(output + curr_output) >= MAX_TELEGRAM_MESSAGE_LENGTH:
- self._send_msg(output, bot=bot)
+ self._send_msg(output)
output = curr_output
else:
output += curr_output
@@ -347,12 +348,12 @@ class Telegram(RPC):
output += "\n*Estimated Value*:\n" \
"\t`BTC: {total: .8f}`\n" \
"\t`{symbol}: {value: .2f}`\n".format(**result)
- self._send_msg(output, bot=bot)
+ self._send_msg(output)
except RPCException as e:
- self._send_msg(str(e), bot=bot)
+ self._send_msg(str(e))
@authorized_only
- def _start(self, bot: Bot, update: Update) -> None:
+ def _start(self, update: Update, context: CallbackContext) -> None:
"""
Handler for /start.
Starts TradeThread
@@ -361,10 +362,10 @@ class Telegram(RPC):
:return: None
"""
msg = self._rpc_start()
- self._send_msg('Status: `{status}`'.format(**msg), bot=bot)
+ self._send_msg('Status: `{status}`'.format(**msg))
@authorized_only
- def _stop(self, bot: Bot, update: Update) -> None:
+ def _stop(self, update: Update, context: CallbackContext) -> None:
"""
Handler for /stop.
Stops TradeThread
@@ -373,10 +374,10 @@ class Telegram(RPC):
:return: None
"""
msg = self._rpc_stop()
- self._send_msg('Status: `{status}`'.format(**msg), bot=bot)
+ self._send_msg('Status: `{status}`'.format(**msg))
@authorized_only
- def _reload_conf(self, bot: Bot, update: Update) -> None:
+ def _reload_conf(self, update: Update, context: CallbackContext) -> None:
"""
Handler for /reload_conf.
Triggers a config file reload
@@ -385,10 +386,10 @@ class Telegram(RPC):
:return: None
"""
msg = self._rpc_reload_conf()
- self._send_msg('Status: `{status}`'.format(**msg), bot=bot)
+ self._send_msg('Status: `{status}`'.format(**msg))
@authorized_only
- def _stopbuy(self, bot: Bot, update: Update) -> None:
+ def _stopbuy(self, update: Update, context: CallbackContext) -> None:
"""
Handler for /stop_buy.
Sets max_open_trades to 0 and gracefully sells all open trades
@@ -397,10 +398,10 @@ class Telegram(RPC):
:return: None
"""
msg = self._rpc_stopbuy()
- self._send_msg('Status: `{status}`'.format(**msg), bot=bot)
+ self._send_msg('Status: `{status}`'.format(**msg))
@authorized_only
- def _forcesell(self, bot: Bot, update: Update) -> None:
+ def _forcesell(self, update: Update, context: CallbackContext) -> None:
"""
Handler for /forcesell {edge_pairs_tab}'
- self._send_msg(message, bot=bot, parse_mode=ParseMode.HTML)
+ self._send_msg(message, parse_mode=ParseMode.HTML)
except RPCException as e:
- self._send_msg(str(e), bot=bot)
+ self._send_msg(str(e))
@authorized_only
- def _help(self, bot: Bot, update: Update) -> None:
+ def _help(self, update: Update, context: CallbackContext) -> None:
"""
Handler for /help.
Show commands of the bot
@@ -559,10 +560,10 @@ class Telegram(RPC):
"*/help:* `This help message`\n" \
"*/version:* `Show version`"
- self._send_msg(message, bot=bot)
+ self._send_msg(message)
@authorized_only
- def _version(self, bot: Bot, update: Update) -> None:
+ def _version(self, update: Update, context: CallbackContext) -> None:
"""
Handler for /version.
Show version information
@@ -570,10 +571,9 @@ class Telegram(RPC):
:param update: message update
:return: None
"""
- self._send_msg('*Version:* `{}`'.format(__version__), bot=bot)
+ self._send_msg('*Version:* `{}`'.format(__version__))
- def _send_msg(self, msg: str, bot: Bot = None,
- parse_mode: ParseMode = ParseMode.MARKDOWN) -> None:
+ def _send_msg(self, msg: str, parse_mode: ParseMode = ParseMode.MARKDOWN) -> None:
"""
Send given markdown message
:param msg: message
@@ -581,7 +581,6 @@ class Telegram(RPC):
:param parse_mode: telegram parse mode
:return: None
"""
- bot = bot or self._updater.bot
keyboard = [['/daily', '/profit', '/balance'],
['/status', '/status table', '/performance'],
@@ -591,7 +590,7 @@ class Telegram(RPC):
try:
try:
- bot.send_message(
+ self._updater.bot.send_message(
self._config['telegram']['chat_id'],
text=msg,
parse_mode=parse_mode,
@@ -604,7 +603,7 @@ class Telegram(RPC):
'Telegram NetworkError: %s! Trying one more time.',
network_err.message
)
- bot.send_message(
+ self._updater.bot.send_message(
self._config['telegram']['chat_id'],
text=msg,
parse_mode=parse_mode,