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


def random_score_change(scores):
    # 如果只有一个分值，则对这个分值随机减去1-2分
    if len(scores) == 1:
        score = scores[0]
        score -= random.randint(1, 2)
        scores = [score]
    # 如果大于一个分值，则随机对其中2-5个数字减去1-2分
    else:
        num_changes = random.randint(2, 5)
        indexes = random.sample(range(len(scores)), num_changes)
        for index in indexes:
            score = scores[index]
            score -= random.randint(1, 2)
            scores[index] = score
    return scores


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, 'card-div')
for element in elements:
    # 查找待批改元素
    card_score_element = element.find_element(By.CLASS_NAME, 'card-score')
    if '待批改' not in card_score_element.text:
        continue
    # 查找题号元素
    subject_number_element = element.find_element(By.CLASS_NAME, 'subject-number')
    subject_number_text = subject_number_element.text
    # 提取题号
    subject_number = ""
    for char in subject_number_text:
        if char.isdigit():
            subject_number += char
    print("待批改的题目号：", subject_number)

    # 定位元素并统计数量
    elements = driver.find_elements(By.ID, f'subject-{subject_number}')
    count = len(elements)

    # 查找题目分值
    scores = []
    for element in elements:
        summary_element = element.find_element(By.CLASS_NAME, 'summary-sub-title')
        summary_text = summary_element.text
        summary_number = ""
        for char in summary_text:
            if char.isdigit():
                summary_number += char
        scores.append(int(summary_number))

    # 随机修改分值
    new_scores = random_score_change(scores)
    for i, element in enumerate(elements):
        summary_element = element.find_element(By.CLASS_NAME, 'summary-sub-title')
        summary_element_text = summary_element.text
        summary_element_text = summary_element_text.replace(str(scores[i]), str(new_scores[i]))
        summary_element.click()
        summary_input_element = element.find_element(By.CLASS_NAME, 'el-input__inner')
        summary_input_element.clear()
        summary_input_element.send_keys(summary_element_text)
        save_button_element = element.find_element(By.CLASS_NAME, 'save-button')
        save_button_element.click()
        time.sleep(1)

driver.quit()