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, 'not-marked')
print('待批改题目数量：', len(elements), '个')

# 逐个点击not-marked元素，并统计出现的题号
for element in elements:
    element.click()
    time.sleep(1)  # 等待页面加载
    subject = driver.find_element(By.CLASS_NAME, 'subject-resort-index')
    print('出现的题号为：', subject.text)

driver.quit()