keyword 插件的 newkeyword 函数 (#120)

keyword 插件的 newkeyword 函数

keyword_func/list.json

@@ -1 +1 @@
-{"list":["ranstr"]}
+{"list": ["ranstr", "newkeyword"]}

keyword_func/newkeyword.py

@@ -0,0 +1,91 @@
+ import json
+from os import remove, mkdir
+from os.path import isfile, exists
+from telethon.tl.types import ChannelParticipantsAdmins
+
+extra_path = "plugins/keyword_func/extra"
+
+
+def get_data(filename):
+    filepath = f"{extra_path}/{filename}"
+    if exists(filepath):
+        with open(filepath, "r") as f:
+            data = f.read()
+        return data
+    else:
+        return ""
+
+
+def write_data(filename, data):
+    filepath = f"{extra_path}/{filename}"
+    with open(filepath, "w") as f:
+        f.write(data)
+
+
+def init_file(filename):
+    if exists(extra_path):
+        if isfile(extra_path):
+            remove(extra_path)
+            mkdir(extra_path)
+    else:
+        mkdir(extra_path)
+    if not exists(f"{extra_path}/{filename}"):
+        with open(f"{extra_path}/{filename}", "w") as f:
+            f.write("{}")
+
+
+async def main(context):
+    chat_id = context.chat_id
+    if chat_id < 0:
+        admins = await context.client.get_participants(context.chat, filter=ChannelParticipantsAdmins)
+        text = context.text
+        if not text:
+            text = ""
+        if text.split()[0] == "/add":
+            if context.sender in admins:
+                try:
+                    parse = text.split("\n")
+                    parse[0] = " ".join(parse[0].split()[1:])
+                    init_file(f"newkeyword_{chat_id}.json")
+                    data = json.loads(get_data(f"newkeyword_{chat_id}.json"))
+                    data[parse[0]] = parse[1]
+                    write_data(f"newkeyword_{chat_id}.json", json.dumps(data))
+                    await context.client.send_message(chat_id, "设置成功", reply_to=context.id)
+                except:
+                    await context.client.send_message(chat_id, "设置失败", reply_to=context.id)
+            else:
+                await context.client.send_message(chat_id, "您无权进行此操作", reply_to=context.id)
+        elif text.split()[0] == "/del":
+            if context.sender in admins:
+                try:
+                    init_file(f"newkeyword_{chat_id}.json")
+                    data = json.loads(get_data(f"newkeyword_{chat_id}.json"))
+                    del data[" ".join(text.split(" ")[1:])]
+                    write_data(f"newkeyword_{chat_id}.json", json.dumps(data))
+                    await context.client.send_message(chat_id, "删除成功", reply_to=context.id)
+                except:
+                    await context.client.send_message(chat_id, "删除失败", reply_to=context.id)
+            else:
+                await context.client.send_message(chat_id, "您无权进行此操作", reply_to=context.id)
+        elif text.split()[0] == "/list":
+            if context.sender in admins:
+                try:
+                    init_file(f"newkeyword_{chat_id}.json")
+                    data = json.loads(get_data(f"newkeyword_{chat_id}.json"))
+                    message = ""
+                    count = 1
+                    for k, v in data.items():
+                        message += f"`{count}` : `{k}` -> `{v}`\n"
+                        count += 1
+                    await context.client.send_message(context.sender_id, message)
+                    await context.client.send_message(chat_id, "已发送私聊", reply_to=context.id)
+                except:
+                    await context.client.send_message(chat_id, "获取失败", reply_to=context.id)
+            else:
+                await context.client.send_message(chat_id, "您无权进行此操作", reply_to=context.id)
+        else:
+            data = json.loads(get_data(f"newkeyword_{chat_id}.json"))
+            for k, v in data.items():
+                if k in text:
+                    await context.client.send_message(chat_id, v, reply_to=context.id)
+    return ""