diff --git a/core/scripts/telegrambot/utils/search.py b/core/scripts/telegrambot/utils/search.py index 494bdf2..31821e0 100644 --- a/core/scripts/telegrambot/utils/search.py +++ b/core/scripts/telegrambot/utils/search.py @@ -1,7 +1,6 @@ from telebot import types from utils.command import * - @bot.inline_handler(lambda query: is_admin(query.from_user.id)) def handle_inline_query(query): command = f"python3 {CLI_PATH} list-users" @@ -14,21 +13,40 @@ def handle_inline_query(query): query_text = query.query.lower() results = [] - for username, details in users.items(): - if query_text in username.lower(): - title = f"{username}" - description = f"Traffic Limit: {details['max_download_bytes'] / (1024 ** 3):.2f} GB, Expiration Days: {details['expiration_days']}" - results.append(types.InlineQueryResultArticle( - id=username, - title=title, - description=description, - input_message_content=types.InputTextMessageContent( - message_text=f"Name: {username}\n" - f"Traffic limit: {details['max_download_bytes'] / (1024 ** 3):.2f} GB\n" - f"Days: {details['expiration_days']}\n" - f"Account Creation: {details['account_creation_date']}\n" - f"Blocked: {details['blocked']}" - ) - )) - bot.answer_inline_query(query.id, results, cache_time=0) \ No newline at end of file + if query_text == "block": + for username, details in users.items(): + if details.get('blocked', False): + title = f"{username} (Blocked)" + description = f"Traffic Limit: {details['max_download_bytes'] / (1024 ** 3):.2f} GB, Expiration Days: {details['expiration_days']}" + results.append(types.InlineQueryResultArticle( + id=username, + title=title, + description=description, + input_message_content=types.InputTextMessageContent( + message_text=f"Name: {username}\n" + f"Traffic limit: {details['max_download_bytes'] / (1024 ** 3):.2f} GB\n" + f"Days: {details['expiration_days']}\n" + f"Account Creation: {details['account_creation_date']}\n" + f"Blocked: {details['blocked']}" + ) + )) + else: + for username, details in users.items(): + if query_text in username.lower(): + title = f"{username}" + description = f"Traffic Limit: {details['max_download_bytes'] / (1024 ** 3):.2f} GB, Expiration Days: {details['expiration_days']}" + results.append(types.InlineQueryResultArticle( + id=username, + title=title, + description=description, + input_message_content=types.InputTextMessageContent( + message_text=f"Name: {username}\n" + f"Traffic limit: {details['max_download_bytes'] / (1024 ** 3):.2f} GB\n" + f"Days: {details['expiration_days']}\n" + f"Account Creation: {details['account_creation_date']}\n" + f"Blocked: {details['blocked']}" + ) + )) + + bot.answer_inline_query(query.id, results, cache_time=0)