[Update]

1 files changed, 55 insertions, 0 deletions

diff --git a/ATRI/plugins/curse.py b/ATRI/plugins/curse.py
index e69de29..2fa4986 100644
--- a/ATRI/plugins/curse.py
+++ b/ATRI/plugins/curse.py
@@ -0,0 +1,55 @@
+from random import choice
+from nonebot.plugin import on_command
+from nonebot.adapters.cqhttp import Bot
+
+from ATRI.request import Request
+from ATRI.utils import count_list, del_list_aim
+from ATRI.config import CURSE_CONFIG
+from ATRI.exceptions import InvalidRequest
+from ATRI.service.plugin import Plugin
+from ATRI.rule import is_in_ban_list, is_in_service, is_in_dormant, to_bot
+
+
+
+# ===========================[Begin Command Processing]===========================
+
+
+__plugin_name__ = 'curse'
+__doc__ = """口臭一下"""
+Plugin.register(__plugin_name__, "func", __doc__,
+                        CURSE_CONFIG['curse']['command'])
+
+curse = on_command(CURSE_CONFIG['curse']['command'][0],
+                    aliases=set(CURSE_CONFIG['curse']['command']),
+                    rule=is_in_ban_list() & is_in_dormant()
+                    & is_in_service(__plugin_name__)
+                    & to_bot())
+
[email protected]()
+async def _(bot: Bot, event) -> None:
+    await bot.send(
+        event, await Function().curse(str(event.get_user_id())))
+
+
+# ===========================[End Command Processing]=============================
+
+curse_list = []
+
+
+class Function:
+    async def curse(self, user: str):
+        global curse_list
+        
+        if count_list(curse_list, user) == 3:
+            curse_list.append(user)
+            return choice(CURSE_CONFIG['curse']['times'][3]['repo'])
+        elif count_list(curse_list, user) == 6:
+            curse_list = del_list_aim(curse_list, user)
+            return choice(CURSE_CONFIG['curse']['times'][6]['repo'])
+        else:
+            try:
+                curse_list.append(user)
+                return str(await Request.get_text(
+                    url=CURSE_CONFIG['curse']['url']))
+            except InvalidRequest:
+                raise InvalidRequest('请求失败')