diff options
Diffstat (limited to 'ATRI/plugins/curse.py')
| -rw-r--r-- | ATRI/plugins/curse.py | 55 |
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()) + +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('请求失败') |