CTYPTO/freqtrade
1
0
Fork 0
mirror of https://github.com/freqtrade/freqtrade.git synced 2025-12-19 06:11:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: add authorized_users support for telegram

Browse Source
This commit is contained in:
Matthias
2025-03-16 13:44:23 +01:00
parent d543feb032
commit a95c3ace92
1 changed files with 12 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
freqtrade/rpc/telegram.py
View File

@@ -25,6 +25,7 @@ from telegram import (
InlineKeyboardButton, InlineKeyboardButton,
InlineKeyboardMarkup, InlineKeyboardMarkup,
KeyboardButton, KeyboardButton,
Message,
ReplyKeyboardMarkup, ReplyKeyboardMarkup,
Update, Update,
) )
@@ -98,12 +99,15 @@ def authorized_only(command_handler: Callable[..., Coroutine[Any, Any, None]]):
@wraps(command_handler) @wraps(command_handler)
async def wrapper(self, *args, **kwargs): async def wrapper(self, *args, **kwargs):
"""Decorator logic""" """Decorator logic"""
update = kwargs.get("update") or args[0] update: Update = kwargs.get("update") or args[0]
# Reject unauthorized messages # Reject unauthorized messages
message = update.message if update.callback_query is None else update.callback_query.message message: Message = (
cchat_id = int(message.chat_id) update.message if update.callback_query is None else update.callback_query.message
ctopic_id = message.message_thread_id )
cchat_id: int = int(message.chat_id)
ctopic_id: int | None = message.message_thread_id
from_user_id: str = str(update.effective_user.id)
chat_id = int(self._config["telegram"]["chat_id"]) chat_id = int(self._config["telegram"]["chat_id"])
if cchat_id != chat_id: if cchat_id != chat_id:
@@ -115,6 +119,10 @@ def authorized_only(command_handler: Callable[..., Coroutine[Any, Any, None]]):
logger.debug(f"Rejected message from wrong channel: {cchat_id}, {ctopic_id}") logger.debug(f"Rejected message from wrong channel: {cchat_id}, {ctopic_id}")
return None return None
authorized = self._config["telegram"].get("authorized_users", [])
if authorized and from_user_id not in authorized:
logger.info(f"Unauthorized user tried to control the bot: {from_user_id}")
return None
# Rollback session to avoid getting data stored in a transaction. # Rollback session to avoid getting data stored in a transaction.
Trade.rollback() Trade.rollback()
logger.debug("Executing handler: %s for chat_id: %s", command_handler.__name__, chat_id) logger.debug("Executing handler: %s for chat_id: %s", command_handler.__name__, chat_id)