-
Notifications
You must be signed in to change notification settings - Fork 0
/
Sign.py
143 lines (136 loc) · 6.49 KB
/
Sign.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
'''每日健康上报的连接模块'''
# !python3
# coding=UTF-8
import requests # HTTP/HTTPS方便接口
# import sys # 系统调用
# from urllib.parse import urlparse # URL解析
# import json # 用于处理json类型的返回数据
# 从单一文件ConnectToUSTC.py中引入依赖地方式
from ConnectToUSTC import getYmlConfig, conn_USTC # 连接USTC统一身份认证
from MyHTMLParser import Parser_lastdata # 自定义HTML解析器
from selenium import webdriver # 浏览器驱动程序
from selenium.webdriver.chrome.options import Options # 选项
import time # 休眠用
# https://weixine.ustc.edu.cn/2020/home
def USTC_dailysign(session, user):
'''进行一次每日上报'''
user = user['user']
headers = {
'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.9',
'Accept-Encoding': 'gzip, deflate, br',
'Accept-Language': 'zh-CN,zh;q=0.9,en-US;q=0.8,en;q=0.7,ja-JP;q=0.6,ja;q=0.5',
'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/89.0.4389.114 Safari/537.36',
'referer': 'https://weixine.ustc.edu.cn/2020/login',
'sec-ch-ua': '"Chromium";v="92", " Not A;Brand";v="99", "Google Chrome";v="92"',
'sec-ch-ua-mobile': '?0',
'sec-fetch-dest': 'document',
'sec-fetch-mode': 'navigate',
'sec-fetch-site': 'same-origin',
'sec-fetch-user': '?1',
'upgrade-insecure-requests': '1',
}
# 首次连接：相当于点击了“统一身份认证登录”，然后通过session内存的cookies自动通过统一身份认证，然后自动跳转到https://weixine.ustc.edu.cn/2020/home，返回的是首页的html信息
login_url = 'https://weixine.ustc.edu.cn/2020/caslogin'
print('connecting to %s ...' % login_url)
res = session.get(url=login_url, headers=headers)
# 解析html信息，获取上一次上报的地址、本次上报所需的token；故变更地址后只需要手动打卡一次，即可自动打卡
parser = Parser_lastdata()
parser.feed(res.text)
parser.close()
last_sign = parser.data
# 二次连接：打卡
sign_url = 'https://weixine.ustc.edu.cn/2020/daliy_report'
headers['referer'] = 'https://weixine.ustc.edu.cn/2020/home'
headers['origin'] = 'https://weixine.ustc.edu.cn'
headers['cache-control'] = 'max-age=0'
headers['content-length'] = '365'
headers['content-type'] = 'application/x-www-form-urlencoded'
params = {
'_token': last_sign['_token'],
'now_address': '1',
'gps_now_address': '',
'now_province': last_sign['now_province_hidden'],
'gps_province': '',
'now_city': last_sign['now_city_hidden'],
'gps_city': '',
'now_detail': '',
'is_inschool': '',
'body_condition': '1',
'body_condition_detail': '',
'now_status': '',
'now_status_detail': '',
'has_fever': '0',
'last_touch_sars': '0',
'last_touch_sars_date': '',
'last_touch_sars_detail': '',
'other_detail': '无'
}
# 如果在合肥，则'是否在校'=中区，'当前状态'=正常在校园内
# 否则(校外，正常在家)
# 具体接口可通过chrome开发者工具抓包获得
if last_sign['now_province_hidden'] == '340000':
params['is_inschool'] = '4'
params['now_status'] = '1'
else:
params['is_inschool'] = '0'
params['now_status'] = '2'
res = session.post(url=sign_url, data=params, headers=headers)
if (res.status_code == 200):
print('Daily Sign for {0} OK!'.format(user['username']))
else:
print('Daily Sign Failed')
def USTC_dailysign_selenium(session):
'''进行一次每日上报，selenium + webdriver实现'''
cookie_jar = session.cookies # 获取上次登录连接用的cookie
cookies = requests.utils.dict_from_cookiejar(cookie_jar) # 转成dict形式
# 预设选项
chrome_options = Options()
chrome_options.add_argument('--no-sandbox')
chrome_options.add_argument('--disable-dev-shm-usage')
chrome_options.add_argument('--headless')
chrome_options.add_argument('blink-settings=imagesEnabled=false')
chrome_options.add_argument('--disable-gpu')
driver = webdriver.Chrome(options=chrome_options) # 创建浏览器对象
driver.get('https://weixine.ustc.edu.cn/2020/home') # 打开网页，为了预先加载
for key, val in cookies.items():
# 将cookies以name:value的形式，逐个添加到selenium创建的新会话中
cookie = {}
cookie['name'] = key
cookie['value'] = val
driver.add_cookie(cookie)
driver.get('https://weixine.ustc.edu.cn/2020/home') # 打开网页
time.sleep(5) # 为了等待网页加载完成
driver.find_element_by_xpath("//button[@id='report-submit-btn']").click()
time.sleep(5) # 等待网页交互
driver.get_screenshot_as_file('screenshot.png') # 截图
driver.quit()
if __name__ == "__main__":
config = getYmlConfig('config.yml')
users = config['users']
for user in users:
session = conn_USTC(user)
USTC_dailysign(session, user)
cookie_jar = session.cookies # 获取上次登录连接用的cookie
cookies = requests.utils.dict_from_cookiejar(cookie_jar) # 转成dict形式
chrome_options = Options()
chrome_options.add_argument('--no-sandbox')
chrome_options.add_argument('--disable-dev-shm-usage')
chrome_options.add_argument('--headless')
chrome_options.add_argument('blink-settings=imagesEnabled=false')
chrome_options.add_argument('--disable-gpu')
driver = webdriver.Chrome(options=chrome_options) #创建浏览器对象
driver.get('https://weixine.ustc.edu.cn/2020/home') #打开网页
print(cookies)
for key, val in cookies.items():
# 将cookies以name:value的形式，逐个添加到selenium中
cookie = {}
cookie['name'] = key
cookie['value'] = val
driver.add_cookie(cookie)
driver.get('https://weixine.ustc.edu.cn/2020/home') #打开网页
time.sleep(5)
print(driver)
driver.find_element_by_xpath("//button[@id='report-submit-btn']").click()
time.sleep(5)
driver.get_screenshot_as_file('screenshot.png') # 截图
driver.quit()