53 lines
2.6 KiB
Python
53 lines
2.6 KiB
Python
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"
|
|
result = run_cli_command(command)
|
|
try:
|
|
users = json.loads(result)
|
|
except json.JSONDecodeError:
|
|
bot.answer_inline_query(query.id, results=[], switch_pm_text="Error retrieving users.", switch_pm_user_id=query.from_user.id)
|
|
return
|
|
|
|
query_text = query.query.lower().replace('\\_', '_')
|
|
results = []
|
|
|
|
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)
|