Python Selenium自动化: 在线考试系统自动打分

from selenium.webdriver.chrome.options import Options
from selenium import webdriver
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

# 循环打分
while True:
    # 查找待批改元素
    while True:
        not_marked_elements = driver.find_elements('class name', 'text ng-binding')
        if len(not_marked_elements) > 0:
            break
        else:
            time.sleep(3)

    # 统计待批改元素对应的题号
    subject_numbers = []
    for element in not_marked_elements:
        row = element.find_element_by_xpath('./../../..')
        column = row.find_element_by_xpath('./td[3]')
        number_element = column.find_element_by_xpath('./span[@class=\'ng-binding ng-scope\']')
        subject_number = int(number_element.text)
        subject_numbers.append(subject_number)

    # 根据题号查找分值元素
    scores = []
    for subject_number in subject_numbers:
        while True:
            elements = driver.find_elements_by_xpath(f'//span[contains(text(), \'题{subject_number}\')]/../following-sibling::td//span[contains(@class, \'summary-sub-title\')]')
            if len(elements) > 0:
                break
            else:
                time.sleep(3)
        score_text = elements[0].text
        score = int(''.join(filter(str.isdigit, score_text)))
        scores.append(score)

    # 随机对2-5个分值减去1-2分
    num_to_reduce = random.randint(2, 5)
    if len(scores) >= num_to_reduce > 0:
        for i in random.sample(range(len(scores)), num_to_reduce):
            scores[i] -= random.randint(1, 2)

    # 打印修改后的分数
    for i, score in enumerate(scores):
        print(f'第{subject_numbers[i]}题分值为：{score}')

        # 判断是否已经填写了答案
        answer = driver.find_element('class name', 'answer-content')
        if answer.text:
            # 填写分数
            print(f'第{subject_numbers[i]}题可以打分')
            score_input = driver.find_elements('css selector', 'input[name=\'score\']')[i]
            score_input.clear()
            score_input.send_keys(str(score))
            time.sleep(1)  # 等待自动保存
            print(f'第{subject_numbers[i]}题已打分：{score}')
    time.sleep(3)
    # 查找下一个学生按钮并点击
    for i in range(3):
        try:
            next_student_button = driver.find_element('css selector', 'span.icon-student-right-narrow.clickable')
            next_student_button.click()
            time.sleep(3)  # 等待页面加载完成
            break
        except:
            time.sleep(3)
            continue
    else:
        print('已到最后一个学生')
        break