-
Notifications
You must be signed in to change notification settings - Fork 24
/
AddUser-noinput.py
173 lines (163 loc) · 7.98 KB
/
AddUser-noinput.py
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
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
import string
import urllib
from process import *
def write_user(filename, newdata):
with open(filename, 'r', encoding='utf-8') as f:
olddata = f.read()
olddata = json.loads(olddata)
olddata.extend(newdata)
with open(filename, 'w', encoding='utf-8') as write_f:
write_f.write(json.dumps(olddata, indent=2, ensure_ascii=False))
def obtain_coordinates(address):
url = f"https://apis.map.qq.com/jsapi?qt=geoc&addr={urllib.parse.quote(address)}&key={config['api_token']}&output=jsonp&pf=jsapi&ref=jsapi&cb=qq.maps._svcb3.geocoder0"
re = requests.get(url=url).text.strip("qq.maps._svcb3.geocoder0(").strip(")")
re = json.loads(re)
return re['detail']['pointx'] + "@" + re['detail']['pointy']
def checkUserData(filename, enabled, day, name, phone, password, device, modify_coordinates, address, pushmode=None,
Ding_secret=None, Ding_token=None,
PushPlus_token=None, Server_Turbo_token=None,
email_username=None, email_password=None, email_address=None, email_port=None, email_receiver=None):
global content
adddate = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
enddate = (datetime.datetime.now() + datetime.timedelta(days=float(day))).strftime("%Y-%m-%d %H:%M:%S")
deviceID = ''.join(random.sample(string.digits + string.ascii_lowercase, 36))
longitude = obtain_coordinates(address=address).split("@")[0]
latitude = obtain_coordinates(address=address).split("@")[1]
newdata = [
{
"enabled": enabled,
"adddate": adddate,
"enddate": enddate,
"name": name,
"phone": phone,
"password": password,
"deviceId": device,
"dToken": deviceID,
"modify_coordinates": modify_coordinates,
"address": address,
"longitude": longitude,
"latitude": latitude,
"pushmode": pushmode,
"pushdata": {
"Ding": {
"Secret": Ding_secret,
"Token": Ding_token
},
"PushPlus": {
"Token": PushPlus_token
},
"Server_Turbo": {
"Token": Server_Turbo_token
},
"Email": {
"Send": email_username,
"Password": email_password,
"Server_Address": email_address,
"Smtp_Port": email_port,
"Receiver": email_receiver
}
}
}
]
now_localtime = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
TDOA = datetime.datetime.strptime(newdata[0]['enddate'], '%Y-%m-%d %H:%M:%S') - datetime.datetime.strptime(
now_localtime, '%Y-%m-%d %H:%M:%S')
pushFeedback = MessagePush.pushMessage(addinfo=True, title='职校家园打卡添加成功！！',
content=f"姓名：{newdata[0]['name']}\n手机：{newdata[0]['phone']}\n密码：{newdata[0]['password']}\n总计：{TDOA.days}天",
pushdata=newdata[0]['pushdata'], pushmode=newdata[0]['pushmode'])
if "失败" in pushFeedback:
feedback = "写入失败！原因：无法推送至指定的方式！"
elif "成功" in pushFeedback:
feedback = "写入成功！"
write_user(filename=filename, newdata=newdata)
else:
feedback = "写入失败！原因未知！"
return newdata, pushFeedback, feedback
if __name__ == '__main__':
# 用户数据
filename = "all-users.json"
# 是否开启打卡
enabled = True
# 打卡天数
day = ""
# 备注名字（留空为职校家园姓名）
name = ""
# 手机号/职教家园账号
phone = ""
# 密码
password = ""
# 手机型号
device = ""
# 随机定位（经纬度最后一位随机）
modify_coordinates = True
# 打卡地址
address = ""
# 推送方式
pushmode = ""
# 推送方式为1时生效
Ding_secret = ""
Ding_token = ""
# 推送方式为2时生效
PushPlus_token = ""
# 推送方式为4时生效
Server_Turbo_token = ""
# 推送方式为4且config.yml里未配置全局邮箱时生效
email_username = "@"
email_password = ""
email_address = ""
email_port = ""
# 推送方式为4时的接收邮件的邮箱
email_receiver = "@"
'''以下为屎山，无需管'''
if name == "":
account_data = get_account_data(phone, password, device)
uid = account_data[1]
token = account_data[2]
name = get_user_info(uid, device, token)[1]
now_localtime = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
if pushmode == "1":
userdata = checkUserData(filename=filename, enabled=enabled, day=day, name=name, phone=phone, password=password,
device=device, modify_coordinates=modify_coordinates, address=address,
pushmode=pushmode,
Ding_secret=Ding_secret, Ding_token=Ding_token)
pushmode = "钉钉WebHook机器人推送"
elif pushmode == "2":
userdata = checkUserData(filename=filename, enabled=enabled, day=day, name=name, phone=phone, password=password,
device=device, modify_coordinates=modify_coordinates, address=address,
pushmode=pushmode,
PushPlus_token=PushPlus_token)
pushmode = "Pushplus推送"
elif pushmode == "3":
userdata = checkUserData(filename=filename, enabled=enabled, day=day, name=name, phone=phone, password=password,
device=device, modify_coordinates=modify_coordinates, address=address,
pushmode=pushmode,
Server_Turbo_token=Server_Turbo_token)
pushmode = "Server酱"
elif pushmode == "4":
if config['push-data']['Email']['email_password'] == "":
userdata = checkUserData(filename=filename, enabled=enabled, day=day, name=name, phone=phone,
password=password,
device=device, modify_coordinates=modify_coordinates, address=address,
pushmode=pushmode,
email_username=email_username, email_password=email_password,
email_address=email_address,
email_port=email_port, email_receiver=email_receiver)
pushmode = "一对一邮件推送"
else:
userdata = checkUserData(filename=filename, enabled=enabled, day=day, name=name, phone=phone,
password=password,
device=device, modify_coordinates=modify_coordinates, address=address,
pushmode=pushmode,
email_receiver=email_receiver)
pushmode = "一对多邮件推送"
else:
userdata = checkUserData(filename=filename, enabled=enabled, day=day, name=name, phone=phone, password=password,
device=device, modify_coordinates=modify_coordinates, address=address,
pushmode=pushmode)
pushmode = "不推送"
userForm = f'''姓名：\033[32m{userdata[0][0]['name']}\033[0m\n打卡时间：\033[32m{(datetime.datetime.strptime(userdata[0][0]['enddate'], '%Y-%m-%d %H:%M:%S') - datetime.datetime.strptime(now_localtime, '%Y-%m-%d %H:%M:%S')).days}\033[0m天\n手机型号：\033[32m{userdata[0][0]['deviceId']}\033[0m\n打卡地址：\033[32m{userdata[0][0]['address']}\033[0m\n推送方式：\033[32m{pushmode}\033[0m\n推送反馈：\033[33m{userdata[1]}\033[0m'''
print(userForm)
if "成功" in userdata[2]:
print("\033[32m写入成功！\033[0m")
else:
print(f"\033[31m{userdata[2]}\033[0m")