This commit closes #13

This commit is contained in:
Profitroll 2023-08-14 12:11:53 +02:00
parent 7a587ececa
commit fdd5ad7ff0
Signed by: profitroll
GPG Key ID: FA35CAB49DACD3B2
8 changed files with 91 additions and 18 deletions

View File

@ -22,6 +22,7 @@ class PyroGroup:
id: int
locale: Union[str, None]
locale_auto: bool
ban_failed: bool
@classmethod
async def create_if_not_exists(
@ -29,6 +30,7 @@ class PyroGroup:
id: int,
locale: Union[str, None] = sync.config_get("locale", "defaults", "group"),
locale_auto: bool = sync.config_get("locale_auto", "defaults", "group"),
ban_failed: bool = sync.config_get("ban_failed", "defaults", "group"),
):
db_entry = await col_groups.find_one(
{
@ -42,6 +44,7 @@ class PyroGroup:
"id": id,
"locale": locale,
"locale_auto": locale_auto,
"ban_failed": ban_failed,
}
)
db_entry = await col_groups.find_one({"_id": inserted.inserted_id})
@ -65,6 +68,16 @@ class PyroGroup:
{"_id": self._id}, {"$set": {"locale_auto": enabled}}
)
async def set_ban_failed(self, enabled: bool) -> None:
logger.debug(
"Banning users that failed the captcha in group %s has been set to %s",
self.id,
enabled,
)
await col_groups.update_one(
{"_id": self._id}, {"$set": {"ban_failed": enabled}}
)
# Group settings
# User locale
def select_locale(

View File

@ -14,11 +14,14 @@
},
"commands": {
"help": "Nützliche Informationen anzeigen",
"ban_failed": "Umschaltverbot bei fehlender Verifizierung",
"language": "Sprache des Bots in dieser Gruppe einstellen",
"language_auto": "Automatische Sprachwahl umschalten"
},
"messages": {
"help": "Hallo! Dieser Bot kann Ihnen helfen, sich gegen nervige Bots zu schützen, indem er Emoji-Captchas beim Beitritt erzwingt.\n\nFügen Sie diesen Bot zu Ihrer gewünschten Gruppe hinzu und geben Sie ihm die Berechtigungen **Nachrichten löschen** und **Benutzer sperren**. Sie können auch die Sprache des Bots in der Gruppe mit /language ändern und die automatische Sprache pro Benutzer mit /language_auto umschalten.\n\nDieser Bot ist Open Source, also können Sie ihn selbst hosten oder nach Ihren Wünschen verändern: https://git.end-play.xyz/profitroll/EmojiCaptchaBot\n\nWenn Sie einen Fehler melden oder einen Vorschlag machen möchten besuchen Sie die [Problemseite](https://git.end-play.xyz/profitroll/EmojiCaptchaBot/issues) des Bots auf git.\n\nEinen schönen Tag noch :)",
"help": "Hallo! Dieser Bot kann Ihnen helfen, sich gegen nervige Bots zu schützen, indem er Emoji-Captchas beim Beitritt erzwingt.\n\nFügen Sie diesen Bot zu Ihrer gewünschten Gruppe hinzu und geben Sie ihm die Berechtigungen **Nachrichten löschen** und **Benutzer sperren**.\n\nVerfügbare Kommandos:\n• /ban_failed - Bannen bei Zeitüberschreitung oder fehlgeschlagenem Captcha umschalten\n• /language - Sprache des Bots ändern\n• /language_auto - automatische Sprache pro Benutzer umschalten\n\nKommandos, die oben aufgelistet sind, müssen in der Gruppe von ihren Admins oder dem Besitzer ausgeführt werden.\n\nDieser Bot ist Open Source, also können Sie ihn selbst hosten oder nach Ihren Wünschen verändern: https://git.end-play.xyz/profitroll/EmojiCaptchaBot\n\nWenn Sie einen Fehler melden oder einen Vorschlag machen möchten besuchen Sie die [Problemseite](https://git.end-play.xyz/profitroll/EmojiCaptchaBot/issues) des Bots auf git.\n\nEinen schönen Tag noch :)",
"ban_failed_disabled": "Bann bei Fehlschlag wurde **deaktiviert**. Wenn Benutzer die Überprüfung ignorieren oder sie nicht bestehen, bleiben sie in der Gruppe nur zum Lesen zugelassen.",
"ban_failed_enabled": "Bann bei Fehlschlag wurde **aktiviert**. Wenn Benutzer die Überprüfung ignorieren oder sie nicht bestehen, werden sie sofort aus der Gruppe geworfen und gesperrt.",
"locale_auto_disabled": "Die automatische Sprachauswahl auf individueller Basis wurde **deaktiviert**. Bitte beachten Sie, dass jetzt die Sprache der Gruppe verwendet wird. Sie können sie mit /language ändern. Wenn \"Standard\" als Sprache eingestellt ist, wird die Standardsprache des Bots ({default_locale}) verwendet.",
"locale_auto_enabled": "Die automatische Sprachauswahl auf individueller Basis wurde **aktiviert**. Bitte beachten Sie, dass die Standardsprache der Gruppe oder des Bots verwendet wird, wenn der Bot kein Sprachpaket für den Benutzer hat. Sie können die Sprache der Gruppe mit /language ändern. Wenn \"Standard\" als Sprache eingestellt ist, wird die Standardsprache des Bots ({default_locale}) verwendet.",
"locale_choice": "Okay. Bitte wählen Sie die Sprache mit der Tastatur unten aus.\n\nBitte beachten Sie, dass Sie auch die automatische Sprachauswahl für jeden Benutzer mit dem Befehl /language_auto",

View File

@ -11,11 +11,14 @@
},
"commands": {
"help": "Show some helpful info",
"language": "Set bot's language in this group",
"language_auto": "Toggle automatic language selection"
"ban_failed": "Toggle ban on verification fail",
"language_auto": "Toggle automatic language selection",
"language": "Set bot's language in this group"
},
"messages": {
"help": "Hi! This bot can help you protect against annoying bots by enforcing emoji captcha on join.\n\nIn order to it, add this bot to your desired group and provide it with **Delete Messages** and **Ban Users** permissions. You can also change bot's language in the group using /language and toggle automatic per-user language using /language_auto.\n\nThis bot is open source, so feel free to host it yourself or modify to your liking: https://git.end-play.xyz/profitroll/EmojiCaptchaBot\n\nIf you want to file an issue or make a suggestion visit bot's [issues page](https://git.end-play.xyz/profitroll/EmojiCaptchaBot/issues) on bot's git.\n\nHave a nice day :)",
"help": "Hi! This bot can help you protect against annoying bots by enforcing emoji captcha on join.\n\nIn order to it, add this bot to your desired group and provide it with **Delete Messages** and **Ban Users** permissions.\n\nAvailable commands:\n• /ban_failed - toggle banning on timeout or captcha fail\n• /language - change bot's language\n• /language_auto - toggle automatic per-user language\n\nCommands listed above must be executes in group by its admins or owner.\n\nThis bot is open source, so feel free to host it yourself or modify to your liking: https://git.end-play.xyz/profitroll/EmojiCaptchaBot\n\nIf you want to file an issue or make a suggestion visit bot's [issues page](https://git.end-play.xyz/profitroll/EmojiCaptchaBot/issues) on bot's git.\n\nHave a nice day :)",
"ban_failed_disabled": "Ban on fail has been **disabled**. If users ignore verification or fail it, they will remain read-only in the group.",
"ban_failed_enabled": "Ban on fail has been **enabled**. If users ignore verification or fail it, they will be immediately kicked from the group and banned.",
"locale_auto_disabled": "Automatic language selection on per-user basis has been **disabled**. Please note that group's language will now be used. You can change it using /language. If \"Default\" language is set bot's default language ({default_locale}) will be used.",
"locale_auto_enabled": "Automatic language selection on per-user basis has been **enabled**. Please note that if bot does not have a language pack for user group's or bot's default language will be used. You can change group's language using /language. If \"Default\" language is set bot's default language ({default_locale}) will be used.",
"locale_choice": "Alright. Please choose the language using keyboard below.\n\nNote, that you can also enable automatic language selection on per-user basis using command /language_auto",

View File

@ -10,11 +10,14 @@
},
"commands": {
"help": "Показати корисну інформацію",
"ban_failed": "Переключити бан при невдалій верифікації",
"language": "Встановити мову бота в цій групі",
"language_auto": "Переключити автоматичний вибір мови"
},
"messages": {
"help": "Привіт! Цей бот допоможе вам захиститися від настирливих ботів, запровадивши емодзі-капчу при вході в групу.\n\nДля цього додайте цього бота в потрібну вам групу і надайте йому права **Видаляти повідомлення** та **Банити користувачів**. Ви також можете змінити мову бота в групі за допомогою /language та увімкнути автоматичну мову для кожного користувача за допомогою /language_auto.\n\nЦей бот має відкритий вихідний код, тому ви можете хостити його самостійно або модифікувати під себе: https://git.end-play.xyz/profitroll/EmojiCaptchaBot\n\nЯкщо ви хочете повідомити про проблему або внести пропозицію відвідайте [сторінку підтримки бота](https://git.end-play.xyz/profitroll/EmojiCaptchaBot/issues) на гіті бота.\n\nГарного дня :)",
"help": "Привіт! Цей бот допоможе вам захиститися від настирливих ботів, запровадивши емодзі-капчу при вході в групу.\n\nДля цього додайте цього бота в потрібну вам групу і надайте йому права **Видаляти повідомлення** та **Банити користувачів**.\n\nДоступні команди:\n• /ban_failed - увімкнути заборону після таймауту або невдалого введення капчі\n• /language - змінити мову бота\n• /language_auto - перемкнути автоматичну мову для кожного користувача\n\nКоманди, перераховані вище, повинні виконуватися в групі її адміністраторами або власником.\n\nЦей бот має відкритий вихідний код, тому ви можете хостити його самостійно або модифікувати під себе: https://git.end-play.xyz/profitroll/EmojiCaptchaBot\n\nЯкщо ви хочете повідомити про проблему або внести пропозицію відвідайте [сторінку підтримки бота](https://git.end-play.xyz/profitroll/EmojiCaptchaBot/issues) на гіті бота.\n\nГарного дня :)",
"ban_failed_disabled": "Блокування при невдалій верифікації було **вимкнено**. Якщо користувачі проігнорують верифікацію або не пройдуть її, вони залишаться в групі тільки для читання.",
"ban_failed_enabled": "Блокування при невдачі було **увімкнено**. Якщо користувач проігнорує верифікацію або не пройде її, його буде негайно вигнано з групи та забанено.",
"locale_auto_disabled": "Автоматичний вибір мови для кожного користувача було **вимкнено**. Зверніть увагу, що тепер буде використовуватися мова групи. Ви можете змінити її за допомогою /language. Якщо встановлено мову \"За замовчуванням\" буде використовуватися мова бота за замовчуванням ({default_locale}).",
"locale_auto_enabled": "Автоматичний вибір мови для кожного користувача був **увімкнений**. Зверніть увагу, що якщо бот не має мовного пакета для користувача буде використовуватися мова групи або бота за замовчуванням. Ви можете змінити мову групи за допомогою /language. Якщо встановлено мову \"За замовчуванням\" буде використовуватися мова бота за замовчуванням ({default_locale}).",
"locale_choice": "Гаразд. Будь ласка, оберіть мову за допомогою клавіатури нижче.\n\nЗауважте, що ви також можете увімкнути автоматичний вибір мови для кожного користувача за допомогою команди /language_auto",

View File

@ -1,18 +1,21 @@
from pyrogram.types import Message
from classes.pyroclient import PyroClient
from classes.pyrogroup import PyroGroup
async def kick_unstarted(
app: PyroClient, user_id: int, group_id: int, message_id: int
) -> None:
user = await app.find_user(user_id, group_id)
group = await PyroGroup.create_if_not_exists(group_id)
if user.score == 0 and user.failed == 0:
banned = await app.ban_chat_member(group_id, user_id)
if group.ban_failed:
banned = await app.ban_chat_member(group_id, user_id)
if isinstance(banned, Message):
await banned.delete()
if isinstance(banned, Message):
await banned.delete()
await app.delete_messages(group_id, message_id)
@ -21,11 +24,13 @@ async def kick_unverified(
app: PyroClient, user_id: int, group_id: int, message_id: int
) -> None:
user = await app.find_user(user_id, group_id)
group = await PyroGroup.create_if_not_exists(group_id)
if user.score < 6 or user.failed:
banned = await app.ban_chat_member(group_id, user_id)
if group.ban_failed:
banned = await app.ban_chat_member(group_id, user_id)
if isinstance(banned, Message):
await banned.delete()
if isinstance(banned, Message):
await banned.delete()
await app.delete_messages(group_id, message_id)

View File

@ -82,10 +82,11 @@ async def callback_emoji_button(app: PyroClient, callback: CallbackQuery):
app._("captcha_failed", "callbacks", locale=locale), show_alert=True
)
banned = await app.ban_chat_member(group.id, callback.from_user.id)
if group.ban_failed:
banned = await app.ban_chat_member(group.id, callback.from_user.id)
if isinstance(banned, Message):
await banned.delete()
if isinstance(banned, Message):
await banned.delete()
await callback.message.delete()
return

View File

@ -0,0 +1,44 @@
import logging
from pyrogram import filters
from pyrogram.enums.chat_member_status import ChatMemberStatus
from pyrogram.types import Message
from classes.pyroclient import PyroClient
from classes.pyrogroup import PyroGroup
logger = logging.getLogger(__name__)
@PyroClient.on_message(
~filters.scheduled
& filters.group
& filters.command(["ban_failed"], prefixes=["/"]) # type: ignore
)
async def command_ban_failed(app: PyroClient, message: Message):
group = await PyroGroup.create_if_not_exists(message.chat.id, None, True)
locale = group.select_locale(app, message.from_user)
if (await app.get_chat_member(group.id, message.from_user.id)).status not in [
ChatMemberStatus.ADMINISTRATOR,
ChatMemberStatus.OWNER,
]:
await message.reply_text(
app._("permission_denied", "messages", locale=locale), quote=True
)
return
if group.ban_failed:
await message.reply_text(
app._("ban_failed_disabled", "messages", locale=locale)
)
else:
await message.reply_text(app._("ban_failed_enabled", "messages", locale=locale))
logger.info(
"Ban on fail in group %s has been set to %s",
group.id,
not group.ban_failed,
)
await group.set_ban_failed(not group.ban_failed)

View File

@ -50,10 +50,11 @@ async def handler_user_join(app: PyroClient, message: Message):
"User %s has previously failed the captcha, kicking and banning him",
user.id,
)
banned = await app.ban_chat_member(group.id, user.id)
if group.ban_failed:
banned = await app.ban_chat_member(group.id, user.id)
if isinstance(banned, Message):
await banned.delete()
if isinstance(banned, Message):
await banned.delete()
return
@ -101,7 +102,7 @@ async def handler_user_join(app: PyroClient, message: Message):
kick_unstarted,
"date",
[app, user.id, verification_request.chat.id, verification_request.id],
run_date=datetime.now() + timedelta(seconds=app.config["timeouts"]["join"]),
run_date=datetime.now() + timedelta(seconds=group.timeout_join),
)
await col_schedule.insert_one(
{"user": user.id, "group": group.id, "job_id": job.id}