from datetime import datetime from os import path, sep from app import app, isAnAdmin 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 classes.holo_user import HoloUser, UserInvalidError, UserNotFoundError from modules.utils import configGet, locale from modules.database import col_applications @app.on_inline_query() async def inline_answer(client, inline_query): if inline_query.chat_type in [ChatType.CHANNEL]: await inline_query.answer( results=[ InlineQueryResultArticle( title=locale("title", "inline", "not_pm", locale=inline_query.from_user), input_message_content=InputTextMessageContent( locale("message_content", "inline", "not_pm", locale=inline_query.from_user) ), description=locale("description", "inline", "not_pm", locale=inline_query.from_user) ) ] ) return try: holo_user = HoloUser(inline_query.from_user) except (UserNotFoundError, UserInvalidError): await inline_query.answer( results=[ InlineQueryResultArticle( title=locale("title", "inline", "forbidden", locale=inline_query.from_user), input_message_content=InputTextMessageContent( locale("message_content", "inline", "forbidden", locale=inline_query.from_user) ), description=locale("description", "inline", "forbidden", locale=inline_query.from_user) ) ] ) return if holo_user.application_approved() or (await isAnAdmin(holo_user.id) is True): 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: application = col_applications.find_one({"user": match.user.id}) if application is None: continue application_content = [] i = 1 for question in application['application']: if i == 2: age = relativedelta(datetime.now(), application['application']['2']) application_content.append(f"{locale(f'question{i}', 'message', 'question_titles', locale=inline_query.from_user)} {application['application']['2'].strftime('%d.%m.%Y')} ({age.years} р.)") elif i == 3: if application['application']['3']['countryCode'] == "UA": application_content.append(f"{locale(f'question{i}', 'message', 'question_titles', locale=inline_query.from_user)} {application['application']['3']['name']}") else: application_content.append(f"{locale(f'question{i}', 'message', 'question_titles', locale=inline_query.from_user)} {application['application']['3']['name']} ({application['application']['3']['adminName1']}, {application['application']['3']['countryName']})") else: application_content.append(f"{locale(f'question{i}', 'message', 'question_titles', locale=inline_query.from_user)} {application['application'][question]}") i += 1 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", locale=inline_query.from_user).format(match.user.first_name, match.user.username, "\n".join(application_content)) ), description=locale("description", "inline", "user", locale=inline_query.from_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", locale=inline_query.from_user).format(match.user.first_name, match.user.username, "\n".join(application_content)) ), description=locale("description", "inline", "user", locale=inline_query.from_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", locale=inline_query.from_user).format(match.user.first_name, match.user.username, "\n".join(application_content)) ), description=locale("description", "inline", "user", locale=inline_query.from_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", locale=inline_query.from_user).format(match.user.first_name, match.user.username, "\n".join(application_content)) ), description=locale("description", "inline", "user", locale=inline_query.from_user).format(match.user.first_name, match.user.username) ) ) await inline_query.answer( results=results, cache_time=10 )