forked from gphwxhq/yiban
-
Notifications
You must be signed in to change notification settings - Fork 0
/
work.py
283 lines (271 loc) · 14.1 KB
/
work.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
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
# coding=utf-8
import re,random
from json.decoder import JSONDecodeError
import requests, json, time
from Crypto.PublicKey import RSA
from Crypto.Cipher import PKCS1_v1_5
import base64
from urllib import parse
from requests.exceptions import SSLError
from logger import logger
class yiban:
def send(self, text,detail=""):
#server酱发送模块
if self.server_url=="":
return
data = {
'text': text,
'desp': detail
}
res=json.loads(requests.post(self.server_url, data=data).text)
if(res['errno']!=0):
logger.error("server酱报错：%s"%res['errmsg'])
def encrypt_passwd(self, pwd):
#密码加密
PUBLIC_KEY = '''-----BEGIN PUBLIC KEY-----
MIICIjANBgkqhkiG9w0BAQEFAAOCAg8AMIICCgKCAgEAxbzZk3gEsbSe7A95iCIk
59Kvhs1fHKE6zRUfOUyTaKd6Rzgh9TB/jAK2ZN7rHzZExMM9kw6lVwmlV0VabSmO
YL9OOHDCiEjOlsfinlZpMZ4VHg8gkFoOeO4GvaBs7+YjG51Am6DKuJWMG9l1pAge
96Uhx8xWuDQweIkjWFADcGLpDJJtjTcrh4fy8toE0/0zJMmg8S4RM/ub0q59+VhM
zBYAfPmCr6YnEZf0QervDcjItr5pTNlkLK9E09HdKI4ynHy7D9lgLTeVmrITdq++
mCbgsF/z5Rgzpa/tCgIQTFD+EPPB4oXlaOg6yFceu0XUQEaU0DvAlJ7Zn+VwPkkq
JEoGudklNePHcK+eLRLHcjd9MPgU6NP31dEi/QSCA7lbcU91F3gyoBpSsp5m7bf5
//OBadjWJDvl2KML7NMQZUr7YXqUQW9AvoNFrH4edn8d5jY5WAxWsCPQlOqNdybM
vKF2jhjIE1fTWOzK+AvvFyNhxer5bWGU4S5LTr7QNXnvbngXCdkQfrcSn/ydQXP0
vXfjf3NhpluFXqWe5qUFKXvjY6+PdrE/lvTmX4DdvUIu9NDa2JU9mhwAPPR1yjjp
4IhgYOTQL69ZQcvy0Ssa6S25Xi3xx2XXbdx8svYcQfHDBF1daK9vca+YRX/DzXxl
1S4uGt+FUWSwuFdZ122ZCZ0CAwEAAQ==
-----END PUBLIC KEY-----
'''
cipher = PKCS1_v1_5.new(RSA.importKey(PUBLIC_KEY))
cipher_text = base64.b64encode(cipher.encrypt(bytes(pwd, encoding="utf8")))
return parse.quote(cipher_text.decode("utf-8"))
def login(self):
params = {
"account": self.account,
"ct": 2,
"identify": 0,
"v": "4.7.12",
"passwd": self.encrypt_passwd(self.passwd)
}
HEADERS = {"Origin": "https://c.uyiban.com", "User-Agent": "yiban"}
try:
data = self.sess.get("https://mobile02.yiban.cn/api/v2/passport/login", params=params, headers=HEADERS).text
except SSLError:
#解决 (Caused by SSLError(SSLError(1, '[SSL: DH_KEY_TOO_SMALL] dh key too small (_ssl.c:897)'),))
requests.packages.urllib3.disable_warnings()
requests.packages.urllib3.util.ssl_.DEFAULT_CIPHERS += 'HIGH:!DH:!aNULL'
try:
requests.packages.urllib3.contrib.pyopenssl.DEFAULT_SSL_CIPHER_LIST += 'HIGH:!DH:!aNULL'
except AttributeError:
# no pyopenssl support used / needed / available
pass
data = self.sess.get("https://mobile.yiban.cn/api/v2/passport/login", params=params, headers=HEADERS,
verify=False).text
data=json.loads(data)
if data is not None and str(data["response"]) == "100":
self.access_token = data["data"]["access_token"]
self.name = data["data"]["user"]["name"]
logger.info("登录成功，用户：%s"%self.name)
return True
else:
return False
def auth(self):
#登录验证
HEADERS = {"logintoken":self.access_token,"Host": "f.yiban.cn", "User-Agent": "yiban"}
# location = self.sess.get('http://f.yiban.cn/iapp/index?act=iapp7463&v=%s' % self.access_token, headers=HEADERS,
# allow_redirects=False)
au=self.sess.get('https://f.yiban.cn/iapp7463?access_token=%s&v_time=%s'%(self.access_token,round(time.time())), headers=HEADERS,allow_redirects=False)
location = self.sess.get('http://f.yiban.cn/iapp/index?act=iapp7463', headers=HEADERS,
allow_redirects=False)
if "Location" in location.headers.keys():
location=location.headers["Location"]
verifyRequest = re.findall(r"verify_request=(.*?)&", location)[0]
else:
verifyRequest=''
# logger.info(verifyRequest)
t=self.sess.get("https://api.uyiban.com/base/c/auth/yiban?verifyRequest=%s&CSRF=%s" % (verifyRequest, self.csrf),
cookies=self.cookie, headers=self.headers)
def get_tasklist(self):
cur = time.time()
starttime = time.strftime("%Y-%m-%d", time.localtime(cur - 86400*7)) # 往前推7天
# starttime 形式='2020-10-01'
endtime = time.strftime("%Y-%m-%d", time.localtime(cur + 86400)) # 往后推一天，解决服务器时区问题
url = 'https://api.uyiban.com/officeTask/client/index/uncompletedList?StartTime={}%2000%3A00&EndTime={}%2023%3A59&CSRF={}'.format(
starttime, endtime, self.csrf)
for i in range(self.max_try_time):
try:
a = json.loads(self.sess.get(url, headers=self.headers, cookies=self.cookie).text) # 获取任务列表
except JSONDecodeError:
logger.error('任务列表解析失败，即将重试..')
time.sleep(10)
if not self.login():
self.finish = '登录失败\n\n'
logger.error(self.account + ':' + '登录失败')
logger.info("运行结束")
return
self.auth()
continue
if a['code'] != 0:
logger.error(a['msg'])
time.sleep(5)
continue
data = a['data']
if len(data) == 0:
if i==0:
logger.info('没有任务')
self.send('%s打卡失败'%self.name,"没有任务")
logger.info("运行结束")
exit()
else:
logger.info('复检结束，无遗漏任务')
return
self.taskidlist = []
self.titlelist = []
for item in data: # 筛选有效任务
if cur > int(item['StartTime']):
self.taskidlist.append(item['TaskId'])
self.titlelist.append(item['Title'])
if len(self.taskidlist)==0:
if i==0:
logger.info('有任务但未到执行时间')
self.send('%s打卡失败'%self.name,'有任务但未到执行时间')
logger.info("运行结束")
exit()
else:
logger.info('复检结束，无遗漏任务')
return
if(i>0):
logger.info("检测到遗漏任务，重新执行")
self.finish += "检测到遗漏任务，重新执行\n\n"
self.post_data()
logger.error("已尝试至最大次数，可能打卡失败")
self.finish += "已尝试至最大次数，可能打卡失败\n\n"
def post_data(self):
self.finish+= '结果：\n\n'
for iter in range(len(self.taskidlist)):
taskid = self.taskidlist[iter]
title = self.titlelist[iter]
url = 'https://api.uyiban.com/officeTask/client/index/detail?TaskId=%s&CSRF=%s' % (taskid, self.csrf)
b = json.loads(self.sess.get(url, headers=self.headers, cookies=self.cookie).text) # 获取参数
url2 = 'https://api.uyiban.com/workFlow/c/my/apply/%s?CSRF=%s' % (b['data']['WFId'], self.csrf)
temper = round(random.uniform(36, 36.6), 1) # 生成区间位于36-36.6带有一位小数的体温
if self.address=="":#在校模式
# 表单数据：体温，本人是否有可疑症状，同住人是否有可疑症状
morning_data = {
'data': '{"ed9ff15f7155ed96682309ea8f865c94":"%s°","adbd34269e63dab3ceda0a9debb57733":"无以上症状","8525b81624577db90dd509b4301d1d21":"无以上症状"}' % temper,
'extend': '{"TaskId":"%s","title":"任务信息","content":[{"label":"任务名称","value":"%s"},{"label":"发布机构","value":"学生工作处"}]}' % (
taskid, title)
}
noon_data = {
'data': '{"ed9ff15f7155ed96682309ea8f865c94":"%s°","adbd34269e63dab3ceda0a9debb57733":"无以上症状","9a9c2732741377699aa2158cb58e54f2":"无以上症状"}' % temper,
'extend':'{"TaskId":"%s","title":"任务信息","content":[{"label":"任务名称","value":"%s"},{"label":"发布机构","value":"%s"},{"label":"发布人","value":"%s"}]}'%(taskid, title,b['data']['PubOrgName'],b['data']['PubPersonName'])
}
if title[-2:] == "晨检":
c = self.sess.post(url2, headers=self.headers, cookies=self.cookie, data=morning_data).text
elif title[-2:] == "午检":
c = self.sess.post(url2, headers=self.headers, cookies=self.cookie, data=noon_data).text
else:
logger.error(title + '晨检午检判断出错')
self.send('%s打卡失败' % self.name, '%s晨检午检判断出错' % title)
continue
else:#假期模式
if title[-1]!='检':
logger.error('表单已更新')
self.send('%s打卡失败' % self.name, '表单已更新')
continue
data = {
'data':'{"41c9b2c46fb085f0383d8590e0cfdd16":"%s°","35812c1e3bdc85acf0fa14b8843283ee":["是"],"3bc2d8a7a0b8901b972a049b8243b3c9":["是"],"53910c5a3ffa064348c5999a59f6601d":[%s]}'%(temper,self.address),
'extend':'{"TaskId":"%s","title":"任务信息","content":[{"label":"任务名称","value":"%s"},{"label":"发布机构","value":"%s"},{"label":"发布人","value":"%s"}]}'%(taskid, title,b['data']['PubOrgName'],b['data']['PubPersonName'])
}
c = self.sess.post(url2, headers=self.headers, cookies=self.cookie, data=data).text
c = json.loads(c)
if c['msg'] == '':
self.finish += '%s打卡成功\n\n' % title
logger.info('%s打卡成功' % title)
else:
self.finish += '%s打卡失败,%s\n\n'% (title,c['msg'])
logger.error('%s打卡失败,%s'% (title,c['msg']))
time.sleep(5)
def start(self):
if not self.login():
logger.error(self.account+':'+'登录失败')
self.send(self.account+':'+'登录失败')
logger.info("运行结束")
exit()
self.auth()
self.get_tasklist()
self.send('%s打卡结果'%self.name,self.finish)
logger.info("运行结束")
def start_night_attendance(self):
if not self.login():
logger.error(self.account + ':' + '登录失败')
self.send(self.account + ':' + '登录失败')
logger.info("运行结束")
exit()
self.auth()
check_url='https://api.uyiban.com/nightAttendance/student/index/signPosition?CSRF=%s'%self.csrf
post_url = 'https://api.uyiban.com/nightAttendance/student/index/signIn?CSRF=%s' % self.csrf
for i in range(self.max_try_time):
try:
res=json.loads(self.sess.get(check_url, headers=self.headers, cookies=self.cookie).text)
state = res['data']['State']
except TypeError:
if i == self.max_try_time-1:
self.send('%s晚检打卡失败' % self.name, '晚检任务解析失败')
logger.error("晚检任务解析失败")
logger.info("运行结束")
exit()
logger.error('晚检任务解析失败，即将重试')
time.sleep(10)
if not self.login():
logger.error(self.account + ':' + '登录失败')
self.send(self.account + ':' + '登录失败')
logger.info("运行结束")
exit()
self.auth()
continue
if state!=0:
if state==3:
if i==0:
self.send('%s晚检打卡失败'%self.name,'之前已完成晚点签到')
logger.info("之前已完成晚点签到")
else:
self.send('%s晚检打卡成功'%self.name,'晚检签到成功')
logger.info("晚检签到成功")
logger.info("运行结束")
exit()
else:
logger.info('晚点签到:%s'%res['data']['Msg'])
time.sleep(5)
continue
add1 = round(random.uniform(111.6987, 111.7011), 4)
add2=round(random.uniform(40.8140, 40.8159), 4)
data={
'Code':'',
'SignInfo':'{"Reason":"","AttachmentFileName":"","LngLat":"%s,%s","Address":"内蒙古自治区 呼和浩特市 赛罕区 金宇巷 109号 靠近内蒙古大学(东校区) "}'% (add1, add2),
'OutState':1
}
res = json.loads(self.sess.post(post_url, headers=self.headers, cookies=self.cookie,data=data).text)
if res['code']!=0:
logger.error('晚点签到失败，错误：%s'%res['msg'])
time.sleep(5)
logger.error('晚点签到:已尝试至最大次数，可能签到失败')
self.send('%s晚检打卡失败'%self.name,'晚点签到:已尝试至最大次数，可能签到失败')
logger.info("运行结束")
def __init__(self,account,pswd,server_url="",address=""):
self.sess = requests.session()
self.account = account
self.passwd = pswd
self.csrf =''.join(random.sample('zyxwvutsrqponmlkjihgfedcba0123456789',32))
self.cookie={'csrf_token':self.csrf}
self.headers = {
'origin': 'https://app.uyiban.com',
'referer': 'https://app.uyiban.com/',
'Host': 'api.uyiban.com',
'user-agent': 'yiban',
}
self.max_try_time=3
self.server_url=server_url
self.finish=""
self.address=address