From 4280ca3112a90f409355d45d304aee1454e2ad6c Mon Sep 17 00:00:00 2001 From: xtaodada Date: Mon, 13 Jun 2022 18:01:12 +0800 Subject: [PATCH] =?UTF-8?q?pmcaptcha=20=E6=94=AF=E6=8C=81=E7=9B=B4?= =?UTF-8?q?=E6=8E=A5=E5=88=A0=E9=99=A4=E5=AF=B9=E8=AF=9D?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- list.json | 2 +- pmcaptcha/main.py | 307 +++++++++++++++++++++++++--------------------- 2 files changed, 170 insertions(+), 139 deletions(-) diff --git a/list.json b/list.json index 7475adc..a6e8020 100644 --- a/list.json +++ b/list.json @@ -72,7 +72,7 @@ }, { "name": "pmcaptcha", - "version": "1.13", + "version": "1.14", "section": "chat", "maintainer": "cloudreflection", "size": "13 kb", diff --git a/pmcaptcha/main.py b/pmcaptcha/main.py index ec772b5..71798eb 100644 --- a/pmcaptcha/main.py +++ b/pmcaptcha/main.py @@ -1,12 +1,14 @@ # pmcaptcha - a pagermaid-pyro plugin by cloudreflection # https://t.me/cloudreflection_channel/268 -# ver 2022/06/11 +# ver 2022/06/13 from pyrogram import Client from pyrogram.enums.chat_type import ChatType from pyrogram.raw.functions.account import UpdateNotifySettings +from pyrogram.raw.functions.messages import DeleteHistory from pyrogram.raw.types import InputNotifyPeer, InputPeerNotifySettings +from pagermaid import log from pagermaid.utils import Message from pagermaid.listener import listener from pagermaid.single_utils import sqlite @@ -16,132 +18,7 @@ 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): - if message.chat.id == 777000 or message.chat.type == ChatType.BOT: - return - cid = message.chat.id - if message.text: - try: - if message.text[0] == ",": #忽略命令 - return - except UnicodeDecodeError: - pass - 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): - # 忽略联系人、服务消息、机器人消息 - if message.from_user.is_contact or message.from_user.id == 777000 or message.chat.type == ChatType.BOT: - return - cid = message.chat.id - data = sqlite.get("pmcaptcha", {}) - if data.get('disable',False) and not captcha_success.check_id(cid): - await message.reply('对方已设置禁止私聊，您已被封禁\n\nYou are not allowed to send private messages to me and been banned') - await client.block_user(user_id=cid) - await client.read_chat_history(cid) - await asyncio.sleep(random.randint(0, 100) / 1000) - return await client.archive_chats(chat_ids=cid) - data['banned'] = data.get('banned',0) + 1 - sqlite['pmcaptcha'] = data - if not captcha_success.check_id(cid) and sqlite.get("pmcaptcha." + str(cid)) is None: - await client.read_chat_history(cid) - if data.get("blacklist", False) and message.text is not None: - 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) - data['banned'] = data.get('banned',0) + 1 - sqlite['pmcaptcha'] = data - 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() - 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) - data['banned'] = data.get('banned',0) + 1 - sqlite['pmcaptcha'] = data - elif sqlite.get("pmcaptcha." + str(cid)): - if message.text == sqlite.get("pmcaptcha." + str(cid)): - await message.safe_delete() - 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 Passed")) - await asyncio.sleep(random.randint(0, 100) / 1000) - await client.unarchive_chats(chat_ids=cid) - data['pass'] = data.get('pass',0) + 1 - sqlite['pmcaptcha'] = data - else: - del sqlite['pmcaptcha.' + str(cid)] - await message.reply('验证错误，您已被封禁\n\nVerification failed.You have been banned.') - await client.block_user(user_id=cid) - await client.read_chat_history(cid) - await asyncio.sleep(random.randint(0, 100) / 1000) - await client.archive_chats(chat_ids=cid) - data['banned'] = data.get('banned',0) + 1 - sqlite['pmcaptcha'] = data - -@listener(is_plugin=True, outgoing=True, command="pmcaptcha", - need_admin=True, - description='一个简单的私聊人机验证 请使用 ```,pmcaptcha h``` 查看可用命令') -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.parameter[0] == 'h': - await message.edit('''```,pmcaptcha``` +pm_captcha_help_msg = '''```,pmcaptcha``` 查询当前私聊用户验证状态 ```,pmcaptcha check id``` @@ -173,20 +50,153 @@ async def pm_captcha(client: Client, message: Message): 查看验证计数器 使用 ```,pmcaptcha stats -clear``` 可重置 +```,pmcaptcha action [ban/delete/archive/none]``` +选择验证失败的处理方式，默认为 archive + 遇到任何问题请先 ```,apt update``` 更新后复现再反馈 -捐赠: cloudreflection.eu.org/donate''') +捐赠: cloudreflection.eu.org/donate''' + + +@listener(is_plugin=False, incoming=False, outgoing=True, ignore_edited=True, privates_only=True) +async def process_pm_captcha_self(_: Client, message: Message): + if message.chat.id == 777000 or message.chat.type == ChatType.BOT: + return + cid = message.chat.id + if message.text: + try: + if message.text[0] == ",": # 忽略命令 + return + except UnicodeDecodeError: + pass + if captcha_success.check_id(cid): + return + else: + return captcha_success.add_id(cid) + + +async def do_action_and_read(client, cid, data): + await client.unarchive_chats(chat_ids=cid) + await asyncio.sleep(random.randint(0, 100) / 1000) + await client.read_chat_history(cid) + await asyncio.sleep(random.randint(0, 100) / 1000) + action = data.get("action", "archive") + if action in ["archive", "delete", "ban"]: + await client.block_user(user_id=cid) + if action in ["archive"]: + await client.archive_chats(chat_ids=cid) + if action in ["delete"]: + try: + await client.invoke(DeleteHistory(max_id=0, peer=await client.resolve_peer(cid))) + except Exception as e: + pass + # log + await log(f"(pmcaptcha) 已对 [{cid}](tg://openmessage?user_id={cid}) 执行 {action} 操作") + data['banned'] = data.get('banned', 0) + 1 + sqlite['pmcaptcha'] = data + + +@listener(is_plugin=False, incoming=True, outgoing=False, ignore_edited=True, privates_only=True) +async def process_pm_captcha(client: Client, message: Message): + # 忽略联系人、服务消息、机器人消息 + if message.from_user.is_contact or message.from_user.id == 777000 or message.chat.type == ChatType.BOT: + return + cid = message.chat.id + data = sqlite.get("pmcaptcha", {}) + if data.get('disable', False) and not captcha_success.check_id(cid): + await message.reply('对方已设置禁止私聊，您已被封禁\n\nYou are not allowed to send private messages to me and been banned') + await do_action_and_read(client, cid, data) + return + if not captcha_success.check_id(cid) and sqlite.get("pmcaptcha." + str(cid)) is None: + await client.read_chat_history(cid) + if data.get("blacklist", False) and message.text is not None: + 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 do_action_and_read(client, cid, data) + 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 do_action_and_read(client, cid, data) + elif sqlite.get("pmcaptcha." + str(cid)): + if message.text == sqlite.get("pmcaptcha." + str(cid)): + await message.safe_delete() + 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) + await message.reply(data.get("welcome", "验证通过\n\nVerification Passed")) + await asyncio.sleep(random.randint(0, 100) / 1000) + await client.unarchive_chats(chat_ids=cid) + data['pass'] = data.get('pass', 0) + 1 + sqlite['pmcaptcha'] = data + else: + del sqlite['pmcaptcha.' + str(cid)] + await message.reply('验证错误，您已被封禁\n\nVerification failed.You have been banned.') + await do_action_and_read(client, cid, data) + + +@listener(is_plugin=True, outgoing=True, command="pmcaptcha", + need_admin=True, + description='一个简单的私聊人机验证 请使用 ```,pmcaptcha h``` 查看可用命令') +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.parameter[0] == 'h': + await message.edit(pm_captcha_help_msg) elif message.parameter[0] == 'disablepm': - if data.get('disable',False): - status='开启' + if data.get('disable', False): + status = '开启' else: - status='关闭' - await message.edit('当前禁止私聊状态: 已'+status+ - '\n如需修改 请使用 ,pmcaptcha disablepm true/false'+ + status = '关闭' + await message.edit('当前禁止私聊状态: 已' + status + + '\n如需修改 请使用 ,pmcaptcha disablepm true/false' + '\n此功能会放行联系人和白名单(已通过验证)用户') elif message.parameter[0] == 'stats': - t = str(data.get('banned',0)+data.get('pass',0)) - await message.edit('自上次重置起，已进行验证 '+str(data.get('pass',0)+data.get('banned',0))+ - ' 次\n其中，通过验证 '+str(data.get('pass',0))+' 次，拦截 '+str(data.get('banned',0))+' 次') + await message.edit('自上次重置起，已进行验证 ' + str(data.get('pass', 0) + data.get('banned', 0)) + + ' 次\n其中，通过验证 ' + str(data.get('pass', 0)) + ' 次，拦截 ' + str(data.get('banned', 0)) + ' 次') elif message.chat.type != ChatType.PRIVATE: await message.edit('请在私聊时使用此命令，或添加id参数执行') await asyncio.sleep(3) @@ -199,6 +209,8 @@ async def pm_captcha(client: Client, message: Message): await message.edit('已删除id ' + cid_ + ' 的验证记录') else: await message.edit('记录不存在') + else: + await message.edit('参数错误') else: if message.parameter[0] == 'add': if message.parameter[1].isnumeric(): @@ -216,7 +228,7 @@ async def pm_captcha(client: Client, message: Message): else: await message.edit('参数错误') elif message.parameter[0] == 'wel': - if message.parameter[1]=='-clear': + if message.parameter[1] == '-clear': if data.get("welcome", False): del data["welcome"] sqlite["pmcaptcha"] = data @@ -233,7 +245,7 @@ async def pm_captcha(client: Client, message: Message): else: await message.edit('错误:不是整数') elif message.parameter[0] == 'bl': - if message.parameter[1]=='-clear': + if message.parameter[1] == '-clear': if data.get("blacklist", False): del data["blacklist"] sqlite["pmcaptcha"] = data @@ -264,3 +276,22 @@ async def pm_captcha(client: Client, message: Message): data["banned"] = 0 sqlite["pmcaptcha"] = data await message.edit('已重置计数器') + elif message.parameter[0] == 'action': + if message.parameter[1] == 'ban': + data["action"] = 'ban' + sqlite["pmcaptcha"] = data + await message.edit('验证失败后将执行**封禁**操作') + elif message.parameter[1] == 'delete': + data["action"] = 'delete' + sqlite["pmcaptcha"] = data + await message.edit('验证失败后将执行**封禁和删除**会话操作') + elif message.parameter[1] == 'archive': + data["action"] = 'archive' + sqlite["pmcaptcha"] = data + await message.edit('验证失败后将执行**封禁和归档**会话操作') + elif message.parameter[1] == 'none': + data["action"] = 'none' + sqlite["pmcaptcha"] = data + await message.edit('验证失败后将不执行任何操作') + else: + await message.edit('参数错误。')