forked from yxKryptonite/FuckPupedu
-
Notifications
You must be signed in to change notification settings - Fork 0
/
fucker.py
287 lines (247 loc) · 11.5 KB
/
fucker.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
import selenium
from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.common.exceptions import StaleElementReferenceException
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions
from selenium.webdriver.chrome.options import Options
from selenium.webdriver import ActionChains
from selenium.webdriver.common.keys import Keys
import time
from tqdm import tqdm
from logger import Logger
IMMED_INTERVAL = 0.1
SHORT_INTERVAL = 1
MID_INTERVAL = 5
LONG_INTERVAL = 300 # 5 minutes
HALF_MINUTE = 30
ONE_MINUTE = 60
PLAYBACK_RATE = 10 # 回放速度
class FuckPupedu(object):
def __init__(self, cfg):
self.cfg = cfg
self.mobile_emulation = {"deviceName": "iPhone 6"}
options = Options()
options.add_experimental_option("mobileEmulation", self.mobile_emulation)
options.add_argument('-mute-audio')
if not cfg['DEBUG']:
options.add_argument('--headless')
self.driver = webdriver.Chrome(options=options)
self.functions = {
"VIDEO": self.play_video,
"PPT": self.watch_ppt,
"NOTES": self.take_notes,
"TEST": self.do_test
}
def login(self, url):
'''
- 初始界面：空
- 目标：登录
- 返回值：是否成功登录
'''
try:
self.driver.get(url)
self.driver.find_element(By.XPATH, "//*[text()='北大学生登录入口']").click()
self.driver.implicitly_wait(MID_INTERVAL)
self.driver.find_element(By.ID, "user_name").send_keys(self.cfg['ID'])
self.driver.find_element(By.ID, "password").send_keys(self.cfg['PASSWORD'])
self.driver.find_element(By.ID, "logon_button").click()
self.driver.implicitly_wait(MID_INTERVAL)
return True
except Exception as e:
Logger.log(f'Login error: \n{e}')
return False
def get_into_course(self, course_name):
'''
- 初始界面：我的课程
- 目标：进入劳动教育课程
- 返回值：是否成功进入
'''
main_boxes = self.driver.find_elements(By.CLASS_NAME, "mainBox")
for main_box in main_boxes:
title_name = main_box.find_element(By.CLASS_NAME, "titleName").text
if title_name == course_name:
btn = main_box.find_element(By.CLASS_NAME, "btn4")
btn.click()
self.driver.implicitly_wait(LONG_INTERVAL)
return True
return False
def get_courses(self, learn_type):
'''
- 初始页面：劳动教育课程
- 目标：搜寻所有章节下的所有符合类型的标题元素
- 返回值: 根据 `learn_type` 返回的 `courses`: len(courses) 为 chapter 数, courses[i] 为第 i 个 chapter 的所有 title 的 list
- 视频 courses: `[[title1, title2, ...], [title1, title2, ...], ...]`
- PPT courses: `[[PPT1], [PPT2], ...]`
- 笔记 courses: `[[PPT1], [PPT2], ...]`
- 测验 courses: `[[TEST1], [TEST2], ...]`
'''
time.sleep(MID_INTERVAL) # 先等待一段时间，等待所有章节加载完毕
container = self.driver.find_element(By.ID, "step2")
chapters = container.find_elements(By.CLASS_NAME, "chapters")
courses = []
for chapter in chapters:
self.driver.execute_script("arguments[0].scrollIntoView();", chapter)
self.driver.implicitly_wait(LONG_INTERVAL)
chapter.click()
self.driver.implicitly_wait(LONG_INTERVAL)
titles = chapter.find_elements(By.CLASS_NAME, "titleName")
if learn_type == "VIDEO":
courses.append(titles[2:-1]) # 去掉前两个和最后一个，剩余的为视频
elif learn_type == "NOTES":
courses.append([titles[1]]) # 在一节中，发布2条笔记、5条评论即可
elif learn_type == "PPT":
courses.append([titles[1]]) # PPT
elif learn_type == "TEST":
courses.append([titles[-1]]) # 测验
else:
raise NotImplementedError("learn_type {} is not implemented!".format(learn_type))
return courses
def get_idx(self, count, courses):
'''
返回值：当前应该学习的 `章节索引` 和 `标题索引`
'''
chapter_idx = 0
title_idx = 0
for chapter in courses:
if count < len(chapter):
title_idx = count
break
else:
chapter_idx += 1
count -= len(chapter)
return chapter_idx, title_idx
def learn(self, learn_type):
'''
- 当前页面：劳动教育课程
- 目标：根据 `learn_type` 决定怎样学习课程
'''
current_url = self.driver.current_url
courses = self.get_courses(learn_type) # [[title1, title2, ...], [title1, title2, ...], ...]
total_num = sum([len(chapter) for chapter in courses])
func = self.functions.get(learn_type)
count = 0
for i in range(self.cfg[learn_type]['START_CHAPTER'] - 1):
count += len(courses[i])
count += self.cfg[learn_type]['START_TITLE'] - 1
while True:
chapter_idx, title_idx = self.get_idx(count, courses)
Logger.log("开始学习第 {} 讲的第 {} 个 {}...".format(chapter_idx + 1, title_idx + 1, self.cfg[learn_type]['NAME']))
func(courses[chapter_idx][title_idx]) # 调用 `func` 函数
count += 1
self.driver.get(current_url)
self.driver.implicitly_wait(LONG_INTERVAL)
if count < total_num:
Logger.log("学习完毕，正在加载下一个 {}...".format(self.cfg[learn_type]['NAME']))
else:
Logger.log("恭喜您，所有 {} 已经学习完毕！".format(self.cfg[learn_type]['NAME']))
break
courses = self.get_courses(learn_type)
while sum(len(chapter) for chapter in courses) < total_num:
self.driver.refresh()
courses = self.get_courses(learn_type)
def play_video(self, title):
'''
- 初始界面：劳动教育课程
- 目标：根据 `title` 这个 `WebElement` 播放视频
'''
self.driver.execute_script("arguments[0].scrollIntoView();", title)
title.click()
self.driver.implicitly_wait(LONG_INTERVAL)
btn = self.driver.find_element(By.CLASS_NAME, "outter")
btn.click()
time.sleep(MID_INTERVAL) # wait for the duration to be loaded
duration_div = self.driver.find_element(By.CLASS_NAME, "duration").text # e.g. 19:25
duration_div = duration_div.split(":")
duration = int(duration_div[0]) * ONE_MINUTE + int(duration_div[1])
for sec in tqdm(range((duration + ONE_MINUTE) // PLAYBACK_RATE + 1)): # 加一分钟，用于补偿刷新的延迟
# 设置播放倍速
self.driver.execute_script(f"document.getElementsByTagName('video')[0].playbackRate = {PLAYBACK_RATE};")
time.sleep(SHORT_INTERVAL)
if sec % (4*ONE_MINUTE) == (4*ONE_MINUTE) - 1:
# 每 4 分钟刷新一次页面，避免弹窗
self.driver.refresh()
self.driver.implicitly_wait(LONG_INTERVAL)
self.driver.find_element(By.CLASS_NAME, "outter").click()
def watch_ppt(self, title):
'''
- 初始界面：劳动教育课程
- 目标：根据 `title` 这个 `WebElement` 看 PPT
'''
self.driver.execute_script("arguments[0].scrollIntoView();", title)
title.click()
time.sleep(MID_INTERVAL)
# remove mobile emulation
self.driver.execute_cdp_cmd("Emulation.clearDeviceMetricsOverride", {})
try:
rb_btn = self.driver.find_element(By.CLASS_NAME, "rb")
except:
Logger.log("该 PPT 暂时无法加载，正在加载下一个...")
return
start_time = time.time()
while True:
curr_time = time.time()
if curr_time - start_time > HALF_MINUTE:
break
rb_btn.click()
# add mobile emulation back
self.driver.create_options().add_experimental_option("mobileEmulation", self.mobile_emulation)
def take_notes(self, title):
'''
- 初始界面：劳动教育课程
- 目标：根据 `title` 这个 `WebElement` 记录笔记
- 注：发布2条笔记、5条评论即可满足要求
'''
self.driver.execute_script("arguments[0].scrollIntoView();", title)
title.click()
time.sleep(MID_INTERVAL)
# remove mobile emulation
self.driver.execute_cdp_cmd("Emulation.clearDeviceMetricsOverride", {})
# 发笔记
for _ in range(2):
self.driver.find_element(By.CLASS_NAME, "addNoteBtn").click()
time.sleep(SHORT_INTERVAL)
self.driver.find_element(By.CLASS_NAME, "el-textarea__inner").send_keys(self.cfg['NOTES']['MY_NOTES'])
submitNoteBtn = self.driver.find_element(By.CLASS_NAME, "submitNoteBtn")
self.driver.execute_script("arguments[0].click();", submitNoteBtn)
time.sleep(SHORT_INTERVAL)
# 发评论
self.driver.find_element(By.CLASS_NAME, "el-switch__core").click()
time.sleep(SHORT_INTERVAL)
self.driver.find_elements(By.CLASS_NAME, "commentBtn")[0].click()
time.sleep(SHORT_INTERVAL)
for _ in range(5):
cmt_area = self.driver.find_element(By.CLASS_NAME, "commentListTop")
cmt_area.find_element(By.CLASS_NAME, "el-input__inner").send_keys(self.cfg['NOTES']['MY_NOTES'])
cmt_area.find_element(By.CSS_SELECTOR, "span").click()
time.sleep(SHORT_INTERVAL)
# add mobile emulation back
self.driver.create_options().add_experimental_option("mobileEmulation", self.mobile_emulation)
def do_test(self, title):
'''
- 初始界面：劳动教育课程
- 目标：根据 `title` 这个 `WebElement` 做测验
'''
self.driver.execute_script("arguments[0].scrollIntoView();", title)
title.click()
time.sleep(MID_INTERVAL)
# 已经做过了但没有被计入
try:
self.driver.switch_to.alert.accept()
self.driver.find_element(By.ID, "homeworkReSubmit").click()
self.driver.implicitly_wait(MID_INTERVAL)
except:
pass
answer_areas = self.driver.find_elements(By.CLASS_NAME, "answerArea")
for answer_area in answer_areas:
choice = answer_area.find_elements(By.CSS_SELECTOR, "a")[0] # 选A
self.driver.execute_script("arguments[0].scrollIntoView();", choice)
self.driver.execute_script("arguments[0].click();", choice)
self.driver.find_element(By.ID, "homeworkSubmit").click()
time.sleep(SHORT_INTERVAL) # 等待弹窗
self.driver.switch_to.alert.accept()
time.sleep(MID_INTERVAL) # 等待下一个弹窗
try:
self.driver.switch_to.alert.accept()
except:
pass