117 lines
6.0 KiB
Python
117 lines
6.0 KiB
Python
from datetime import datetime
|
||
from os import path, sep
|
||
from pyrogram.types import InlineQueryResultArticle, InputTextMessageContent
|
||
from pyrogram.enums.chat_type import ChatType
|
||
from pyrogram.enums.chat_members_filter import ChatMembersFilter
|
||
from dateutil.relativedelta import relativedelta
|
||
|
||
from app import app, isAnAdmin
|
||
from modules.utils import configGet, jsonLoad, locale
|
||
|
||
@app.on_inline_query()
|
||
async def inline_answer(client, inline_query):
|
||
|
||
if inline_query.chat_type not in [ChatType.BOT, ChatType.PRIVATE]:
|
||
await inline_query.answer(
|
||
results=[
|
||
InlineQueryResultArticle(
|
||
title=locale("title", "inline", "not_pm"),
|
||
input_message_content=InputTextMessageContent(
|
||
locale("message_content", "inline", "not_pm")
|
||
),
|
||
description=locale("description", "inline", "not_pm")
|
||
)
|
||
]
|
||
)
|
||
return
|
||
|
||
if (path.exists(f"{configGet('data', 'locations')}{sep}users{sep}{inline_query.from_user.id}.json") and jsonLoad(f"{configGet('data', 'locations')}{sep}users{sep}{inline_query.from_user.id}.json")["approved"]) or (await isAnAdmin(inline_query.from_user.id)):
|
||
|
||
list_of_users = []
|
||
async for m in app.get_chat_members(configGet("destination_group"), limit=configGet("inline_preview_count"), filter=ChatMembersFilter.SEARCH, query=inline_query.query):
|
||
list_of_users.append(m)
|
||
|
||
results = []
|
||
|
||
for match in list_of_users:
|
||
|
||
try:
|
||
user_data = jsonLoad(f"{configGet('data', 'locations')}{sep}users{sep}{match.user.id}.json")
|
||
application_content = []
|
||
i = 1
|
||
for question in configGet("application", file=str(match.user.id)):
|
||
if i == 2:
|
||
age = relativedelta(datetime.now(), datetime.strptime(configGet('application', file=str(match.user.id))['2'], '%d.%m.%Y'))
|
||
application_content.append(f"{locale('question'+str(i), 'message', 'question_titles')} {configGet('application', file=str(match.user.id))['2']} ({age.years} р.)")
|
||
else:
|
||
application_content.append(f"{locale('question'+str(i), 'message', 'question_titles')} {configGet('application', file=str(match.user.id))[question]}")
|
||
i += 1
|
||
except FileNotFoundError:
|
||
continue
|
||
except TypeError:
|
||
continue
|
||
|
||
if match.user.photo != None:
|
||
try:
|
||
if not path.exists(f'{configGet("cache", "locations")}{sep}avatars{sep}{match.user.photo.big_file_id}'):
|
||
print(f'Downloaded avatar {match.user.photo.big_file_id} of {match.user.id} and uploaded to {configGet("api")}/?avatar_id={match.user.photo.big_file_id}', flush=True)
|
||
await app.download_media(match.user.photo.big_file_id, file_name=f'{configGet("cache", "locations")}{sep}avatars{sep}{match.user.photo.big_file_id}')
|
||
results.append(
|
||
InlineQueryResultArticle(
|
||
title=str(match.user.first_name),
|
||
input_message_content=InputTextMessageContent(
|
||
locale("message_content", "inline", "user").format(match.user.first_name, match.user.username, "\n".join(application_content))
|
||
),
|
||
description=locale("description", "inline", "user").format(match.user.first_name, match.user.username),
|
||
thumb_url=f'{configGet("api")}/?avatar_id={match.user.photo.big_file_id}'
|
||
)
|
||
)
|
||
except ValueError:
|
||
results.append(
|
||
InlineQueryResultArticle(
|
||
title=str(match.user.first_name),
|
||
input_message_content=InputTextMessageContent(
|
||
locale("message_content", "inline", "user").format(match.user.first_name, match.user.username, "\n".join(application_content))
|
||
),
|
||
description=locale("description", "inline", "user").format(match.user.first_name, match.user.username)
|
||
)
|
||
)
|
||
except FileNotFoundError:
|
||
results.append(
|
||
InlineQueryResultArticle(
|
||
title=str(match.user.first_name),
|
||
input_message_content=InputTextMessageContent(
|
||
locale("message_content", "inline", "user").format(match.user.first_name, match.user.username, "\n".join(application_content))
|
||
),
|
||
description=locale("description", "inline", "user").format(match.user.first_name, match.user.username)
|
||
)
|
||
)
|
||
else:
|
||
results.append(
|
||
InlineQueryResultArticle(
|
||
title=str(match.user.first_name),
|
||
input_message_content=InputTextMessageContent(
|
||
locale("message_content", "inline", "user").format(match.user.first_name, match.user.username, "\n".join(application_content))
|
||
),
|
||
description=locale("description", "inline", "user").format(match.user.first_name, match.user.username)
|
||
)
|
||
)
|
||
|
||
await inline_query.answer(
|
||
results=results,
|
||
cache_time=10
|
||
)
|
||
|
||
else:
|
||
await inline_query.answer(
|
||
results=[
|
||
InlineQueryResultArticle(
|
||
title=locale("title", "inline", "forbidden"),
|
||
input_message_content=InputTextMessageContent(
|
||
locale("message_content", "inline", "forbidden")
|
||
),
|
||
description=locale("description", "inline", "forbidden")
|
||
)
|
||
]
|
||
)
|
||
return |