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

    # 查找题目分值
    summary_numbers = []
    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
        summary_numbers.append(int(summary_number))
        print(f'第{subject_number}题的分值为：{summary_number}')

    # 随机减分
    if len(summary_numbers) <= 1:
        random_index = random.randint(0, len(summary_numbers) - 1)
        random_score = random.randint(1, 2)
        summary_numbers[random_index] -= random_score
        print(f'第{subject_number}题的分值随机减少{random_score}分')
    else:
        random_count = random.randint(2, 5)
        random_indexes = random.sample(range(len(summary_numbers)), random_count)
        for index in random_indexes:
            random_score = random.randint(1, 2)
            summary_numbers[index] -= random_score
            print(f'第{subject_number}题第{index+1}个分值随机减少{random_score}分')

    # 更新分值
    for i, element in enumerate(elements):
        summary_element = element.find_element(By.CLASS_NAME, 'summary-sub-title')
        summary_text = summary_element.text
        summary_text = summary_text.replace(str(summary_numbers[i]), str(summary_numbers[i]-1))
        summary_element.clear()
        summary_element.send_keys(summary_text)

    # 提交批改
    submit_button = element.find_element(By.CLASS_NAME, 'submit-btn')
    submit_button.click()
    time.sleep(1)
    submit_confirm_button = driver.find_element(By.CLASS_NAME, 'layui-layer-btn0')
    submit_confirm_button.click()
    print(f'第{subject_number}题批改完成')

driver.quit()