Telegram/modules/inline.py

from datetime import datetime
from os import path, sep
from app import app, isAnAdmin
from pyrogram.types import InlineQueryResultArticle, InputTextMessageContent, InlineQuery
from pyrogram.client import Client
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: Client, inline_query: InlineQuery):

    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):

        max_results = configGet("inline_preview_count") if inline_query.query != "" else 200

        list_of_users = []
        async for m in app.get_chat_members(configGet("destination_group"), limit=max_results, 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
        )