Add support for user ids in config

* Add support for user ids

* Allow both usernames and user ids at the same time
This commit is contained in:
Clément Sauvage
2023-03-14 13:22:06 +01:00
committed by GitHub
parent 35077ec20d
commit 76da87bb31
2 changed files with 6 additions and 5 deletions
+5 -4
View File
@@ -308,10 +308,11 @@ def run_bot() -> None:
)
# add handlers
if len(config.allowed_telegram_usernames) == 0:
user_filter = filters.ALL
else:
user_filter = filters.User(username=config.allowed_telegram_usernames)
user_filter = filters.ALL
if len(config.allowed_telegram_usernames) > 0:
usernames = [x for x in config.allowed_telegram_usernames if isinstance(x, str)]
user_ids = [x for x in config.allowed_telegram_usernames if isinstance(x, int)]
user_filter = filters.User(username=usernames) | filters.User(user_id=user_ids)
application.add_handler(CommandHandler("start", start_handle, filters=user_filter))
application.add_handler(CommandHandler("help", help_handle, filters=user_filter))
+1 -1
View File
@@ -1,7 +1,7 @@
telegram_token: ""
openai_api_key: ""
use_chatgpt_api: true
allowed_telegram_usernames: [] # if empty, the bot is available to anyone
allowed_telegram_usernames: [] # if empty, the bot is available to anyone. pass a username string to allow it and/or user ids as integers
new_dialog_timeout: 600 # new dialog starts after timeout (in seconds)
chatgpt_price_per_1000_tokens: 0.002