blob: 79ed7ad72d3ed2ff26560e6868a4eacb413a3e73 (
plain)
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
|
import json
from pathlib import Path
from typing import Optional
SERVICE_SWITCH_PATH = Path('.') / 'ATRI' / 'data' / 'service' / 'switch.service.json'
SERVICE_BANLIST_PATH = Path('.') / 'ATRI' / 'data' / 'service' / 'banlist.service.json'
state = 0
class Service():
class Switch():
if not SERVICE_SWITCH_PATH.is_file:
SERVICE_SWITCH_PATH.write_text(json.dumps({}))
data = {}
else:
data = json.loads(SERVICE_SWITCH_PATH.read_bytes())
def get_service(self) -> dict:
return self.data
def auth_service(self, service: str, group: Optional[int]) -> bool:
try:
self.data['global']
except:
self.data['global'] = {}
SERVICE_SWITCH_PATH.write_text(json.dumps(self.data))
try:
self.data[group]
except:
self.data[group] = {}
SERVICE_SWITCH_PATH.write_text(json.dumps(self.data))
if (not self.data['global'].get('service', None)
or not self.data[group][service].get('service', None)):
self.data['global'][service] = True
self.data[group][service] = True
SERVICE_SWITCH_PATH.write_text(json.dumps(self.data))
else:
pass
if self.data['global'][service]:
return True if self.data[group][service] else False
else:
return False
def control_service(self, service: str, _type: bool, group: Optional[int]) -> bool:
if service not in self.data:
self.data['global'][service] = True
self.data[group][service] = True
SERVICE_SWITCH_PATH.write_text(json.dumps(self.data))
if group:
try:
self.data[group][service] = _type
SERVICE_SWITCH_PATH.write_text(json.dumps(self.data))
return True
except:
return False
else:
try:
self.data['global'][service] = _type
SERVICE_SWITCH_PATH.write_text(json.dumps(self.data))
return True
except:
return False
class BanList():
if not SERVICE_BANLIST_PATH.is_file():
SERVICE_BANLIST_PATH.write_text(json.dumps({}))
data = {}
else:
data = json.loads(SERVICE_BANLIST_PATH.read_bytes())
def get_banlist(self) -> dict:
return self.data
def is_in_list(self, user: Optional[int]) -> bool:
return False if user in self.data else True
def add_list(self, user: Optional[int]) -> bool:
try:
self.data[user] = user
SERVICE_BANLIST_PATH.write_text(json.dumps(self.data))
return True
except:
return False
def del_list(self, user: Optional[int]) -> bool:
try:
del self.data[user]
SERVICE_BANLIST_PATH.write_text(json.dumps(self.data))
return True
except:
return False
class Dormant():
def is_sleep(self) -> bool:
return True if state != 1 else False
def cont_wake(self, _type: bool) -> bool:
global state
try:
if _type:
state = 0
else:
state = 1
return True
except:
return False
|
