AutoOpenLearning/bot1.py at master · Confusion-ymc/AutoOpenLearning · GitHub

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
import datetime
import logging
import time
import traceback

from selenium.webdriver.common.by import By

from base import MyBrowser
from driver_helper import DriverHelper

logging.getLogger().setLevel(logging.INFO)

driver_helper = DriverHelper()
execute_path = driver_helper.get_driver()


class LearnBot(MyBrowser):
    def __init__(self, remote_port=None):
        super(LearnBot, self).__init__()

        self.chrome_opt.add_argument('log-level=3')
        self.chrome_opt.add_argument("--disable-popup-blocking")
        if remote_port:
            logging.info(f'please start chrome with  --remote-debugging-port={remote_port}')
            # f' --remote-debugging-port={remote_port}')
            self.chrome_opt.add_experimental_option("debuggerAddress", f"127.0.0.1:{remote_port}")
            logging.info(f'wait chrome start...')
        self.load_browser()
        logging.info(f'connect success')
        self.count = 0
        self.driver.get('https://student.uestcedu.com/console/')

    def login(self, username, password):
        while True:
            self.driver.get('https://student.uestcedu.com/console/')
            self.find_element(by=By.XPATH, value='//*[@id="txtLoginName"]').send_keys(username)
            self.find_element(by=By.XPATH, value='//*[@id="txtPassword"]').send_keys(password)
            self.find_element(by=By.XPATH, value='//*[@id="verify_button"]').click()
            self.driver.switch_to.alert.accept()
            msm_code = input('验证码:')
            self.find_element(by=By.XPATH, value='//*[@id="txtSmsCode"]').send_keys(msm_code)
            self.find_element(by=By.XPATH, value='//*[@id="login_button"]').click()
            time.sleep(5)
            if 'https://student.uestcedu.com/console/main.html?' in self.driver.current_url:
                break

    def qr_login(self):
        self.find_element(by=By.XPATH, value='/html/body/div/div/div/div/div/ul/li[2]').click()

    def run(self):
        while True:
            try:
                choice = input('选择你要的操作，(默认)1.开始刷课，2.退出程序\n1):Run... \n2):Quit\n')
                if not choice or choice == '1':
                    pass
                elif choice == '2':
                    break
                else:
                    continue
                handles = self.driver.window_handles
                self.driver.switch_to.window(handles[-1])
                self.driver.switch_to.default_content()
                self.driver.switch_to.frame(
                    self.find_element(by=By.XPATH, value='/html/body/table/tbody/tr[2]/td/iframe'))
                self.driver.switch_to.frame(self.find_element(by=By.XPATH, value='//*[@id="w_code"]'))
                self.expand_all()
                time.sleep(2)
                node_list = self.find_not_finish()
                self.play(node_list)
                print('全部完成...')
            except Exception as e:
                traceback.print_exc()

    def play(self, a_list):
        for i, a in enumerate(a_list):
            a.click()
            time.sleep(5)
            title = a.text
            print(f'开始学习 {title}')
            self.wait_finish()
            self.count += 1
            print(f'{self.count} {datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")} {title} 学习完成')

    def expand_all(self):
        """
        展开所有加号
        :return:
        """
        for i in self.find_elements(by=By.CSS_SELECTOR, value="img[src='/_web_api/images/treeicon/plusnode.gif']"):
            try:
                i.click()
            except Exception as e:
                # traceback.print_exc()
                pass
        for i in self.find_elements(by=By.CSS_SELECTOR,
                                    value="img[src='/_web_api/images/treeicon/cornerplusnode.gif']"):
            try:
                i.click()
            except Exception as e:
                # traceback.print_exc()
                pass

    def find_not_finish(self):
        """
        查找待学习的标签
        :return:
        """
        # 未开始
        not_attempt_list = self.find_elements(by=By.CSS_SELECTOR, value="span.scorm.notattempt")
        # 未完成
        incomplete = self.find_elements(by=By.CSS_SELECTOR, value="span.scorm.incomplete")
        return [i.find_element(By.XPATH, '../a') for i in (incomplete + not_attempt_list)]

    def wait_finish(self):
        self.driver.switch_to.default_content()
        self.driver.switch_to.frame(self.find_element('/html/body/table/tbody/tr[2]/td/iframe'))
        self.driver.switch_to.frame(self.find_element('//*[@id="w_lms_content"]'))
        wait_count = 0
        while wait_count <= 60 * 60:
            if '您正在学习' in self.driver.find_elements_by_xpath('//td')[-1].text:
                time.sleep(5)
                wait_count += 5
            else:
                break
        self.driver.switch_to.default_content()
        self.driver.switch_to.frame(self.find_element('/html/body/table/tbody/tr[2]/td/iframe'))
        self.driver.switch_to.frame(self.find_element('//*[@id="w_code"]'))

    def stop(self):
        try:
            self.driver.quit()
        except:
            pass
        print('正常退出')


if __name__ == '__main__':
    learn_bot = LearnBot()
    learn_bot.qr_login()
    learn_bot.run()
    learn_bot.stop()