This repository has been archived by the owner on Jan 20, 2022. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 206
/
index.py
390 lines (347 loc) · 15.8 KB
/
index.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
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
# -*- coding: utf-8 -*-
import sys
import requests
import json
import yaml
import oss2
from urllib.parse import urlparse
from datetime import datetime, timedelta, timezone
from urllib3.exceptions import InsecureRequestWarning
import smtplib
from email.mime.text import MIMEText
from email.utils import formataddr
# debug模式
debug = False
if debug:
requests.packages.urllib3.disable_warnings(InsecureRequestWarning)
# 读取yml配置
def getYmlConfig(yaml_file='config.yml'):
file = open(yaml_file, 'r', encoding="utf-8")
file_data = file.read()
file.close()
config = yaml.load(file_data, Loader=yaml.FullLoader)
return dict(config)
# 全局配置
config = getYmlConfig(yaml_file='config.yml')
# 获取今日校园api
def getCpdailyApis(user):
apis = {}
user = user['user']
schools = requests.get(
url='https://www.cpdaily.com/v6/config/guest/tenant/list', verify=not debug).json()['data']
flag = True
for one in schools:
if one['name'] == user['school']:
if one['joinType'] == 'NONE':
log(user['school'] + ' 未加入今日校园')
exit(-1)
flag = False
params = {
'ids': one['id']
}
res = requests.get(url='https://www.cpdaily.com/v6/config/guest/tenant/info', params=params,
verify=not debug)
data = res.json()['data'][0]
joinType = data['joinType']
idsUrl = data['idsUrl']
ampUrl = data['ampUrl']
if 'campusphere' in ampUrl or 'cpdaily' in ampUrl:
parse = urlparse(ampUrl)
host = parse.netloc
res = requests.get(parse.scheme + '://' + host)
parse = urlparse(res.url)
apis[
'login-url'] = idsUrl + '/login?service=' + parse.scheme + r"%3A%2F%2F" + host + r'%2Fportal%2Flogin'
apis['host'] = host
ampUrl2 = data['ampUrl2']
if 'campusphere' in ampUrl2 or 'cpdaily' in ampUrl2:
parse = urlparse(ampUrl2)
host = parse.netloc
res = requests.get(parse.scheme + '://' + host)
parse = urlparse(res.url)
apis[
'login-url'] = idsUrl + '/login?service=' + parse.scheme + r"%3A%2F%2F" + host + r'%2Fportal%2Flogin'
apis['host'] = host
break
if flag:
log(user['school'] + ' 未找到该院校信息,请检查是否是学校全称错误')
exit(-1)
log(apis)
return apis
# 获取当前utc时间,并格式化为北京时间
def getTimeStr():
utc_dt = datetime.utcnow().replace(tzinfo=timezone.utc)
bj_dt = utc_dt.astimezone(timezone(timedelta(hours=8)))
return bj_dt.strftime("%Y-%m-%d %H:%M:%S")
# 输出调试信息,并及时刷新缓冲区
def log(content):
print(getTimeStr() + ' ' + str(content))
sys.stdout.flush()
# 登陆并返回session
def getSession(user, loginUrl):
user = user['user']
params = {
'login_url': loginUrl,
# 保证学工号和密码正确下面两项就不需要配置
'needcaptcha_url': '',
'captcha_url': '',
'username': user['username'],
'password': user['password']
}
cookies = {}
# 借助上一个项目开放出来的登陆API,模拟登陆
res = requests.post(config['login']['api'], params, verify=not debug)
cookieStr = str(res.json()['cookies'])
log(cookieStr)
if cookieStr == 'None':
log(res.json())
return None
# 解析cookie
for line in cookieStr.split(';'):
name, value = line.strip().split('=', 1)
cookies[name] = value
session = requests.session()
session.cookies = requests.utils.cookiejar_from_dict(cookies)
return session
# 查询表单
def queryForm(session, apis):
host = apis['host']
headers = {
'Accept': 'application/json, text/plain, */*',
'User-Agent': 'Mozilla/5.0 (Linux; Android 4.4.4; OPPO R11 Plus Build/KTU84P) AppleWebKit/537.36 (KHTML, like Gecko) Version/4.0 Chrome/33.0.0.0 Safari/537.36 yiban/8.1.11 cpdaily/8.1.11 wisedu/8.1.11',
'content-type': 'application/json',
'Accept-Encoding': 'gzip,deflate',
'Accept-Language': 'zh-CN,en-US;q=0.8',
'Content-Type': 'application/json;charset=UTF-8'
}
queryCollectWidUrl = 'https://{host}/wec-counselor-collector-apps/stu/collector/queryCollectorProcessingList'.format(
host=host)
params = {
'pageSize': 6,
'pageNumber': 1
}
res = session.post(queryCollectWidUrl, headers=headers,
data=json.dumps(params), verify=not debug)
if len(res.json()['datas']['rows']) < 1:
return None
collectWid = res.json()['datas']['rows'][0]['wid']
formWid = res.json()['datas']['rows'][0]['formWid']
detailCollector = 'https://{host}/wec-counselor-collector-apps/stu/collector/detailCollector'.format(
host=host)
res = session.post(url=detailCollector, headers=headers,
data=json.dumps({"collectorWid": collectWid}), verify=not debug)
schoolTaskWid = res.json()['datas']['collector']['schoolTaskWid']
getFormFields = 'https://{host}/wec-counselor-collector-apps/stu/collector/getFormFields'.format(
host=host)
res = session.post(url=getFormFields, headers=headers, data=json.dumps(
{"pageSize": 100, "pageNumber": 1, "formWid": formWid, "collectorWid": collectWid}), verify=not debug)
form = res.json()['datas']['rows']
return {'collectWid': collectWid, 'formWid': formWid, 'schoolTaskWid': schoolTaskWid, 'form': form}
# 填写form
def fillForm(session, form, host):
sort = 1
for formItem in form[:]:
# 只处理必填项
if formItem['isRequired'] == 1:
default = config['cpdaily']['defaults'][sort - 1]['default']
if formItem['title'] != default['title']:
log('第%d个默认配置不正确,请检查' % sort)
exit(-1)
# 文本直接赋值
if formItem['fieldType'] == 1 or formItem['fieldType'] == 5:
formItem['value'] = default['value']
# 单选框需要删掉多余的选项
if formItem['fieldType'] == 2:
# 填充默认值
formItem['value'] = default['value']
fieldItems = formItem['fieldItems']
for i in range(0, len(fieldItems))[::-1]:
if fieldItems[i]['content'] != default['value']:
del fieldItems[i]
# 多选需要分割默认选项值,并且删掉无用的其他选项
if formItem['fieldType'] == 3:
fieldItems = formItem['fieldItems']
defaultValues = default['value'].split(',')
for i in range(0, len(fieldItems))[::-1]:
flag = True
for j in range(0, len(defaultValues))[::-1]:
if fieldItems[i]['content'] == defaultValues[j]:
# 填充默认值
formItem['value'] += defaultValues[j] + ' '
flag = False
if flag:
del fieldItems[i]
# 图片需要上传到阿里云oss
if formItem['fieldType'] == 4:
fileName = uploadPicture(session, default['value'], host)
formItem['value'] = getPictureUrl(session, fileName, host)
log('必填问题%d:' % sort + formItem['title'])
log('答案%d:' % sort + formItem['value'])
sort += 1
else:
form.remove(formItem)
# print(form)
return form
# 上传图片到阿里云oss
def uploadPicture(session, image, host):
url = 'https://{host}/wec-counselor-collector-apps/stu/collector/getStsAccess'.format(
host=host)
res = session.post(url=url, headers={
'content-type': 'application/json'}, data=json.dumps({}), verify=not debug)
datas = res.json().get('datas')
fileName = datas.get('fileName')
accessKeyId = datas.get('accessKeyId')
accessSecret = datas.get('accessKeySecret')
securityToken = datas.get('securityToken')
endPoint = datas.get('endPoint')
bucket = datas.get('bucket')
bucket = oss2.Bucket(oss2.Auth(access_key_id=accessKeyId,
access_key_secret=accessSecret), endPoint, bucket)
with open(image, "rb") as f:
data = f.read()
bucket.put_object(key=fileName, headers={
'x-oss-security-token': securityToken}, data=data)
res = bucket.sign_url('PUT', fileName, 60)
# log(res)
return fileName
# 获取图片上传位置
def getPictureUrl(session, fileName, host):
url = 'https://{host}/wec-counselor-collector-apps/stu/collector/previewAttachment'.format(
host=host)
data = {
'ossKey': fileName
}
res = session.post(url=url, headers={
'content-type': 'application/json'}, data=json.dumps(data), verify=not debug)
photoUrl = res.json().get('datas')
return photoUrl
# 提交表单
def submitForm(formWid, address, collectWid, schoolTaskWid, form, session, host):
headers = {
'User-Agent': 'Mozilla/5.0 (Linux; Android 4.4.4; OPPO R11 Plus Build/KTU84P) AppleWebKit/537.36 (KHTML, like Gecko) Version/4.0 Chrome/33.0.0.0 Safari/537.36 okhttp/3.12.4',
'CpdailyStandAlone': '0',
'extension': '1',
'Cpdaily-Extension': '1wAXD2TvR72sQ8u+0Dw8Dr1Qo1jhbem8Nr+LOE6xdiqxKKuj5sXbDTrOWcaf v1X35UtZdUfxokyuIKD4mPPw5LwwsQXbVZ0Q+sXnuKEpPOtk2KDzQoQ89KVs gslxPICKmyfvEpl58eloAZSZpaLc3ifgciGw+PIdB6vOsm2H6KSbwD8FpjY3 3Tprn2s5jeHOp/3GcSdmiFLYwYXjBt7pwgd/ERR3HiBfCgGGTclquQz+tgjJ PdnDjA==',
'Content-Type': 'application/json; charset=utf-8',
# 请注意这个应该和配置文件中的host保持一致
'Host': host,
'Connection': 'Keep-Alive',
'Accept-Encoding': 'gzip'
}
# 默认正常的提交参数json
params = {"formWid": formWid, "address": address, "collectWid": collectWid, "schoolTaskWid": schoolTaskWid,
"form": form}
# print(params)
submitForm = 'https://{host}/wec-counselor-collector-apps/stu/collector/submitForm'.format(
host=host)
r = session.post(url=submitForm, headers=headers,
data=json.dumps(params), verify=not debug)
msg = r.json()['message']
return msg
title_text = '今日校园疫结果通知'
# 发送邮件通知
def sendMessage(send, msg):
if send != '':
log('正在发送邮件通知。。。')
res = requests.post(url='http://www.zimo.wiki:8080/mail-sender/sendMail',
data={'title': title_text, 'content': getTimeStr() + str(msg), 'to': send})
code = res.json()['code']
if code == 0:
log('发送邮件通知成功。。。')
else:
log('发送邮件通知失败。。。')
log(res.json())
def sendEmail(send,msg):
my_sender= config['Info']['Email']['account'] # 发件人邮箱账号
my_pass = config['Info']['Email']['password'] # 发件人邮箱密码
my_user = send # 收件人邮箱账号,我这边发送给自己
try:
msg=MIMEText(getTimeStr() + str(msg),'plain','utf-8')
msg['From']=formataddr(["FromRunoob",my_sender]) # 括号里的对应发件人邮箱昵称、发件人邮箱账号
msg['To']=formataddr(["FK",my_user]) # 括号里的对应收件人邮箱昵称、收件人邮箱账号
msg['Subject']=title_text # 邮件的主题,也可以说是标题
server=smtplib.SMTP_SSL(config['Info']['Email']['server'], config['Info']['Email']['port']) # 发件人邮箱中的SMTP服务器,端口是25
server.login(my_sender, my_pass) # 括号中对应的是发件人邮箱账号、邮箱密码
server.sendmail(my_sender,[my_user,],msg.as_string()) # 括号中对应的是发件人邮箱账号、收件人邮箱账号、发送邮件
server.quit() # 关闭连接
except Exception: # 如果 try 中的语句没有执行,则会执行下面的 ret=False
log("邮件发送失败")
else: print("邮件发送成功")
# server酱通知
def sendServerChan(msg):
log('正在发送Server酱。。。')
res = requests.post(url='https://sc.ftqq.com/{0}.send'.format(config['Info']['ServerChan']),
data={'text': title_text, 'desp': getTimeStr() + "\n" + str(msg)})
code = res.json()['errmsg']
if code == 'success':
log('发送Server酱通知成功。。。')
else:
log('发送Server酱通知失败。。。')
log('Server酱返回结果'+code)
# Qmsg酱通知
def sendQmsgChan(msg):
log('正在发送Qmsg酱。。。')
res = requests.post(url='https://qmsg.zendee.cn:443/send/{0}'.format(config['Info']['Qsmg']),
data={'msg': title_text + '\n时间:' + getTimeStr() + "\n 返回结果:" + str(msg)})
code = res.json()['success']
if code:
log('发送Qmsg酱通知成功。。。')
else:
log('发送Qmsg酱通知失败。。。')
log('Qmsg酱返回结果'+code)
# 综合提交
def InfoSubmit(msg, send=None):
if(None != send):
if(config['Info']['Email']['enable']): sendEmail(send,msg)
else: sendMessage(send, msg)
if(config['Info']['ServerChan']): sendServerChan(msg)
if(config['Info']['Qsmg']): sendQmsgChan(msg)
def main_handler(event, context):
try:
for user in config['users']:
log('当前用户:' + str(user['user']['username']))
apis = getCpdailyApis(user)
log('脚本开始执行。。。')
log('开始模拟登陆。。。')
session = getSession(user, apis['login-url'])
if session != None:
log('模拟登陆成功。。。')
log('正在查询最新待填写问卷。。。')
params = queryForm(session, apis)
if str(params) == 'None':
log('获取最新待填写问卷失败,可能是辅导员还没有发布。。。')
InfoSubmit('没有新问卷')
exit(-1)
log('查询最新待填写问卷成功。。。')
log('正在自动填写问卷。。。')
form = fillForm(session, params['form'], apis['host'])
log('填写问卷成功。。。')
log('正在自动提交。。。')
msg = submitForm(params['formWid'], user['user']['address'], params['collectWid'],
params['schoolTaskWid'], form, session, apis['host'])
if msg == 'SUCCESS':
log('自动提交成功!')
InfoSubmit('自动提交成功!', user['user']['email'])
elif msg == '该收集已填写无需再次填写':
log('今日已提交!')
# InfoSubmit('今日已提交!', user['user']['email'])
InfoSubmit('今日已提交!')
else:
log('自动提交失败。。。')
log('错误是' + msg)
InfoSubmit('自动提交失败!错误是' + msg, user['user']['email'])
exit(-1)
else:
log('模拟登陆失败。。。')
log('原因可能是学号或密码错误,请检查配置后,重启脚本。。。')
exit(-1)
except Exception as e:
InfoSubmit("出现问题了!"+str(e))
raise e
else:
return 'success'
# 配合Windows计划任务等使用
if __name__ == '__main__':
print(main_handler({}, {}))
# for user in config['users']:
# log(getCpdailyApis(user))