WIP: /language system

This commit is contained in:
Profitroll 2023-06-28 10:37:18 +02:00
parent 6f8b560acc
commit 10c60ae932
Signed by: profitroll
GPG Key ID: FA35CAB49DACD3B2
9 changed files with 203 additions and 205 deletions

View File

@ -1,58 +0,0 @@
from datetime import datetime
from libbot import sync
from modules.database import col_banned, col_users
class PosterUser:
def __init__(self, id: int):
self.id = id
def is_blocked(self) -> bool:
"""Check if user is banned from submitting content.
### Returns:
`bool`: Must be `True` if banned and `False` if not
"""
return False if col_banned.find_one({"user": self.id}) is None else True
def block(self) -> None:
"""Ban user from using command and submitting content."""
if col_banned.find_one({"user": self.id}) is None:
col_banned.insert_one({"user": self.id, "date": datetime.now()})
def unblock(self) -> None:
"""Allow user to use command and submit posts again."""
col_banned.find_one_and_delete({"user": self.id})
def is_limited(self) -> bool:
"""Check if user is on a cooldown after submitting something.
### Returns:
`bool`: Must be `True` if on the cooldown and `False` if not
"""
if self.id in sync.config_get("admins", "bot"):
return False
db_record = col_users.find_one({"user": self.id})
if db_record is None:
return False
return (
True
if (datetime.now() - db_record["cooldown"]).total_seconds()
< sync.config_get("timeout", "submission")
else False
)
def limit(self) -> None:
"""Restart user's cooldown. Used after post has been submitted."""
if (
col_users.find_one_and_update(
{"user": self.id}, {"$set": {"cooldown": datetime.now()}}
)
is None
):
col_users.insert_one({"user": self.id, "cooldown": datetime.now()})

View File

@ -7,6 +7,6 @@ from classes.pyroclient import PyroClient
@Client.on_callback_query(filters.regex("nothing"))
async def callback_query_nothing(app: PyroClient, clb: CallbackQuery):
await clb.answer(
text=app._("nothing", "callback", locale=clb.from_user.language_code)
)
user = await app.find_user(clb.from_user)
await clb.answer(text=app._("nothing", "callback", locale=user.locale))

View File

@ -15,7 +15,6 @@ from classes.exceptions import (
SubmissionUnsupportedError,
)
from classes.pyroclient import PyroClient
from classes.user import PosterUser
from modules.database import col_submitted
logger = logging.getLogger(__name__)
@ -23,22 +22,22 @@ logger = logging.getLogger(__name__)
@Client.on_callback_query(filters.regex("sub_yes_[\s\S]*"))
async def callback_query_yes(app: PyroClient, clb: CallbackQuery):
user = await app.find_user(clb.from_user)
fullclb = str(clb.data).split("_")
user_locale = clb.from_user.language_code
db_entry = col_submitted.find_one({"_id": ObjectId(fullclb[2])})
try:
submission = await app.submit_photo(fullclb[2])
submission = await app.submit_media(fullclb[2])
except SubmissionUnavailableError:
await clb.answer(
text=app._("sub_msg_unavail", "callback", locale=user_locale),
text=app._("sub_msg_unavail", "callback", locale=user.locale),
show_alert=True,
)
return
except SubmissionUnsupportedError:
await clb.answer(
text=app._("mime_not_allowed", "message", locale=user_locale).format(
text=app._("mime_not_allowed", "message", locale=user.locale).format(
", ".join(app.config["submission"]["mime_types"]), quote=True
),
show_alert=True,
@ -46,11 +45,11 @@ async def callback_query_yes(app: PyroClient, clb: CallbackQuery):
return
except SubmissionDuplicatesError as exp:
await clb.answer(
text=app._("sub_duplicates_found", "callback", locale=user_locale),
text=app._("sub_duplicates_found", "callback", locale=user.locale),
show_alert=True,
)
await clb.message.reply_text(
app._("sub_media_duplicates_list", "message", locale=user_locale).format(
app._("sub_media_duplicates_list", "message", locale=user.locale).format(
"\n".join(exp.duplicates)
),
quote=True,
@ -69,14 +68,25 @@ async def callback_query_yes(app: PyroClient, clb: CallbackQuery):
if submission[0] is not None:
await submission[0].reply_text(
app._("sub_yes", "message", locale=submission[0].from_user.language_code),
app._(
"sub_yes",
"message",
locale=(await app.find_user(submission[0].from_user)).locale,
),
quote=True,
)
elif db_entry is not None:
await app.send_message(db_entry["user"], app._("sub_yes", "message"))
await app.send_message(
db_entry["user"],
app._(
"sub_yes",
"message",
locale=(await app.find_user(db_entry["user"])).locale,
),
)
await clb.answer(
text=app._("sub_yes", "callback", locale=user_locale).format(fullclb[2]),
text=app._("sub_yes", "callback", locale=user.locale).format(fullclb[2]),
show_alert=True,
)
@ -84,7 +94,7 @@ async def callback_query_yes(app: PyroClient, clb: CallbackQuery):
[
[
InlineKeyboardButton(
text=str(app._("accepted", "button", locale=user_locale)),
text=str(app._("accepted", "button", locale=user.locale)),
callback_data="nothing",
)
],
@ -94,7 +104,7 @@ async def callback_query_yes(app: PyroClient, clb: CallbackQuery):
else [
[
InlineKeyboardButton(
text=str(app._("accepted", "button", locale=user_locale)),
text=str(app._("accepted", "button", locale=user.locale)),
callback_data="nothing",
)
]
@ -103,7 +113,7 @@ async def callback_query_yes(app: PyroClient, clb: CallbackQuery):
if await config_get("send_uploaded_id", "submission"):
await clb.message.edit_caption(
clb.message.caption + f"\n\nID: `{submission[1]}`"
f"{clb.message.caption}\n\nID: `{submission[1]}`"
)
await clb.message.edit_reply_markup(
@ -121,8 +131,8 @@ async def callback_query_yes(app: PyroClient, clb: CallbackQuery):
@Client.on_callback_query(filters.regex("sub_no_[\s\S]*"))
async def callback_query_no(app: PyroClient, clb: CallbackQuery):
user = await app.find_user(clb.from_user)
fullclb = str(clb.data).split("_")
user_locale = clb.from_user.language_code
db_entry = col_submitted.find_one_and_delete({"_id": ObjectId(fullclb[2])})
@ -145,17 +155,21 @@ async def callback_query_no(app: PyroClient, clb: CallbackQuery):
)
except Exception as exp:
await clb.answer(
text=app._("sub_msg_unavail", "message", locale=user_locale),
text=app._("sub_msg_unavail", "message", locale=user.locale),
show_alert=True,
)
return
await submission.reply_text(
app._("sub_no", "message", locale=submission.from_user.language_code),
app._(
"sub_no",
"message",
locale=(await app.find_user(submission.from_user)).locale,
),
quote=True,
)
await clb.answer(
text=app._("sub_no", "callback", locale=user_locale).format(fullclb[2]),
text=app._("sub_no", "callback", locale=user.locale).format(fullclb[2]),
show_alert=True,
)
@ -163,7 +177,7 @@ async def callback_query_no(app: PyroClient, clb: CallbackQuery):
[
[
InlineKeyboardButton(
text=str(app._("declined", "button", locale=user_locale)),
text=str(app._("declined", "button", locale=user.locale)),
callback_data="nothing",
)
],
@ -173,7 +187,7 @@ async def callback_query_no(app: PyroClient, clb: CallbackQuery):
else [
[
InlineKeyboardButton(
text=str(app._("declined", "button", locale=user_locale)),
text=str(app._("declined", "button", locale=user.locale)),
callback_data="nothing",
)
]
@ -193,17 +207,21 @@ async def callback_query_no(app: PyroClient, clb: CallbackQuery):
@Client.on_callback_query(filters.regex("sub_block_[\s\S]*"))
async def callback_query_block(app: PyroClient, clb: CallbackQuery):
user = await app.find_user(clb.from_user)
fullclb = str(clb.data).split("_")
user_locale = clb.from_user.language_code
await app.send_message(
int(fullclb[2]),
app._("sub_blocked", "message"),
app._(
"sub_blocked",
"message",
locale=(await app.find_user(int(fullclb[2]))).locale,
),
)
PosterUser(int(fullclb[2])).block()
await user.block()
await clb.answer(
text=app._("sub_block", "callback", locale=user_locale).format(fullclb[2]),
text=app._("sub_block", "callback", locale=user.locale).format(fullclb[2]),
show_alert=True,
)
@ -211,7 +229,7 @@ async def callback_query_block(app: PyroClient, clb: CallbackQuery):
clb.message.reply_markup.inline_keyboard[0],
[
InlineKeyboardButton(
text=str(app._("sub_unblock", "button", locale=user_locale)),
text=str(app._("sub_unblock", "button", locale=user.locale)),
callback_data=f"sub_unblock_{fullclb[2]}",
)
],
@ -230,15 +248,22 @@ async def callback_query_block(app: PyroClient, clb: CallbackQuery):
@Client.on_callback_query(filters.regex("sub_unblock_[\s\S]*"))
async def callback_query_unblock(app: PyroClient, clb: CallbackQuery):
user = await app.find_user(clb.from_user)
fullclb = str(clb.data).split("_")
user_locale = clb.from_user.language_code
await app.send_message(int(fullclb[2]), app._("sub_unblocked", "message"))
await app.send_message(
int(fullclb[2]),
app._(
"sub_unblocked",
"message",
locale=(await app.find_user(int(fullclb[2]))).locale,
),
)
PosterUser(int(fullclb[2])).unblock()
await user.unblock()
await clb.answer(
text=app._("sub_unblock", "callback", locale=user_locale).format(fullclb[2]),
text=app._("sub_unblock", "callback", locale=user.locale).format(fullclb[2]),
show_alert=True,
)
@ -246,7 +271,7 @@ async def callback_query_unblock(app: PyroClient, clb: CallbackQuery):
clb.message.reply_markup.inline_keyboard[0],
[
InlineKeyboardButton(
text=str(app._("sub_block", "button", locale=user_locale)),
text=str(app._("sub_block", "button", locale=user.locale)),
callback_data=f"sub_block_{fullclb[2]}",
)
],

View File

@ -18,16 +18,18 @@ async def cmd_kill(app: PyroClient, msg: Message):
if msg.from_user.id not in app.admins:
return
user = await app.find_user(msg.from_user)
if len(USERS_WITH_CONTEXT) > 0:
await msg.reply_text(
app._("shutdown_confirm", "message").format(len(USERS_WITH_CONTEXT)),
app._("shutdown_confirm", "message", locale=user.locale).format(
len(USERS_WITH_CONTEXT)
),
reply_markup=InlineKeyboardMarkup(
[
[
InlineKeyboardButton(
app._(
"shutdown", "button", locale=msg.from_user.language_code
),
app._("shutdown", "button", locale=user.locale),
callback_data="shutdown",
)
]

View File

@ -3,22 +3,25 @@ from pyrogram.client import Client
from pyrogram.types import Message
from classes.pyroclient import PyroClient
from classes.user import PosterUser
@Client.on_message(~filters.scheduled & filters.command(["start"], prefixes="/"))
async def cmd_start(app: PyroClient, msg: Message):
if PosterUser(msg.from_user.id).is_blocked():
user = await app.find_user(msg.from_user)
if user.banned:
return
await msg.reply_text(app._("start", "message", locale=msg.from_user.language_code))
await msg.reply_text(app._("start", "message", locale=user.locale))
@Client.on_message(
~filters.scheduled & filters.command(["rules", "help"], prefixes="/")
)
async def cmd_rules(app: PyroClient, msg: Message):
if PosterUser(msg.from_user.id).is_blocked():
user = await app.find_user(msg.from_user)
if user.banned:
return
await msg.reply_text(app._("rules", "message", locale=msg.from_user.language_code))
await msg.reply_text(app._("rules", "message", locale=user.locale))

View File

@ -48,9 +48,9 @@ async def cmd_import(app: PyroClient, msg: Message):
else:
return
await msg.reply_text(
app._("import_request", "message", locale=msg.from_user.language_code)
)
user = await app.find_user(msg.from_user)
await msg.reply_text(app._("import_request", "message", locale=user.locale))
answer = await listen_message(app, msg.chat.id, timeout=600)
@ -58,15 +58,13 @@ async def cmd_import(app: PyroClient, msg: Message):
if answer is None:
await msg.reply_text(
app._("import_ignored", "message", locale=msg.from_user.language_code),
app._("import_ignored", "message", locale=user.locale),
quote=True,
)
return
if answer.text == "/cancel":
await answer.reply_text(
app._("import_abort", "message", locale=msg.from_user.language_code)
)
await answer.reply_text(app._("import_abort", "message", locale=user.locale))
return
if answer.document is None:
@ -74,7 +72,7 @@ async def cmd_import(app: PyroClient, msg: Message):
app._(
"import_invalid_media",
"message",
locale=msg.from_user.language_code,
locale=user.locale,
),
quote=True,
)
@ -82,16 +80,14 @@ async def cmd_import(app: PyroClient, msg: Message):
if answer.document.mime_type != "application/zip":
await answer.reply_text(
app._("import_invalid_mime", "message", locale=msg.from_user.language_code),
app._("import_invalid_mime", "message", locale=user.locale),
quote=True,
)
return
if disk_usage(getcwd())[2] < (answer.document.file_size) * 3:
await msg.reply_text(
app._(
"import_too_big", "message", locale=msg.from_user.language_code
).format(
app._("import_too_big", "message", locale=user.locale).format(
answer.document.file_size // (2**30),
disk_usage(getcwd())[2] // (2**30),
)
@ -111,14 +107,12 @@ async def cmd_import(app: PyroClient, msg: Message):
tmp_path = Path(f"{app.config['locations']['tmp']}/{answer.document.file_id}")
downloading = await answer.reply_text(
app._("import_downloading", "message", locale=msg.from_user.language_code),
app._("import_downloading", "message", locale=user.locale),
quote=True,
)
await app.download_media(answer, file_name=str(tmp_path))
await downloading.edit(
app._("import_unpacking", "message", locale=msg.from_user.language_code)
)
await downloading.edit(app._("import_unpacking", "message", locale=user.locale))
try:
with ZipFile(tmp_path, "r") as handle:
@ -137,17 +131,15 @@ async def cmd_import(app: PyroClient, msg: Message):
format_exc(),
)
await answer.reply_text(
app._(
"import_unpack_error", "message", locale=msg.from_user.language_code
).format(exp, format_exc())
app._("import_unpack_error", "message", locale=user.locale).format(
exp, format_exc()
)
)
return
logger.info("Downloaded '%s' - awaiting upload", answer.document.file_name)
await downloading.edit(
app._("import_uploading", "message", locale=msg.from_user.language_code)
)
await downloading.edit(app._("import_uploading", "message", locale=user.locale))
remove(tmp_path)
@ -184,7 +176,7 @@ async def cmd_import(app: PyroClient, msg: Message):
app._(
"import_upload_error_other",
"message",
locale=msg.from_user.language_code,
locale=user.locale,
).format(path.basename(filename)),
disable_notification=True,
)
@ -205,7 +197,7 @@ async def cmd_import(app: PyroClient, msg: Message):
app._(
"import_upload_error_duplicate",
"message",
locale=msg.from_user.language_code,
locale=user.locale,
).format(path.basename(filename)),
disable_notification=True,
)
@ -214,7 +206,7 @@ async def cmd_import(app: PyroClient, msg: Message):
app._(
"import_upload_error_other",
"message",
locale=msg.from_user.language_code,
locale=user.locale,
).format(path.basename(filename)),
disable_notification=True,
)
@ -235,7 +227,7 @@ async def cmd_import(app: PyroClient, msg: Message):
rmtree(Path(f"{app.config['locations']['tmp']}/{tmp_dir}"), ignore_errors=True)
await answer.reply_text(
app._("import_finished", "message", locale=msg.from_user.language_code),
app._("import_finished", "message", locale=user.locale),
quote=True,
)
@ -260,9 +252,9 @@ async def cmd_remove(app: PyroClient, msg: Message):
else:
return
await msg.reply_text(
app._("remove_request", "message", locale=msg.from_user.language_code)
)
user = await app.find_user(msg.from_user)
await msg.reply_text(app._("remove_request", "message", locale=user.locale))
answer_id = await listen_message(app, msg.chat.id, timeout=600)
@ -270,28 +262,22 @@ async def cmd_remove(app: PyroClient, msg: Message):
if answer_id is None:
await msg.reply_text(
app._("remove_ignored", "message", locale=msg.from_user.language_code),
app._("remove_ignored", "message", locale=user.locale),
quote=True,
)
return
if answer_id.text == "/cancel":
await answer_id.reply_text(
app._("remove_abort", "message", locale=msg.from_user.language_code)
)
await answer_id.reply_text(app._("remove_abort", "message", locale=user.locale))
return
await msg.reply_text(
app._("remove_kind", "message", locale=msg.from_user.language_code),
app._("remove_kind", "message", locale=user.locale),
reply_markup=ReplyKeyboardMarkup(
[
[
KeyboardButton(
app._("photo", "button", locale=msg.from_user.language_code)
),
KeyboardButton(
app._("video", "button", locale=msg.from_user.language_code)
),
KeyboardButton(app._("photo", "button", locale=user.locale)),
KeyboardButton(app._("video", "button", locale=user.locale)),
]
],
resize_keyboard=True,
@ -307,7 +293,7 @@ async def cmd_remove(app: PyroClient, msg: Message):
if answer_kind is None:
await msg.reply_text(
app._("remove_ignored", "message", locale=msg.from_user.language_code),
app._("remove_ignored", "message", locale=user.locale),
quote=True,
reply_markup=ReplyKeyboardRemove(),
)
@ -315,7 +301,7 @@ async def cmd_remove(app: PyroClient, msg: Message):
if answer_kind.text == "/cancel":
await answer_kind.reply_text(
app._("remove_abort", "message", locale=msg.from_user.language_code),
app._("remove_abort", "message", locale=user.locale),
reply_markup=ReplyKeyboardRemove(),
)
return
@ -326,11 +312,9 @@ async def cmd_remove(app: PyroClient, msg: Message):
func = video_delete
else:
await answer_kind.reply_text(
app._(
"remove_unknown", "message", locale=msg.from_user.language_code
).format(
app._("photo", "button", locale=msg.from_user.language_code),
app._("video", "button", locale=msg.from_user.language_code),
app._("remove_unknown", "message", locale=user.locale).format(
app._("photo", "button", locale=user.locale),
app._("video", "button", locale=user.locale),
),
reply_markup=ReplyKeyboardRemove(),
)
@ -346,9 +330,9 @@ async def cmd_remove(app: PyroClient, msg: Message):
answer_id.from_user.id,
)
await answer_kind.reply_text(
app._(
"remove_success", "message", locale=msg.from_user.language_code
).format(answer_id.text),
app._("remove_success", "message", locale=user.locale).format(
answer_id.text
),
reply_markup=ReplyKeyboardRemove(),
)
else:
@ -359,9 +343,9 @@ async def cmd_remove(app: PyroClient, msg: Message):
answer_id.from_user.id,
)
await answer_kind.reply_text(
app._(
"remove_failure", "message", locale=msg.from_user.language_code
).format(answer_id.text),
app._("remove_failure", "message", locale=user.locale).format(
answer_id.text
),
reply_markup=ReplyKeyboardRemove(),
)

View File

@ -1,7 +1,8 @@
from pyrogram.client import Client
from pyrogram import filters
from pyrogram.types import Message, User
from libbot import sync
from pyrogram import filters
from pyrogram.client import Client
from pyrogram.types import Message, User
from classes.pyroclient import PyroClient
@ -12,14 +13,16 @@ from classes.pyroclient import PyroClient
& filters.command(["report"], prefixes=["", "/"])
)
async def command_report(app: PyroClient, msg: Message):
if msg.reply_to_message.forward_from_chat.id == app.config["posting"]["channel"]:
if msg.reply_to_message.forward_from_chat.id != app.config["posting"]["channel"]:
return
user = await app.find_user(msg.from_user)
await msg.reply_text(
app._(
"report_sent",
"message",
locale=msg.from_user.language_code
if msg.from_user is not None
else None,
locale=user.locale if msg.from_user is not None else None,
)
)
@ -28,14 +31,10 @@ async def command_report(app: PyroClient, msg: Message):
report_sent = await msg.reply_to_message.forward(app.owner)
sender = msg.from_user if msg.from_user is not None else msg.sender_chat
sender_name = (
sender.first_name if isinstance(sender, User) else sender.title
)
sender_name = sender.first_name if isinstance(sender, User) else sender.title
# ACTION NEEDED
# Name and username are somehow None
await report_sent.reply_text(
app._("report_received", "message").format(
app._("report_received", "message", locale=user.locale).format(
sender_name, sender.username, sender.id
),
quote=True,

View File

@ -13,7 +13,6 @@ from pyrogram.types import InlineKeyboardButton, InlineKeyboardMarkup, Message
from classes.enums.submission_types import SubmissionType
from classes.exceptions import SubmissionDuplicatesError, SubmissionUnsupportedError
from classes.pyroclient import PyroClient
from classes.user import PosterUser
from modules.database import col_banned, col_submitted
from modules.utils import USERS_WITH_CONTEXT
@ -35,20 +34,22 @@ async def get_submission(app: PyroClient, msg: Message):
if msg.from_user.id in USERS_WITH_CONTEXT:
return
user = await app.find_user(msg.from_user)
user_owner = await app.find_user(app.owner)
try:
if col_banned.find_one({"user": msg.from_user.id}) is not None:
return
await app.send_chat_action(msg.chat.id, ChatAction.TYPING)
user_locale = msg.from_user.language_code
save_tmp = True
contents = None
if PosterUser(msg.from_user.id).is_limited():
if await user.is_limited():
await msg.reply_text(
app._("sub_cooldown", "message", locale=user_locale).format(
str(app.config["submission"]["timeout"])
app._("sub_cooldown", "message", locale=user.locale).format(
app.config["submission"]["timeout"]
)
)
return
@ -63,7 +64,7 @@ async def get_submission(app: PyroClient, msg: Message):
)
if msg.document.mime_type not in app.config["submission"]["mime_types"]:
await msg.reply_text(
app._("mime_not_allowed", "message", locale=user_locale).format(
app._("mime_not_allowed", "message", locale=user.locale).format(
", ".join(app.config["submission"]["mime_types"])
),
quote=True,
@ -71,8 +72,8 @@ async def get_submission(app: PyroClient, msg: Message):
return
if msg.document.file_size > app.config["submission"]["file_size"]:
await msg.reply_text(
app._("document_too_large", "message", locale=user_locale).format(
str(app.config["submission"]["file_size"] / 1024 / 1024)
app._("document_too_large", "message", locale=user.locale).format(
app.config["submission"]["file_size"] / 1024 / 1024
),
quote=True,
)
@ -93,8 +94,8 @@ async def get_submission(app: PyroClient, msg: Message):
)
if msg.video.file_size > app.config["submission"]["file_size"]:
await msg.reply_text(
app._("document_too_large", "message", locale=user_locale).format(
str(app.config["submission"]["file_size"] / 1024 / 1024)
app._("document_too_large", "message", locale=user.locale).format(
app.config["submission"]["file_size"] / 1024 / 1024
),
quote=True,
)
@ -112,7 +113,7 @@ async def get_submission(app: PyroClient, msg: Message):
# )
# if msg.animation.file_size > app.config["submission"]["file_size"]:
# await msg.reply_text(
# app._("document_too_large", "message", locale=user_locale).format(
# app._("document_too_large", "message", locale=user.locale).format(
# str(app.config["submission"]["file_size"] / 1024 / 1024)
# ),
# quote=True,
@ -179,7 +180,7 @@ async def get_submission(app: PyroClient, msg: Message):
buttons = [
[
InlineKeyboardButton(
text=app._("sub_yes", "button"),
text=app._("sub_yes", "button", locale=user_owner.locale),
callback_data=f"sub_yes_{str(inserted.inserted_id)}",
)
]
@ -189,7 +190,7 @@ async def get_submission(app: PyroClient, msg: Message):
caption = str(msg.caption)
buttons[0].append(
InlineKeyboardButton(
text=app._("sub_yes_caption", "button"),
text=app._("sub_yes_caption", "button", locale=user_owner.locale),
callback_data=f"sub_yes_{str(inserted.inserted_id)}_caption",
)
)
@ -198,11 +199,11 @@ async def get_submission(app: PyroClient, msg: Message):
buttons[0].append(
InlineKeyboardButton(
text=app._("sub_no", "button"),
text=app._("sub_no", "button", locale=user_owner.locale),
callback_data=f"sub_no_{str(inserted.inserted_id)}",
)
)
caption += app._("sub_by", "message")
caption += app._("sub_by", "message", locale=user_owner.locale)
if msg.from_user.first_name is not None:
caption += f" {msg.from_user.first_name}"
@ -220,7 +221,7 @@ async def get_submission(app: PyroClient, msg: Message):
try:
submitted = await app.submit_media(str(inserted.inserted_id))
await msg.reply_text(
app._("sub_yes_auto", "message", locale=user_locale),
app._("sub_yes_auto", "message", locale=user.locale),
disable_notification=True,
quote=True,
)
@ -230,7 +231,7 @@ async def get_submission(app: PyroClient, msg: Message):
return
except SubmissionUnsupportedError:
await msg.reply_text(
app._("mime_not_allowed", "message", locale=user_locale).format(
app._("mime_not_allowed", "message", locale=user.locale).format(
", ".join(app.config["submission"]["mime_types"]), quote=True
),
quote=True,
@ -239,7 +240,7 @@ async def get_submission(app: PyroClient, msg: Message):
except SubmissionDuplicatesError as exp:
await msg.reply_text(
app._(
"sub_media_duplicates_list", "message", locale=user_locale
"sub_media_duplicates_list", "message", locale=user.locale
).format("\n".join(exp.duplicates)),
quote=True,
)
@ -254,7 +255,7 @@ async def get_submission(app: PyroClient, msg: Message):
try:
submitted = await app.submit_photo(str(inserted.inserted_id))
await msg.reply_text(
app._("sub_yes_auto", "message", locale=user_locale),
app._("sub_yes_auto", "message", locale=user.locale),
disable_notification=True,
quote=True,
)
@ -264,22 +265,22 @@ async def get_submission(app: PyroClient, msg: Message):
return
except SubmissionUnsupportedError:
await msg.reply_text(
app._("mime_not_allowed", "message", locale=user_locale).format(
app._("mime_not_allowed", "message", locale=user.locale).format(
", ".join(app.config["submission"]["mime_types"]), quote=True
)
)
return
except SubmissionDuplicatesError as exp:
await msg.reply_text(
app._("sub_dup", "message", locale=user_locale), quote=True
app._("sub_dup", "message", locale=user.locale), quote=True
)
return
except Exception as exp:
await app.send_message(
app.owner,
app._("sub_error_admin", "message").format(
msg.from_user.id, format_exc()
),
app._(
"sub_error_admin", "message", locale=user_owner.locale
).format(msg.from_user.id, format_exc()),
)
await msg.reply_text("sub_error", quote=True)
return
@ -288,17 +289,17 @@ async def get_submission(app: PyroClient, msg: Message):
buttons += [
[
InlineKeyboardButton(
text=app._("sub_block", "button"),
text=app._("sub_block", "button", locale=user_owner.locale),
callback_data=f"sub_block_{msg.from_user.id}",
)
]
]
PosterUser(msg.from_user.id).limit()
await user.update_cooldown()
if msg.from_user.id != app.owner:
await msg.reply_text(
app._("sub_sent", "message", locale=user_locale),
app._("sub_sent", "message", locale=user.locale),
disable_notification=True,
quote=True,
)

42
plugins/language.py Normal file
View File

@ -0,0 +1,42 @@
from pykeyboard import InlineButton, InlineKeyboard
from pyrogram import filters
from pyrogram.client import Client
from pyrogram.types import CallbackQuery, Message
from classes.pyroclient import PyroClient
@Client.on_message(
~filters.scheduled & filters.private & filters.command(["language"], prefixes=["/"]) # type: ignore
)
async def command_language(app: PyroClient, message: Message):
user = await app.find_user(message.from_user)
keyboard = InlineKeyboard(row_width=2)
buttons = []
for locale, data in app.in_every_locale("metadata").items():
buttons.append(
InlineButton(f"{data['flag']} {data['name']}", f"language:{locale}")
)
keyboard.add(*buttons)
await message.reply_text(
app._("locale_choice", "messages", locale=user.locale),
reply_markup=keyboard,
)
@Client.on_callback_query(filters.regex(r"language:[\s\S]*")) # type: ignore
async def callback_language(app: PyroClient, callback: CallbackQuery):
user = await app.find_user(callback.from_user)
language = str(callback.data).split(":")[1]
await user.update_locale(language)
await callback.answer(
app._("locale_set", "callbacks", locale=language).format(
locale=app._("name", "metadata", locale=language)
),
show_alert=True,
)