-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathtraversal.py
175 lines (143 loc) · 7.96 KB
/
traversal.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
import random
import time
from selenium.common import NoSuchElementException, StaleElementReferenceException
from selenium.webdriver.common.by import By
import MainWindow
class link_changed(Exception):
def __init__(self, message):
super().__init__(message)
class traversal_over(Exception):
def __init__(self, message):
super().__init__(message)
def waitRandomTime():
time.sleep(random.uniform(MainWindow.minTime, MainWindow.maxTime))
def traversal_course(driver, ocr, platform):
cIndex = 0
while True:
courseL = driver.find_element(By.CLASS_NAME, value="user-course")
courseList = courseL.find_elements(By.CLASS_NAME, value="item")
if cIndex >= len(courseList):
print('没有待刷课程')
break
no_complete_course = courseList[cIndex]
if no_complete_course.find_element(By.CLASS_NAME, value="txt").text == "100%":
cIndex += 1
continue
no_complete_course.find_element(By.CLASS_NAME, value="name").find_element(By.TAG_NAME, value="a").click()
continueButton = driver.find_element(By.XPATH, value="/html/body/div[3]/div[2]/div/div[1]/div[2]/div[6]/div[1]/a")
if continueButton.text == '尚未开始' or continueButton.text == '已结束':
driver.back()
cIndex += 1
time.sleep(0.5)
continue
continueButton.click()
while True:
try:
browser_url = driver.current_url
videoSlot = driver.find_element(By.CLASS_NAME, value="detmain-navlist")
videoList = videoSlot.find_elements(By.XPATH, value="./*")
currVideoListIndex = 0
currVideoIndex = 0
for element in videoList:
if element.get_attribute("class") == "group two on":
break
else:
currVideoListIndex += 1
itemList = videoList[currVideoListIndex].find_element(By.CLASS_NAME, value="list").find_elements(
By.CLASS_NAME, value="item")
for video in itemList:
if video.find_element(By.TAG_NAME, value="a").get_attribute("class") == "on":
break
else:
currVideoIndex += 1
if browser_url != driver.current_url:
raise link_changed("页面已跳转")
print(f"""正在播放第{currVideoListIndex + 1}章, 第{currVideoIndex + 1}节...""")
isNew = False
for i in range(currVideoListIndex, len(videoList)):
videoSlot = driver.find_element(By.CLASS_NAME, value="detmain-navlist")
videoList = videoSlot.find_elements(By.XPATH, value="./*")
itemList = videoList[i].find_element(By.CLASS_NAME, value="list").find_elements(By.CLASS_NAME, value="item")
if browser_url != driver.current_url:
raise link_changed("页面已跳转")
for j in range(currVideoIndex, len(itemList)):
if j != currVideoIndex or isNew:
# 等待一段随机时间
waitRandomTime()
if browser_url != driver.current_url:
raise link_changed("页面已跳转")
itemList[j].click()
print('播放下一个视频')
browser_url = driver.current_url
elif browser_url != driver.current_url:
raise link_changed("页面已跳转")
videoSlot = driver.find_element(By.CLASS_NAME, value="detmain-navlist")
videoList = videoSlot.find_elements(By.XPATH, value="./*")
itemList = videoList[i].find_element(By.CLASS_NAME, value="list").find_elements(By.CLASS_NAME, value="item")
playButton = driver.find_element(By.XPATH, value='//*[@id="videoContent"]/div/div[2]/div[1]')
# actions = ActionChains(driver)
# actions.move_to_element(playButton).perform()
while True:
try:
# 等待一段随机时间
waitRandomTime()
playButton.click()
time.sleep(0.5)
except:
print('点击播放按钮异常')
if browser_url != driver.current_url:
raise link_changed("页面已跳转")
try:
if 'none' in playButton.get_attribute('style'):
print('已开始播放')
break
dialog = driver.find_element(By.CSS_SELECTOR, value='[id^="layui-layer"]')
codeImg = driver.find_element(By.ID, value='codeImg').screenshot_as_png
res = ocr.classification(codeImg)
print(res)
vertify_input = dialog.find_element(By.CSS_SELECTOR, value='input:not([id="yzCode"])')
vertify_input.send_keys(res)
waitRandomTime()
if browser_url != driver.current_url:
raise link_changed("页面已跳转")
playButton2 = driver.find_element(By.LINK_TEXT, value='开始播放')
playButton2.click()
time.sleep(0.5)
except NoSuchElementException as e:
print('无弹窗与验证码')
continue
while True:
try:
if browser_url != driver.current_url:
raise link_changed("页面已跳转")
playingTimeElement = driver.find_element(By.XPATH, value="//*[starts-with(@class, 'timetext')]")
try:
playingTimeText = playingTimeElement.text
textList = playingTimeText.split()
if len(textList) == 0:
pass
elif textList[0] == textList[2]:
print('播放完毕！')
break
except StaleElementReferenceException as e:
print('未获取到播放时间文本')
except NoSuchElementException as e:
print('找不到播放时间元素')
if j == len(itemList) - 1:
currVideoIndex = 0
if browser_url != driver.current_url:
raise link_changed("页面已跳转")
if i + 1 < len(videoList):
driver.execute_script("arguments[0].className = 'group two on'", videoList[i + 1])
isNew = True
print('展开下一章')
else:
raise traversal_over("已刷完")
except link_changed as e:
print('页面已跳转')
continue
except traversal_over as o:
print('已刷完一门网课')
js = f"window.location.replace('{platform}')"
driver.execute_script(js)
break