✨🐛 更新

新增：老婆！
新增：涩图

修复：manage中出现的bug

优化：nsfw不再用float，换为int

1 files changed, 129 insertions, 0 deletions

diff --git a/ATRI/plugins/wife/__init__.py b/ATRI/plugins/wife/__init__.py
new file mode 100644
index 0000000..3f89214
--- /dev/null
+++ b/ATRI/plugins/wife/__init__.py
@@ -0,0 +1,129 @@
+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})回心转意了！')