From 0439a8f4b829f141b864b1cba06b0baa22b5f1fc Mon Sep 17 00:00:00 2001 From: cloudreflection <39823819+cloudreflection@users.noreply.github.com> Date: Mon, 6 Jun 2022 20:20:41 +0800 Subject: [PATCH] =?UTF-8?q?pmcaptcha=20=E4=B8=80=E4=B8=AA=E7=AE=80?= =?UTF-8?q?=E5=8D=95=E7=9A=84=E7=A7=81=E8=81=8A=E4=BA=BA=E6=9C=BA=E9=AA=8C?= =?UTF-8?q?=E8=AF=81=20(#6)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit * pmcaptcha Co-authored-by: omg-xtao <100690902+omg-xtao@users.noreply.github.com> --- list.json | 10 +++ pmcaptcha/main.py | 183 ++++++++++++++++++++++++++++++++++++++++++++++ 2 files changed, 193 insertions(+) create mode 100644 pmcaptcha/main.py diff --git a/list.json b/list.json index d0c1687..6cd68f8 100644 --- a/list.json +++ b/list.json @@ -70,6 +70,16 @@ "des-short": "Base64编码/解码。", "des": "Base64编码/解码。\n命令：,b64e ,b64d" }, + { + "name": "pmcaptcha", + "version": "1.00", + "section": "chat", + "maintainer": "cloudreflection", + "size": "8.27 kb", + "supported": true, + "des-short": "简单的私聊人机验证", + "des": "简单的私聊人机验证，支持自定义关键词黑名单，通过后欢迎语，验证超时时间" + }, { "name": "bc", "version": "1.232", diff --git a/pmcaptcha/main.py b/pmcaptcha/main.py new file mode 100644 index 0000000..0df1b47 --- /dev/null +++ b/pmcaptcha/main.py @@ -0,0 +1,183 @@ +# pmcaptcha - a pagermaid-pyro plugin by cloudreflection +# https://t.me/cloudreflection_channel/268 +# ver 2022/06/06 + +from pyrogram import Client +from pyrogram.enums.chat_type import ChatType +from pyrogram.raw.functions.account import UpdateNotifySettings +from pyrogram.raw.types import InputNotifyPeer, InputPeerNotifySettings + +from pagermaid.utils import Message +from pagermaid.listener import listener +from pagermaid.single_utils import sqlite +from pagermaid.sub_utils import Sub + +import asyncio +import random + +captcha_success = Sub("pmcaptcha.success") + + +@listener(is_plugin=False, incoming=False, outgoing=True, ignore_edited=True, privates_only=True) +async def process_pm_captcha_self(_: Client, message: Message): + cid = message.chat.id + if message.text: + if message.text[0] == ",": + return + if captcha_success.check_id(cid): + return + else: + return captcha_success.add_id(cid) + + +@listener(is_plugin=False, incoming=True, outgoing=False, ignore_edited=True, privates_only=True) +async def process_pm_captcha(client: Client, message: Message): + cid = message.chat.id + data = sqlite.get("pmcaptcha", {}) + if message.text is not None: + try: + message.text == "114514" + except UnicodeDecodeError: + await message.reply('您触犯了风控规则，已被封禁\n\nYou have violated the risk control rules and been banned') + await client.block_user(user_id=cid) + await asyncio.sleep(random.randint(0, 100) / 1000) + return await client.archive_chats(chat_ids=cid) + if not captcha_success.check_id(cid) and sqlite.get("pmcaptcha." + str(cid)) is None: + # 忽略联系人 + if message.from_user.is_contact: + return captcha_success.add_id(cid) + await client.read_chat_history(message.chat.id) + if message.text is not None: + if data.get("blacklist", ""): + for i in data.get("blacklist", "").split(","): + if i in message.text: + await message.reply( + '您触犯了黑名单规则，已被封禁\n\nYou have violated the blacklist rules and been banned') + await client.block_user(user_id=cid) + await asyncio.sleep(random.randint(0, 100) / 1000) + return await client.archive_chats(chat_ids=cid) + try: + await client.invoke(UpdateNotifySettings(peer=InputNotifyPeer(peer=await client.resolve_peer(cid)), + settings=InputPeerNotifySettings(silent=True))) + except: # noqa + pass + await asyncio.sleep(random.randint(0, 100) / 1000) + await client.archive_chats(chat_ids=cid) + wait = data.get("wait", 20) + key1 = random.randint(1, 10) + key2 = random.randint(1, 10) + await asyncio.sleep(random.randint(0, 100) / 1000) + sqlite['pmcaptcha.' + str(cid)] = str(key1 + key2) + msg = await message.reply( + '已启用私聊验证。请发送 \"' + str(key1) + '+' + str(key2) + '\" 的答案(阿拉伯数字)来与我私聊\n请在' + str(wait) + + '秒内完成验证。您只有一次验证机会\n\nHuman verification is enabled.Please send the answer of this question \"' + + str(key1) + '+' + str(key2) + '\" (numbers only) first.\nYou have ' + str(wait) + + ' seconds to complete the verification.') + await asyncio.sleep(wait) + await msg.safe_delete() # noqa + if sqlite.get('pmcaptcha.' + str(cid)) is not None: + del sqlite['pmcaptcha.' + str(cid)] + await message.reply('验证超时,您已被封禁\n\nVerification timeout.You have been banned.') + await client.block_user(user_id=cid) + await asyncio.sleep(random.randint(0, 100) / 1000) + await client.archive_chats(chat_ids=cid) + elif sqlite.get("pmcaptcha." + str(cid)): + await message.safe_delete() + if message.text == sqlite.get("pmcaptcha." + str(cid)): + del sqlite['pmcaptcha.' + str(cid)] + captcha_success.add_id(cid) + try: + await client.invoke(UpdateNotifySettings(peer=InputNotifyPeer(peer=await client.resolve_peer(cid)), + settings=InputPeerNotifySettings(silent=False))) + except: # noqa + pass + await asyncio.sleep(random.randint(0, 100) / 1000) + msg = await message.reply(data.get("welcome", "验证通过\n\nVerification Passe")) + await asyncio.sleep(random.randint(0, 100) / 1000) + await client.unarchive_chats(chat_ids=cid) + await asyncio.sleep(5) + await msg.safe_delete() # noqa + else: + del sqlite['pmcaptcha.' + str(cid)] + await message.reply('验证错误，您已被封禁\n\nVerification failed.You have been banned.') + await client.block_user(user_id=cid) + await asyncio.sleep(random.randint(0, 100) / 1000) + await client.archive_chats(chat_ids=cid) + + +@listener(is_plugin=True, outgoing=True, command="pmcaptcha", + description='一个简单的私聊人机验证 可用命令列表请见 t.me/cloudreflection_channel/298') +async def pm_captcha(client: Client, message: Message): + cid_ = str(message.chat.id) + data = sqlite.get("pmcaptcha", {}) + if len(message.parameter) == 0: + if message.chat.type != ChatType.PRIVATE: + await message.edit('请在私聊时使用此命令，或添加参数执行') + await asyncio.sleep(3) + await message.safe_delete() + if captcha_success.check_id(message.chat.id): + text = "已验证用户" + else: + text = "未验证/验证中用户" + await message.edit(text) + elif len(message.parameter) == 1: + if message.parameter[0] == "bl": + await message.edit( + '当前黑名单规则:\n' + str(data.get('blacklist', '无')) + '\n如需编辑，请使用 ,pmcaptcha bl +关键词（英文逗号分隔）') + elif message.parameter[0] == 'wel': + await message.edit( + '当前通过时消息规则:\n' + str(data.get('welcome', '无')) + '\n如需编辑，请使用 ,pmcaptcha wel +要发送的消息') + elif message.parameter[0] == 'wait': + await message.edit( + '当前验证等待时间(秒): ' + str(data.get('wait', '无')) + '\n如需编辑，请使用 ,pmcaptcha wait +等待秒数(整数)') + elif message.chat.type != ChatType.PRIVATE: + await message.edit('请在私聊时使用此命令，或添加id参数执行') + await asyncio.sleep(3) + await message.safe_delete() + elif message.parameter[0] == 'add': + await message.edit('已将id ' + cid_ + ' 添加至白名单') + captcha_success.add_id(message.chat.id) + elif message.parameter[0] == 'del': + if captcha_success.del_id(message.chat.id): + await message.edit('已删除id ' + cid_ + ' 的验证记录') + else: + await message.edit('记录不存在') + else: + if message.parameter[0] == 'add': + if message.parameter[1].isnumeric(): + await message.edit('已将id ' + message.parameter[1] + ' 添加至白名单') + captcha_success.add_id(int(message.parameter[1])) + await client.unarchive_chats(chat_ids=int(message.parameter[1])) + else: + await message.edit('参数错误') + elif message.parameter[0] == 'del': + if message.parameter[1].isnumeric(): + if captcha_success.del_id(int(message.parameter[1])): + await message.edit('已删除id ' + message.parameter[1] + ' 的验证记录') + else: + await message.edit('记录不存在') + else: + await message.edit('参数错误') + elif message.parameter[0] == 'wel': + data["welcome"] = " ".join(message.parameter[1:]) + sqlite["pmcaptcha"] = data + await message.edit('规则已更新') + elif message.parameter[0] == 'wait': + if message.parameter[1].isnumeric(): + data["wait"] = int(message.parameter[1]) + sqlite["pmcaptcha"] = data + await message.edit('等待时间已更新') + else: + await message.edit('错误:不是整数') + elif message.parameter[0] == 'bl': + data["blacklist"] = " ".join(message.parameter[1:]) + sqlite["pmcaptcha"] = data + await message.edit('规则已更新') + elif message.parameter[0] == 'check': + if message.parameter[1].isnumeric(): + if captcha_success.check_id(int(message.parameter[1])): + await message.edit('id ' + message.parameter[1] + ' 已验证') + else: + await message.edit('id ' + message.parameter[1] + ' 未验证') + else: + await message.edit('未知用户/无效id')