This repository has been archived on 2024-10-10. You can view files and clone it, but cannot push or open issues or pull requests.
TelegramBot/modules/search_nearby.py
2023-11-05 13:20:01 +00:00

75 lines
2.3 KiB
Python

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_api 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)
locations.reverse()
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:
app.contexts.append(message.from_user.id)
answer = await listen_message(app, message.chat.id, 300)
app.contexts.remove(message.from_user.id)
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)