diff options
Diffstat (limited to 'ATRI/plugins/wife/__init__.py')
-rw-r--r-- | ATRI/plugins/wife/__init__.py | 116 |
1 files changed, 0 insertions, 116 deletions
diff --git a/ATRI/plugins/wife/__init__.py b/ATRI/plugins/wife/__init__.py deleted file mode 100644 index 2b6f6a3..0000000 --- a/ATRI/plugins/wife/__init__.py +++ /dev/null @@ -1,116 +0,0 @@ -import re -import asyncio -from random import choice - -from nonebot.typing import T_State -from nonebot.adapters.cqhttp import ( - Bot, - MessageEvent, - GroupMessageEvent, - PrivateMessageEvent, -) - -from ATRI.service import Service as sv -from ATRI.rule import is_in_service -from ATRI.utils.limit import is_too_exciting - -from .data_source import Tsuma - - -__doc__ = """ -好欸!是老婆! -权限组:所有人 -用法: - 抽老婆 # 获取一位老婆 - 查老婆 # 查询老婆,如+at对象可查询对方 - 我要离婚 # 离婚... -""" - -roll_wife = sv.on_command(cmd="抽老婆", docs=__doc__, rule=is_in_service("抽老婆")) - - -@roll_wife.handle() -async def _roll_wife(bot: Bot, event: GroupMessageEvent) -> None: - user = event.user_id - gender = event.sender.sex - group = event.group_id - user_name = await bot.get_group_member_info(group_id=group, user_id=user) - user_name = user_name["nickname"] - run = is_too_exciting(user, 1, seconds=5) - if not run: - return - - check_repo, if_h = Tsuma.check_tsuma(str(user)) - if if_h: - await roll_wife.finish(check_repo) - - msg = "5秒后咱将随机抽取一位群友成为\n" f"{user_name} 的老婆!究竟是谁呢~?" - await bot.send(event, msg) - await asyncio.sleep(5) - - async def get_luck_user(): - luck_list = await bot.get_group_member_list(group_id=group) - return choice(luck_list) - - while True: - luck_user = await get_luck_user() - luck_qq = luck_user["user_id"] - if user != luck_qq: - break - - luck_gender = luck_user["sex"] - luck_user = luck_user["nickname"] - d = { - "nickname": user_name, - "gender": gender, - "lassie": {"nickname": luck_user, "qq": luck_qq, "gender": luck_gender}, - } - - if str(luck_qq) == str(event.self_id): - Tsuma.got_tsuma(str(user), d) - msg = "老婆竟是我自己~❤" - else: - msg = Tsuma.got_tsuma(str(user), d) - - await roll_wife.finish(msg) - - -@roll_wife.handle() -async def _no_pr(bot: Bot, event: PrivateMessageEvent) -> None: - await roll_wife.finish("对8起...该功能只对群聊开放(") - - -inquire_wife = sv.on_command(cmd="查老婆", rule=is_in_service("抽老婆")) - - -@inquire_wife.handle() -async def _inq_wife(bot: Bot, event: MessageEvent) -> None: - msg = str(event.message).split(" ") - if msg[0] == "": - user = str(event.user_id) - await inquire_wife.finish(Tsuma.inquire_tsuma(user)) - else: - aim = re.findall(r"qq=(.*?)]", msg[0])[0] - await inquire_wife.finish(Tsuma.inquire_tsuma(aim).replace("你", "ta")) - - -want_divorce = sv.on_command(cmd="我要离婚", rule=is_in_service("抽老婆")) - - -@want_divorce.handle() -async def _want_div(bot: Bot, event: MessageEvent, state: T_State) -> None: - msg = str(event.message).strip() - if msg: - state["is_d"] = msg - - -@want_divorce.got("is_d", prompt="你确定吗?(是/否)") -async def _deal_div(bot: Bot, event: MessageEvent, state: T_State) -> None: - msg = state["is_d"] - user = str(event.user_id) - name = event.sender.nickname - - if msg in ["是", "确定"]: - await want_divorce.finish(Tsuma.divorce(user)) - else: - await want_divorce.finish(f"({name})回心转意了!") |