1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
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})回心转意了！')
|
