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
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
|
import json
from pathlib import Path
from datetime import datetime
from nonebot.permission import SUPERUSER
from nonebot.plugin import on_command, on_message
from nonebot.adapters.cqhttp import (
Bot,
MessageEvent,
GroupMessageEvent
)
from ATRI.exceptions import WriteError
from ATRI.log import logger
ADMIN_DIR = Path('.') / 'ATRI' / 'data' / 'database' / 'admin'
# 收集bot所在的群聊聊天记录
chat_monitor = on_message()
@chat_monitor.handle()
async def _chat_monitor(bot: Bot, event: GroupMessageEvent) -> None:
now_time = datetime.now().strftime('%Y-%m-%d')
file_name = f"{now_time}-chat.json"
path = ADMIN_DIR / f"{event.group_id}" / file_name
path.parent.mkdir(exist_ok=True, parents=True)
path.parent.touch(exist_ok=True)
try:
data = json.loads(path.read_bytes())
except FileExistsError:
data = {}
data[event.message_id] = {
"post_type": event.post_type,
"sub_type": event.sub_type,
"user_id": event.user_id,
"group_id": event.group_id,
"message_type": event.message_type,
"message": event.message,
"raw_message": event.raw_message,
"font": event.font,
"sender": event.sender,
"to_me": event.to_me
}
try:
with open(path, 'w', encoding='utf-8') as r:
r.write(
json.dumps(
data, indent=4
)
)
logger.debug(f"写入消息成功，id: {event.message_id}")
except WriteError:
logger.error("消息记录失败，可能是缺少文件的原因！")
ESSENTIAL_DIR = Path('.') / 'ATRI' / 'data' / 'database' / 'essential'
request_friend = on_command(
"好友申请",
permission=SUPERUSER
)
@request_friend.handle()
async def _request_friend(bot: Bot, event: MessageEvent) -> None:
msg = str(event.message).split(" ")
key = msg[0]
data = {}
path = ESSENTIAL_DIR / "request_friend.json"
try:
data = json.loads(path.read_bytes())
except FileExistsError:
await request_friend.finish("读取数据失败，可能并没有请求...")
if key == "list":
msg0 = ""
for i in data.keys():
msg0 += f"{i} | {data[i]['user_id']} | {data[i]['comment']}\n"
msg = "好友申请列表如下：\n"
msg += msg0
await request_friend.finish(msg)
elif key == "y":
arg = msg[1]
await bot.set_friend_add_request(flag=arg, approve=True)
await request_friend.finish(f"完成~！已同意 {data[arg]['user_id']} 的申请")
elif key == "n":
arg = msg[1]
await bot.set_friend_add_request(flag=arg, approve=False)
await request_friend.finish(f"完成~！已拒绝 {data[arg]['user_id']} 的申请")
else:
await request_friend.finish("阿...请检查输入——！")
request_group = on_command(
"群聊申请",
permission=SUPERUSER
)
@request_group.handle()
async def _request_group(bot: Bot, event: MessageEvent) -> None:
msg = str(event.message).split(" ")
key = msg[0]
data = {}
path = ESSENTIAL_DIR / "request_group.json"
try:
data = json.loads(path.read_bytes())
except FileExistsError:
await request_friend.finish("读取数据失败，可能并没有请求...")
if key == "list":
msg0 = ""
for i in data.keys():
msg0 += f"{i} | {data[i]['sub_type']} | {data[i]['user_id']} | {data[i]['comment']}\n"
msg = "群申请列表如下：\n"
msg += msg0
await request_friend.finish(msg)
elif key == "y":
arg = msg[1]
try:
await bot.set_group_add_request(flag=arg,
sub_type=data[arg]['sub_type'],
approve=False)
await request_friend.finish(f"完成~！已同意 {data[arg]['user_id']} 的申请")
except:
await request_friend.finish("请检查输入的值是否正确——！")
elif key == "n":
arg = msg[1]
try:
await bot.set_group_add_request(flag=arg,
sub_type=data[arg]['sub_type'],
approve=False)
await request_friend.finish(f"完成~！已拒绝 {data[arg]['user_id']} 的申请")
except:
await request_friend.finish("请检查输入的值是否正确——！")
else:
await request_friend.finish("阿...请检查输入——！")
|
