TeamPGM/PagerMaid_Plugins
3
0
Fork 0
mirror of https://github.com/TeamPGM/PagerMaid_Plugins.git synced 2024-11-25 15:22:20 +00:00
Code Issues Projects Releases Wiki Activity

keyword 添加消息自动删除功能 (#94)

Browse Source 
* 添加消息自动删除功能

* Update list.json
This commit is contained in:
c3b2a 2021-01-17 09:42:11 +08:00 committed by GitHub
parent 87173dac8c
commit e4d280e28f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 43 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

47
keyword.py
View File

@ -1,4 +1,5 @@
import re, time, uuid import re, time, asyncio
from uuid import uuid4
from base64 import b64encode, b64decode from base64 import b64encode, b64decode
from pagermaid import bot, redis, log, redis_status from pagermaid import bot, redis, log, redis_status
from pagermaid.listener import listener from pagermaid.listener import listener
@ -20,7 +21,7 @@ def decode(s: str):
return str(b64decode(s.encode('utf-8')), 'utf-8') return str(b64decode(s.encode('utf-8')), 'utf-8')
def random_str(): def random_str():
return str(uuid.uuid4()).replace('-', '') return str(uuid4()).replace('-', '')
def parse_rules(rules: str): def parse_rules(rules: str):
n_rules = {} n_rules = {}
@ -56,16 +57,22 @@ def get_redis(db_key: str):
byte_data = str(byte_data, "ascii") byte_data = str(byte_data, "ascii")
return parse_rules(byte_data) return parse_rules(byte_data)
async def del_msg(context, t_lim):
await asyncio.sleep(t_lim)
await context.delete()
@listener(is_plugin=True, outgoing=True, command="keyword", @listener(is_plugin=True, outgoing=True, command="keyword",
description="关键词自动回复。", description="关键词自动回复",
parameters="``new <plain|regex> '<规则>' '<回复信息>'` 或者 `del <plain|regex> '<规则>'` 或者 `list` 或者 `clear <plain|regex>") parameters="``new <plain|regex> '<规则>' '<回复信息>'` 或者 `del <plain|regex> '<规则>'` 或者 `list` 或者 `clear <plain|regex>")
async def reply(context): async def reply(context):
if not redis_status(): if not redis_status():
await context.edit("出错了呜呜呜 ~ Redis 离线，无法运行。") await context.edit("出错了呜呜呜 ~ Redis 离线，无法运行")
await del_msg(context, 5)
return return
chat_id = context.chat_id chat_id = context.chat_id
if chat_id > 0: if chat_id > 0:
await context.edit("请在群组中使用") await context.edit("请在群组中使用")
await del_msg(context, 5)
return return
plain_dict = get_redis(f"keyword.{chat_id}.plain") plain_dict = get_redis(f"keyword.{chat_id}.plain")
regex_dict = get_redis(f"keyword.{chat_id}.regex") regex_dict = get_redis(f"keyword.{chat_id}.regex")
@ -80,6 +87,7 @@ async def reply(context):
parse.append(tmp_parse[i]) parse.append(tmp_parse[i])
if len(parse) == 0 or (len(parse[0].split()) == 1 and parse[0].split()[0] in ("new", "del", "clear")) or len(parse[0].split()) > 2: if len(parse) == 0 or (len(parse[0].split()) == 1 and parse[0].split()[0] in ("new", "del", "clear")) or len(parse[0].split()) > 2:
await context.edit("[Code: -1] 格式错误，格式为 `-keyword` 加上 `new <plain|regex> '<规则>' '<回复信息>'` 或者 `del <plain|regex> '<规则>'` 或者 `list` 或者 `clear <plain|regex>`") await context.edit("[Code: -1] 格式错误，格式为 `-keyword` 加上 `new <plain|regex> '<规则>' '<回复信息>'` 或者 `del <plain|regex> '<规则>'` 或者 `list` 或者 `clear <plain|regex>`")
await del_msg(context, 10)
return return
else: parse[0] = parse[0].split() else: parse[0] = parse[0].split()
if parse[0][0] == "new" and len(parse) == 3: if parse[0][0] == "new" and len(parse) == 3:
@ -91,8 +99,10 @@ async def reply(context):
redis.set(f"keyword.{chat_id}.regex", save_rules(regex_dict, placeholder)) redis.set(f"keyword.{chat_id}.regex", save_rules(regex_dict, placeholder))
else: else:
await context.edit("格式错误，格式为 `-keyword` 加上 `new <plain|regex> '<规则>' '<回复信息>'` 或者 `del <plain|regex> '<规则>'` 或者 `list` 或者 `clear <plain|regex>`") await context.edit("格式错误，格式为 `-keyword` 加上 `new <plain|regex> '<规则>' '<回复信息>'` 或者 `del <plain|regex> '<规则>'` 或者 `list` 或者 `clear <plain|regex>`")
await del_msg(context, 10)
return return
await context.edit("设置成功") await context.edit("设置成功")
await del_msg(context, 5)
elif parse[0][0] == "del" and len(parse) == 2: elif parse[0][0] == "del" and len(parse) == 2:
if parse[0][1] == "plain": if parse[0][1] == "plain":
if parse[1] in plain_dict: if parse[1] in plain_dict:
@ -100,6 +110,7 @@ async def reply(context):
redis.set(f"keyword.{chat_id}.plain", save_rules(plain_dict, placeholder)) redis.set(f"keyword.{chat_id}.plain", save_rules(plain_dict, placeholder))
else: else:
await context.edit("规则不存在") await context.edit("规则不存在")
await del_msg(context, 5)
return return
elif parse[0][1] == "regex": elif parse[0][1] == "regex":
if parse[1] in regex_dict: if parse[1] in regex_dict:
@ -107,11 +118,14 @@ async def reply(context):
redis.set(f"keyword.{chat_id}.regex", save_rules(regex_dict, placeholder)) redis.set(f"keyword.{chat_id}.regex", save_rules(regex_dict, placeholder))
else: else:
await context.edit("规则不存在") await context.edit("规则不存在")
await del_msg(context, 5)
return return
else: else:
await context.edit("格式错误，格式为 -keyword 加上 new <plain|regex> '<规则>' '<回复信息>' 或者 del <plain|regex> '<规则>' 或者 list 或者 clear <plain|regex>") await context.edit("格式错误，格式为 -keyword 加上 new <plain|regex> '<规则>' '<回复信息>' 或者 del <plain|regex> '<规则>' 或者 list 或者 clear <plain|regex>")
await del_msg(context, 10)
return return
await context.edit("删除成功") await context.edit("删除成功")
await del_msg(context, 5)
elif parse[0][0] == "list" and len(parse) == 1: elif parse[0][0] == "list" and len(parse) == 1:
plain_msg = "Plain: \n" plain_msg = "Plain: \n"
for k, v in plain_dict.items(): for k, v in plain_dict.items():
@ -127,22 +141,27 @@ async def reply(context):
redis.set(f"keyword.{chat_id}.regex", "") redis.set(f"keyword.{chat_id}.regex", "")
else: else:
await context.edit("参数错误") await context.edit("参数错误")
await del_msg(context, 5)
return return
await context.edit("清除成功") await context.edit("清除成功")
await del_msg(context, 5)
else: else:
await context.edit("[Code -2] 格式错误，格式为 `-keyword` 加上 `new <plain|regex> '<规则>' '<回复信息>'` 或者 `del <plain|regex> '<规则>'` 或者 `list` 或者 `clear <plain|regex>`") await context.edit("[Code -2] 格式错误，格式为 `-keyword` 加上 `new <plain|regex> '<规则>' '<回复信息>'` 或者 `del <plain|regex> '<规则>'` 或者 `list` 或者 `clear <plain|regex>`")
await del_msg(context, 10)
return return
@listener(outgoing=True, command="replyset", @listener(outgoing=True, command="replyset",
description="自动回复设置。", description="自动回复设置",
parameters="help") parameters="help")
async def reply_set(context): async def reply_set(context):
if not redis_status(): if not redis_status():
await context.edit("出错了呜呜呜 ~ Redis 离线，无法运行。") await context.edit("出错了呜呜呜 ~ Redis 离线，无法运行")
await del_msg(context, 5)
return return
chat_id = context.chat_id chat_id = context.chat_id
if chat_id > 0: if chat_id > 0:
await context.edit("请在群组中使用") await context.edit("请在群组中使用")
await del_msg(context, 5)
return return
params = context.parameter params = context.parameter
is_global = len(params) >= 1 and params[0] == "global" is_global = len(params) >= 1 and params[0] == "global"
@ -154,6 +173,7 @@ async def reply_set(context):
cmd_dict = {"help": (1, ), "mode": (2, ), "list": (2, 3), "show": (1, ), "clear": (1, )} cmd_dict = {"help": (1, ), "mode": (2, ), "list": (2, 3), "show": (1, ), "clear": (1, )}
if len(params) < 1: if len(params) < 1:
await context.edit("参数错误") await context.edit("参数错误")
await del_msg(context, 5)
return return
if params[0] in cmd_list and len(params) in cmd_dict[params[0]]: if params[0] in cmd_list and len(params) in cmd_dict[params[0]]:
if params[0] == "help": if params[0] == "help":
@ -163,6 +183,7 @@ async def reply_set(context):
`-replyset mode <0/1/clear>` ( 0 表示黑名单，1 表示白名单 ) 或 `-replyset mode <0/1/clear>` ( 0 表示黑名单，1 表示白名单 ) 或
`-replyset list <add/del/show/clear> [user_id]`。 `-replyset list <add/del/show/clear> [user_id]`。
在 `-replyset` 后面加上 `global` 即为全局设置''') 在 `-replyset` 后面加上 `global` 即为全局设置''')
await del_msg(context, 15)
return return
elif params[0] == "show": elif params[0] == "show":
defaults = {"mode": "0", "list": "无"} defaults = {"mode": "0", "list": "无"}
@ -177,14 +198,17 @@ async def reply_set(context):
redis.set(redis_data, save_rules(settings_dict, None)) redis.set(redis_data, save_rules(settings_dict, None))
if params[1] == "0": await context.edit("模式已更改为黑名单") if params[1] == "0": await context.edit("模式已更改为黑名单")
elif params[1] == "1": await context.edit("模式已更改为白名单") elif params[1] == "1": await context.edit("模式已更改为白名单")
await del_msg(context, 5)
return return
elif params[1] == "clear": elif params[1] == "clear":
if "mode" in settings_dict: del settings_dict["mode"] if "mode" in settings_dict: del settings_dict["mode"]
redis.set(redis_data, save_rules(settings_dict, None)) redis.set(redis_data, save_rules(settings_dict, None))
await context.edit("清除成功") await context.edit("清除成功")
await del_msg(context, 5)
return return
else: else:
await context.edit("参数错误") await context.edit("参数错误")
await del_msg(context, 5)
return return
elif params[0] == "list": elif params[0] == "list":
if params[1] == "show" and len(params) == 2: if params[1] == "show" and len(params) == 2:
@ -197,6 +221,7 @@ async def reply_set(context):
return return
else: else:
await context.edit("列表为空") await context.edit("列表为空")
await del_msg(context, 5)
return return
elif params[1] == "add" and len(params) == 3: elif params[1] == "add" and len(params) == 3:
if is_num(params[2]): if is_num(params[2]):
@ -205,9 +230,11 @@ async def reply_set(context):
else: settings_dict["list"] += f",{params[2]}" else: settings_dict["list"] += f",{params[2]}"
redis.set(redis_data, save_rules(settings_dict, None)) redis.set(redis_data, save_rules(settings_dict, None))
await context.edit("添加成功") await context.edit("添加成功")
await del_msg(context, 5)
return return
else: else:
await context.edit("user_id 需为整数") await context.edit("user_id 需为整数")
await del_msg(context, 5)
return return
elif params[1] == "del" and len(params) == 3: elif params[1] == "del" and len(params) == 3:
if is_num(params[2]): if is_num(params[2]):
@ -219,30 +246,38 @@ async def reply_set(context):
settings_dict["list"] = ",".join(user_list) settings_dict["list"] = ",".join(user_list)
redis.set(redis_data, save_rules(settings_dict, None)) redis.set(redis_data, save_rules(settings_dict, None))
await context.edit("删除成功") await context.edit("删除成功")
await del_msg(context, 5)
return return
else: else:
await context.edit("user_id 不在列表") await context.edit("user_id 不在列表")
await del_msg(context, 5)
return return
else: else:
await context.edit("列表为空") await context.edit("列表为空")
await del_msg(context, 5)
return return
else: else:
await context.edit("user_id 需为整数") await context.edit("user_id 需为整数")
await del_msg(context, 5)
return return
elif params[1] == "clear" and len(params) == 2: elif params[1] == "clear" and len(params) == 2:
if "list" in settings_dict: del settings_dict["list"] if "list" in settings_dict: del settings_dict["list"]
redis.set(redis_data, save_rules(settings_dict, None)) redis.set(redis_data, save_rules(settings_dict, None))
await context.edit("清除成功") await context.edit("清除成功")
await del_msg(context, 5)
return return
else: else:
await context.edit("参数错误") await context.edit("参数错误")
await del_msg(context, 5)
return return
elif params[0] == "clear": elif params[0] == "clear":
redis.delete(redis_data) redis.delete(redis_data)
await context.edit("清除成功") await context.edit("清除成功")
await del_msg(context, 5)
return return
else: else:
await context.edit("参数错误") await context.edit("参数错误")
await del_msg(context, 5)
return return

4
list.json
View File

@ -362,10 +362,10 @@
}, },
{ {
"name": "keyword", "name": "keyword",
"version": "1.00", "version": "1.1",
"section": "chat", "section": "chat",
"maintainer": "c3b2a", "maintainer": "c3b2a",
"size": "11.9 kb", "size": "12.9 kb",
"supported": true, "supported": true,
"des-short": "群组关键词自动回复插件", "des-short": "群组关键词自动回复插件",
"des": "keyword, replyset。" "des": "keyword, replyset。"