78 lines
2.6 KiB
Python
78 lines
2.6 KiB
Python
from app import app
|
|
from pyrogram import filters
|
|
from pyrogram.types import ForceReply, ReplyKeyboardMarkup, Message
|
|
from pyrogram.client import Client
|
|
from classes.holo_user import HoloUser
|
|
from modules.utils import all_locales, locale, logWrite
|
|
from modules import custom_filters
|
|
|
|
# Welcome check ================================================================================================================
|
|
welcome_1 = []
|
|
for pattern in all_locales("welcome", "keyboard"):
|
|
welcome_1.append(pattern[0][0])
|
|
for pattern in all_locales("return", "keyboard"):
|
|
welcome_1.append(pattern[0][0])
|
|
|
|
|
|
@app.on_message(
|
|
custom_filters.enabled_applications
|
|
& ~filters.scheduled
|
|
& filters.private
|
|
& filters.command(welcome_1, prefixes=[""])
|
|
& ~custom_filters.banned
|
|
)
|
|
async def welcome_pass(app: Client, msg: Message, once_again: bool = False) -> None:
|
|
"""Set user's stage to 1 and start a fresh application
|
|
|
|
### Args:
|
|
* app (app): Pyrogram Client to use
|
|
* msg (Message): Message with .from_user.id attribute equal to the end-user ID whose application will be started
|
|
* once_again (bool, optional): Set to False if it's the first time as user applies. Defaults to True.
|
|
"""
|
|
|
|
if not once_again:
|
|
await msg.reply_text(locale("privacy_notice", "message"))
|
|
|
|
holo_user = HoloUser(msg.from_user)
|
|
|
|
if once_again is False:
|
|
holo_user.application_restart()
|
|
|
|
if once_again is True:
|
|
logWrite(f"User {msg.from_user.id} confirmed starting the application")
|
|
else:
|
|
logWrite(
|
|
f"User {msg.from_user.id} confirmed starting the application once again"
|
|
)
|
|
await msg.reply_text(
|
|
locale("question1", "message", locale=msg.from_user),
|
|
reply_markup=ForceReply(
|
|
placeholder=locale("question1", "force_reply", locale=msg.from_user)
|
|
),
|
|
)
|
|
|
|
|
|
welcome_2 = []
|
|
for pattern in all_locales("welcome", "keyboard"):
|
|
welcome_2.append(pattern[1][0])
|
|
|
|
|
|
@app.on_message(
|
|
custom_filters.enabled_applications
|
|
& ~filters.scheduled
|
|
& filters.private
|
|
& filters.command(welcome_2, prefixes=[""])
|
|
& ~custom_filters.banned
|
|
)
|
|
async def welcome_reject(app: Client, msg: Message):
|
|
logWrite(f"User {msg.from_user.id} rejected to start the application")
|
|
await msg.reply_text(
|
|
locale("goodbye", "message", locale=msg.from_user),
|
|
reply_markup=ReplyKeyboardMarkup(
|
|
locale("return", "keyboard", locale=msg.from_user), resize_keyboard=True
|
|
),
|
|
)
|
|
|
|
|
|
# ==============================================================================================================================
|