from typing import Union from convopyro import listen_message from pykeyboard import ReplyButton, ReplyKeyboard from pyrogram.types import Message, ReplyKeyboardRemove from classes.location import Location from classes.pyroclient import PyroClient from modules.database import col_locations from modules.search_name import search_name async def search_nearby(app: PyroClient, message: Message) -> Union[Location, None]: user = await app.find_user(message.from_user) query = { "location": { "$within": { "$center": [ [message.location.longitude, message.location.latitude], app.config["search"]["radius"], ] } } } locations = await col_locations.find(query).limit(6).to_list() if len(locations) == 0: await message.reply_text( app._("search_nearby_empty", "messages", locale=user.locale) ) return await search_name(app, message) keyboard = ReplyKeyboard(resize_keyboard=True, row_width=2) keyboard.add(*[ReplyButton(db_record["name"]) for db_record in locations]) await message.reply_text( app._("location_select", "messages", locale=user.locale), reply_markup=keyboard, ) while True: answer = await listen_message(app, message.chat.id, 300) location: Union[Location, None] = None if answer is None or answer.text == "/cancel": await message.reply_text( app._("cancelled", "messages", locale=user.locale), reply_markup=ReplyKeyboardRemove(), ) return for db_record in locations: if answer.text == db_record["name"]: location = Location(**db_record) break if answer.text is None or location is None: await answer.reply_text( app._("selection_invalid", "messages", locale=user.locale).format( cancel_notice=app._("cancel", "messages", locale=user.locale) ) ) continue break return location or await search_name(app, message)