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:
    not_marked_elements = driver.find_elements('css selector', 'div.status.not-marked.ng-scope')
    if len(not_marked_elements) > 0:
        not_marked_nums = []
        for element in not_marked_elements:
            num_element = element.find_element('css selector', 'div.subject-number span.ng-binding')
            not_marked_nums.append(int(num_element.text))
        print(f'待批改元素个数为：{len(not_marked_elements)}')
        print(f'待批改的题号为：{not_marked_nums}')
        break
    else:
        time.sleep(3)

# 循环打分
while True:
    # 在网页内查找所有class属性为summary-sub-title的元素，并提取其中的数字
    while True:
        elements = driver.find_elements('class name', 'summary-sub-title')
        if len(elements) > 0:
            break
        else:
            time.sleep(3)

    # 获取所有待批改题目的分数元素
    score_elements = [driver.find_elements('css selector', 'input[name=\'score\']')[i - 1] for i in not_marked_nums]

    # 对每个待批改题目进行打分
    for i, score_element in zip(not_marked_nums, score_elements):
        score = int(''.join(filter(str.isdigit, elements[i - 1].text)))
        score -= random.randint(1, 2)
        print(f'第{i}题分值为：{score}')

        # 填写分数
        print(f'第{i}题可以打分')
        score_element.clear()
        score_element.send_keys(str(score))
        time.sleep(1)  # 等待自动保存
        print(f'第{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