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
147
148
149
150
151
|
#!/usr/bin/env python3
# -*- encoding: utf-8 -*-
'''
@File : __init__.py
@Time : 2020/11/06 19:27:00
@Author : Kyomotoi
@Contact : kyomotoiowo@gmail.com
@Github : https://github.com/Kyomotoi
@License : Copyright © 2018-2020 Kyomotoi, All Rights Reserved.
'''
__author__ = 'kyomotoi'
import os
import json
from pathlib import Path
from typing import Optional
from nonebot.rule import Rule
from nonebot.typing import Bot, Event
def check_banlist(for_group: Optional[bool] = False) -> Rule:
'''
检查目标是否存在于封禁名单
:return: bool
'''
async def _chech_banlist(bot: Bot, event: Event, state: dict) -> bool:
# 获取目标信息
user = str(event.user_id)
group = str(event.group_id)
# 名单目录
BAN_LIST_USER_PATH = Path(
'.') / 'ATRI' / 'utils' / 'utils_rule' / 'ban_list_user.json'
BAN_LIST_GROUP_PATH = Path(
'.') / 'ATRI' / 'utils' / 'utils_rule' / 'ban_list_group.json'
# 检查文件是否存在，如不存在，自动创建并写入默认值
if not BAN_LIST_USER_PATH.is_file():
with open(BAN_LIST_USER_PATH, 'w') as f:
f.write(json.dumps({}))
if not BAN_LIST_GROUP_PATH.is_file():
with open(BAN_LIST_GROUP_PATH, 'w') as f:
f.write(json.dumps({}))
# 读取文件
with open(BAN_LIST_USER_PATH, 'r') as f:
data_user = json.load(f)
with open(BAN_LIST_GROUP_PATH, 'r') as f:
data_group = json.load(f)
# 判断目标
if not for_group:
if user:
if user not in data_user:
if group:
if group not in data_group:
return True
else:
return False
else:
return True
else:
return False
elif group:
if group not in data_group:
if user:
if user not in data_user:
return True
else:
return False
else:
return True
else:
return False
else:
return False
else:
if group not in data_group:
return True
else:
return False
return Rule(_chech_banlist)
def check_switch(func_name: str, notice: bool) -> Rule:
'''
检查目标功能是否开启
:return: bool
'''
async def _check_switch(bot: Bot, event: Event, state: dict) -> bool:
# 获取目标信息
group = str(event.group_id)
# 文件目录
SWITCH_ALL_PATH = Path('.') / 'ATRI' / 'utils' / 'utils_rule' / 'switch.json'
SWITCH_ALONE_PATH = Path(
'.') / 'ATRI' / 'data' / 'data_Group' / f'{group}' / 'switch.json'
# 检查文件是否存在，如不存在，自动创建并写入默认值
if not SWITCH_ALL_PATH.is_file():
with open(SWITCH_ALL_PATH, 'ws') as f:
f.write(json.dumps({}))
if not SWITCH_ALONE_PATH.is_file():
try:
os.mkdir(
Path('.') / 'ATRI' / 'data' / 'data_Group' / f'{group}')
except:
pass
with open(SWITCH_ALONE_PATH, 'w') as f:
f.write(json.dumps({}))
# 读取文件
with open(SWITCH_ALL_PATH, 'r') as f:
data_all = json.load(f)
with open(SWITCH_ALONE_PATH, 'r') as f:
data_alone = json.load(f)
# 判断目标是否存在于将要读取的文件，如不存在，写入
# 此项举措是为了适应以后版本更新出现新插件的状况
# 不至于每次都需要修改
if func_name not in data_all:
data_all[func_name] = "True"
with open(SWITCH_ALL_PATH, 'w') as f:
f.write(json.dumps(data_all))
if func_name not in data_alone:
data_alone[func_name] = "True"
with open(SWITCH_ALONE_PATH, 'w') as f:
f.write(json.dumps(data_alone))
# 判断目标
if data_all[func_name] == "True":
if data_alone[func_name] == "True":
return True
else:
return False
else:
if notice:
await bot.send(event, f"Service-{func_name} has been closed.")
return False
return Rule(_check_switch)
|
