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), '组')

# 查找待批改和题号元素
total_score = 0
for element in elements:
    subject_number_element = element.find_element(By.CLASS_NAME, 'subject-number')
    card_score_element = element.find_element(By.CLASS_NAME, 'card-score')
    if '待批改' in card_score_element.text:
        print('题号：', subject_number_element.text, '待批改')
    else:
        try:
            score = int(card_score_element.text)
            total_score += score
            print('题号：', subject_number_element.text, '得分：', score)
        except ValueError:
            print('题号：', subject_number_element.text, '分数格式错误：', card_score_element.text)
print('总分:', total_score)