-
Notifications
You must be signed in to change notification settings - Fork 46
/
daka.py
186 lines (162 loc) · 7.65 KB
/
daka.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
from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support.wait import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.common.exceptions import WebDriverException
from DingRobot import dingpush
import argparse
import datetime
import time
import os
class AutoDaka:
def __init__(self, args):
self.url = args.url
self.username = args.username
self.password = args.password
self.latitude = args.latitude
self.longitude = args.longitude
self.DD_BOT_TOKEN = args.DD_BOT_TOKEN
self.DD_BOT_SECRET = args.DD_BOT_SECRET
def init_driver(self):
chrome_options = webdriver.ChromeOptions()
chrome_options.add_argument("--disable-infobars") # 禁用浏览器正在受到自动化软件的控制的提示
if args.headless:
chrome_options.add_argument('--headless')
chrome_options.add_argument('--no-sandbox')
chrome_options.add_argument('window-size=1920x1080')
chrome_options.add_argument('--disable-gpu')
chrome_options.add_argument('--hide-scrollbars')
if args.proxy:
chrome_options.add_argument(f'--proxy-server={args.proxy-server}')
driver = webdriver.Chrome(options=chrome_options)
try:
driver.get(self.url)
except WebDriverException:
print("page down")
driver.maximize_window()
return driver
def login(self, driver):
print("\n[Time] %s" %
datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S'))
print("🚌 打卡任务启动")
username_input = driver.find_element(by=By.ID, value="username")
password_input = driver.find_element(by=By.ID, value="password")
login_button = driver.find_element(by=By.ID, value="dl")
print("登录到浙大统一身份认证平台...")
try:
username_input.send_keys(self.username)
password_input.send_keys(self.password)
print("已登录到浙大统一身份认证平台")
login_button.click()
time.sleep(1)
except Exception as err:
print(str(err))
raise Exception
def click_by_xpath(self, driver, form, xpath, submit=False):
try:
WebDriverWait(driver, 10).until(EC.element_to_be_clickable((By.XPATH, xpath))).click()
except Exception as err:
print(f'write {form} Information wrong...')
else:
if submit:
print('submit success...')
else:
print(f'write {form} Information success...')
time.sleep(1)
def daka(self, driver):
print("打卡任务启动...")
print("正在获得虚拟地理位置信息...")
# 获取虚拟地理位置信息
driver.execute_cdp_cmd(
"Browser.grantPermissions", # 授权地理位置信息
{
"origin": self.url,
"permissions": ["geolocation"]
},
)
driver.execute_cdp_cmd(
"Emulation.setGeolocationOverride", # 虚拟位置信息
{
"latitude": self.latitude,
"longitude": self.longitude,
"accuracy": args.accuracy
},
)
time.sleep(2) # 等待位置信息
# 需要提交的表单
forms = {'inSchool': '/html/body/div[1]/div[1]/div/section/div[4]/ul/li[4]/div/div/div[1]/span[1]', # on campus
'roomMate': '/html/body/div[1]/div[1]/div/section/div[4]/ul/li[7]/div/div/div[2]/span[1]', # people living with you
'inPractice': '/html/body/div[1]/div[1]/div/section/div[4]/ul/li[8]/div/div/div[3]/span[1]', # internship
'location': '/html/body/div[1]/div[1]/div/section/div[4]/ul/li[11]/div/input', # location
'commit': '/html/body/div[1]/div[1]/div/section/div[4]/ul/li[28]/div/div/div/span[1]', # commit
}
for form in forms:
self.click_by_xpath(driver, form, forms[form])
print("表单信息填写成功，正在提交表单...")
time.sleep(1)
# 提交表单
submit_xpath = '/html/body/div[1]/div[1]/div/section/div[5]/div/a'
self.click_by_xpath(driver, 'submit', submit_xpath, submit=True)
time.sleep(2)
# 弹出的确认提交窗口，点击确定
try:
# 寻找<div class="wapcf-btn wapcf-btn-ok">确认提交</div>的按钮
submit=driver.find_element(by=By.CLASS_NAME, value="wapcf-btn-ok")
submit = WebDriverWait(driver, 10).until(
EC.element_to_be_clickable(submit))
submit.click()
print("确认提交...")
self.Reminder("今天的打卡完成了🚌，耶！")
except:
try:
# 寻找<div class="wapat-title">每天只能填报一次，你已提交过</div>的按钮
HaveSubmitted=driver.find_element(by=By.CLASS_NAME, value="wapat-title")
print('您今天已提交过了...')
self.Reminder("您今天已提交过")
except Exception as error:
print('提交失败...')
# self.Reminder("提交失败,请手动填写或检查代码仓库更新情况")
time.sleep(1)
def Reminder(self, content):
if self.DD_BOT_TOKEN:
ding= dingpush('浙江大学每日健康打卡小助手', content, self.DD_BOT_TOKEN,self.DD_BOT_SECRET)
ding.SelectAndPush()
else:
print("钉钉推送未配置，请自行查看签到结果")
print("推送完成！")
def run(self):
driver = self.init_driver()
self.login(driver)
self.daka(driver)
driver.close()
print("打卡完成")
def get_parser():
user_name = os.getenv("account")
password = os.getenv("password")
token = os.getenv("DD_BOT_TOKEN")
secret = os.getenv("DD_BOT_SECRET")
url = 'https://healthreport.zju.edu.cn/ncov/wap/default/index'
parser = argparse.ArgumentParser()
parser.add_argument('--username', default=user_name, help='浙江大学统一身份认证平台用户名')
parser.add_argument('--password', default=password, help='浙江大学统一身份认证平台密码')
parser.add_argument('--latitude', type=float ,default=30.27, help='虚拟地理位置纬度, 默认为杭州市西湖区浙江大学')
parser.add_argument('--longitude', type=float, default=120.13, help='虚拟地理位置经度')
parser.add_argument('--accuracy', default=50, help='虚拟地理位置精度')
parser.add_argument('--url', type=str, default=url, help='浙江大学统一身份认证平台地址')
parser.add_argument('--headless', type=bool, default=True, help='是否开启无头模式')
parser.add_argument('--proxy', type=bool, default=False, help='是否使用代理')
parser.add_argument('--proxy-server', type=str, help='代理服务器地址 (e.g. http://')
parser.add_argument('--DD_BOT_TOKEN', type=str, default=token, help='钉钉机器人token')
parser.add_argument('--DD_BOT_SECRET', type=str, default=secret, help='钉钉机器人secret')
return parser
def print_arguments(args):
print("--------------------- Configuration Arguments ---------------------")
for arg, value in sorted(vars(args).items()):
print("%s: %s" % (arg, value))
print("--------------------------------------------------------------------")
if __name__ == "__main__":
parser = get_parser()
args = parser.parse_args()
print_arguments(args)
daka = AutoDaka(args)
daka.run()