mohammadian7/freqtrade
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Authorized-only and /version to async

Browse Source
This commit is contained in:
Matthias
2023-04-15 09:46:51 +02:00
parent cb45689c1d
commit 5134bf8ec3
1 changed files with 13 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
freqtrade/rpc/telegram.py
+13 -13
View File
@@ -58,7 +58,7 @@ def authorized_only(command_handler: Callable[..., None]) -> Callable[..., Any]:
:return: decorated function :return: decorated function
""" """
def wrapper(self, *args, **kwargs): async def wrapper(self, *args, **kwargs):
""" Decorator logic """ """ Decorator logic """
update = kwargs.get('update') or args[0] update = kwargs.get('update') or args[0]
@@ -80,9 +80,9 @@ def authorized_only(command_handler: Callable[..., None]) -> Callable[..., Any]:
chat_id chat_id
) )
try: try:
return command_handler(self, *args, **kwargs) return await command_handler(self, *args, **kwargs)
except RPCException as e: except RPCException as e:
self._send_msg(str(e)) await self._send_msg(str(e))
except BaseException: except BaseException:
logger.exception('Exception occurred within Telegram module') logger.exception('Exception occurred within Telegram module')
finally: finally:
@@ -1589,7 +1589,7 @@ class Telegram(RPCHandler):
self._send_msg(message) self._send_msg(message)
@authorized_only @authorized_only
def _version(self, update: Update, context: CallbackContext) -> None: async def _version(self, update: Update, context: CallbackContext) -> None:
""" """
Handler for /version. Handler for /version.
Show version information Show version information
@@ -1602,7 +1602,7 @@ class Telegram(RPCHandler):
if strategy_version is not None: if strategy_version is not None:
version_string += f', *Strategy version: * `{strategy_version}`' version_string += f', *Strategy version: * `{strategy_version}`'
self._send_msg(version_string) await self._send_msg(version_string)
@authorized_only @authorized_only
def _show_config(self, update: Update, context: CallbackContext) -> None: def _show_config(self, update: Update, context: CallbackContext) -> None:
@@ -1680,12 +1680,12 @@ class Telegram(RPCHandler):
except TelegramError as telegram_err: except TelegramError as telegram_err:
logger.warning('TelegramError: %s! Giving up on that message.', telegram_err.message) logger.warning('TelegramError: %s! Giving up on that message.', telegram_err.message)
def _send_msg(self, msg: str, parse_mode: str = ParseMode.MARKDOWN, async def _send_msg(self, msg: str, parse_mode: str = ParseMode.MARKDOWN,
disable_notification: bool = False, disable_notification: bool = False,
keyboard: Optional[List[List[InlineKeyboardButton]]] = None, keyboard: Optional[List[List[InlineKeyboardButton]]] = None,
callback_path: str = "", callback_path: str = "",
reload_able: bool = False, reload_able: bool = False,
query: Optional[CallbackQuery] = None) -> None: query: Optional[CallbackQuery] = None) -> None:
""" """
Send given markdown message Send given markdown message
:param msg: message :param msg: message
@@ -1708,7 +1708,7 @@ class Telegram(RPCHandler):
reply_markup = ReplyKeyboardMarkup(self._keyboard, resize_keyboard=True) reply_markup = ReplyKeyboardMarkup(self._keyboard, resize_keyboard=True)
try: try:
try: try:
self._updater.bot.send_message( await self._app.bot.send_message(
self._config['telegram']['chat_id'], self._config['telegram']['chat_id'],
text=msg, text=msg,
parse_mode=parse_mode, parse_mode=parse_mode,
@@ -1722,7 +1722,7 @@ class Telegram(RPCHandler):
'Telegram NetworkError: %s! Trying one more time.', 'Telegram NetworkError: %s! Trying one more time.',
network_err.message network_err.message
) )
self._updater.bot.send_message( await self._app.bot.send_message(
self._config['telegram']['chat_id'], self._config['telegram']['chat_id'],
text=msg, text=msg,
parse_mode=parse_mode, parse_mode=parse_mode,