Python Selenium 自动化：获取内蒙古开放大学待批改题目

from selenium.webdriver.chrome.options import Options
from selenium import webdriver
from selenium.webdriver.common.by import By
import time
import os
import random


def get_driver():
    # 检查是否已经打开浏览器
    browser_opened = False
    for handle in webdriver.Chrome().window_handles:
        browser_opened = True
        break

    # 创建浏览器实例或在已有浏览器中操作
    if browser_opened:
        options = Options()
        options.debugger_address = '127.0.0.1:9222'
        driver = webdriver.Chrome(options=options)
    else:
        os.system(r'start chrome --remote-debugging-port=9222 --user-data-dir="D:\评阅用"')
        options = Options()
        options.add_experimental_option('debuggerAddress', '127.0.0.1:9222')
        driver = webdriver.Chrome(options=options)

    return driver


driver = get_driver()

# 找到有'内蒙古开放大学'字样的标签页
while True:
    for handle in driver.window_handles:
        driver.switch_to.window(handle)
        if '内蒙古开放大学' in driver.title:
            print('登录成功')
            break
    else:
        time.sleep(3)
        continue
    break

# 统计题目组
elements = driver.find_elements(By.CLASS_NAME, 'subject-table-row')
print('题目组数量：', len(elements), '组')

# 遍历所有题目组，提取待批改题目和题号
for element in elements:
    # 查找待批改和题号俩个元素
    number_element = element.find_element(By.CLASS_NAME, 'subject-number')
    score_element = element.find_element(By.CLASS_NAME, 'card-score')
    # 判断是否同时出现
    if '待批改' in score_element.text and '题号' in number_element.text:
        # 提取题号元素中的数字
        number = int(''.join(filter(str.isdigit, number_element.text)))
        print('待批改的题号是：', number)
    else:
        continue

driver.quit()