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/plugins/commands/start.py

120 lines
3.9 KiB
Python

from convopyro import listen_message
from pyrogram import filters
from pyrogram.types import (
KeyboardButton,
Message,
ReplyKeyboardMarkup,
ReplyKeyboardRemove,
)
from classes.pyroclient import PyroClient
from modules import custom_filters
@PyroClient.on_message(
~filters.scheduled & filters.private & filters.command(["start"], prefixes=["/"]) & ~custom_filters.context # type: ignore
)
async def command_start(app: PyroClient, message: Message):
user = await app.find_user(message.from_user)
await message.reply_text(
app._("start", "messages", locale=user.locale).format(
terms_of_service="https://garbagebot.eu/community/public/terms-of-service",
privacy_policy="https://garbagebot.eu/community/public/privacy-policy",
)
)
join_code = None if len(message.command) == 1 else message.command[1]
if join_code is not None:
try:
location = await app.get_location(int(join_code))
except ValueError:
await message.reply_text(
app._("start_code_invalid", "messages", locale=user.locale)
)
return
keyboard = ReplyKeyboardMarkup(
[
[
KeyboardButton(
app._("start_code_yes", "buttons", locale=user.locale)
)
],
[KeyboardButton(app._("start_code_no", "buttons", locale=user.locale))],
],
resize_keyboard=True,
one_time_keyboard=True,
)
await message.reply_text(
app._("start_code", "messages", locale=user.locale).format(
name=location.name
),
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)
if answer is None or answer.text == "/cancel":
await message.reply_text(
app._("cancelled", "messages", locale=user.locale),
reply_markup=ReplyKeyboardRemove(),
)
return
if answer.text not in app.in_all_locales(
"start_code_yes", "buttons"
) + app.in_all_locales("start_code_no", "buttons"):
await answer.reply_text(
app._("selection_invalid", "messages", locale=user.locale).format(
cancel_notice=app._("cancel", "messages", locale=user.locale)
)
)
continue
if answer.text in app.in_all_locales("start_code_no", "buttons"):
await answer.reply_text(
app._("start_selection_no", "messages", locale=user.locale),
reply_markup=ReplyKeyboardRemove(),
)
return
break
await user.update_location(location.id)
user_time = user.get_reminder_time().strftime(
app._("time", "formats", locale=user.locale)
)
await answer.reply_text(
app._("start_selection_yes", "messages", locale=user.locale).format(
name=location.name,
offset=user.offset,
time=user_time,
),
reply_markup=ReplyKeyboardRemove(),
)
return
if user.location is None:
await message.reply_text(
app._("start_configure", "messages", locale=user.locale),
reply_markup=ReplyKeyboardMarkup(
[
[
KeyboardButton(
app._("start_configure", "buttons", locale=user.locale)
)
]
],
resize_keyboard=True,
one_time_keyboard=True,
),
)