diff options
Diffstat (limited to 'ATRI/plugins/wife')
| -rw-r--r-- | ATRI/plugins/wife/__init__.py | 116 | ||||
| -rw-r--r-- | ATRI/plugins/wife/data_source.py | 87 |
2 files changed, 0 insertions, 203 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})回心转意了！") diff --git a/ATRI/plugins/wife/data_source.py b/ATRI/plugins/wife/data_source.py deleted file mode 100644 index 9665351..0000000 --- a/ATRI/plugins/wife/data_source.py +++ /dev/null @@ -1,87 +0,0 @@ -import os -import json -from pathlib import Path - - -WIFE_DIR = Path(".") / "ATRI" / "data" / "database" / "wife" -MERRY_LIST_PATH = WIFE_DIR / "merry_list.json" -os.makedirs(WIFE_DIR, exist_ok=True) - - -class Tsuma: - @staticmethod - def _load_tsuma() -> dict: - try: - return json.loads(MERRY_LIST_PATH.read_bytes()) - except FileNotFoundError: - with open(MERRY_LIST_PATH, "w") as r: - r.write(json.dumps({}, indent=4)) - return dict() - - @staticmethod - def _store_tsuma(data: dict) -> None: - with open(MERRY_LIST_PATH, "w") as r: - r.write(json.dumps(data, indent=4)) - - @classmethod - def check_tsuma(cls, user: str): - data = cls._load_tsuma() - if user in data: - msg = "阿，你已经有老婆惹！" f"ta是：{data[user]['lassie']['nickname']}" - return msg, True - else: - return "悲——你还没老婆...", False - - @classmethod - def inquire_tsuma(cls, user: str) -> str: - data = cls._load_tsuma() - if user in data: - return f"你的老婆是：{data[user]['lassie']['nickname']} 哦~❤" - else: - return "悲——你还没老婆..." - - @classmethod - def got_tsuma(cls, user: str, d: dict) -> str: - check_repo, if_h = cls.check_tsuma(user) # 防止出现多人同时操作导致 NTR 事件 - if if_h: - return check_repo - else: - data = cls._load_tsuma() - data[user] = { - "nickname": d["nickname"], - "gender": d["gender"], - "lassie": { - "nickname": d["lassie"]["nickname"], - "qq": d["lassie"]["qq"], - "gender": d["lassie"]["gender"], - }, - } - cls._store_tsuma(data) - - data[d["lassie"]["qq"]] = { - "nickname": d["lassie"]["nickname"], - "gender": d["lassie"]["gender"], - "lassie": { - "nickname": d["nickname"], - "qq": user, - "gender": d["gender"], - }, - } - cls._store_tsuma(data) - - msg = ( - f"> {d['lassie']['nickname']}({d['lassie']['qq']})\n" - f"恭喜成为 {d['nickname']} 的老婆~⭐" - ) - return msg - - @classmethod - def divorce(cls, user: str) -> str: - data = cls._load_tsuma() - if not user in data: - return "悲——你还没老婆。。" - - msg = f"悲——，({data[user]['nickname']})抛弃了({data[user]['lassie']['nickname']})" - del data[user] - cls._store_tsuma(data) - return msg |