Python+Selenium 自动化实现考后照片核验和分类

import os
import time
import psutil
from selenium.webdriver.chrome.options import Options
from selenium import webdriver
from selenium.webdriver.common.by import By
import openpyxl

# 检查是否已经存在浏览器实例，如果存在则不再创建
for proc in psutil.process_iter():
    try:
        if 'chrome' in proc.name() and '--remote-debugging-port=9224' in proc.cmdline():
            options = Options()
            options.add_experimental_option('debuggerAddress', '127.0.0.1:9224')
            break
    except:
        pass
else:
    # 创建浏览器实例
    os.system(r'start chrome --remote-debugging-port=9224 --user-data-dir="D:\评阅用"')
    options = Options()
    options.add_experimental_option('debuggerAddress', '127.0.0.1:9224')

# 在已有的浏览器实例中查找标签页
driver = webdriver.Chrome(options=options)
tabs = driver.window_handles
for tab in tabs:
    driver.switch_to.window(tab)
    if driver.title == '考后核验':
        print('登陆成功')
        break

# 打开Excel表格
wb = openpyxl.load_workbook(r'C:\Users\Administrator\Desktop\考后核验.xlsx')
sheet = wb.active

# 创建照片文件夹
if not os.path.exists(r'C:\Users\Administrator\Desktop\照片'):
    os.mkdir(r'C:\Users\Administrator\Desktop\照片')

# 遍历时间单元K列, 创建时间文件夹
for row in sheet.iter_rows(min_row=2, max_row=sheet.max_row, min_col=11, max_col=11):
    for cell in row:
        # 获取时间单元内的数据
        data = cell.value
        if data:
            # 创建子文件夹
            folder_name = time.strftime('%Y-%m-%d %H-%M-%S', time.strptime(data, '%Y-%m-%d %H:%M:%S'))
            folder_path = os.path.join(r'C:\Users\Administrator\Desktop\照片', folder_name)
            if not os.path.exists(folder_path):
                os.mkdir(folder_path)

# 遍历表格内'A'列数据
for row in sheet.iter_rows(min_row=2, max_row=sheet.max_row, min_col=1, max_col=1):
    for cell in row:
        # 获取学号
        student_id = cell.value
        if student_id:
            # 在网页内查找考生信息
            search_input = driver.find_element(By.XPATH, '//input[@placeholder="姓名/考生号"]')
            search_input.clear()
            search_input.send_keys(student_id)
            time.sleep(1)  # 等待页面加载完成
            search_button = driver.find_element(By.XPATH, '//button[@class="ant-btn ant-btn-primary"]')
            driver.execute_script('arguments[0].click();', search_button)
            time.sleep(1)  # 等待页面加载完成
            
            # 获取考生信息
            student_info = driver.find_element(By.XPATH, f'//td[contains(text(), "{student_id}")]').text
            student_number = student_info.split(' ')[0]
            student_name = student_info.split(' ')[1]
            
            # 遍历表格内“K”列和“H”列数据
            for time_row, room_row in zip(sheet.iter_rows(min_row=2, max_row=sheet.max_row, min_col=11, max_col=11), 
                                          sheet.iter_rows(min_row=2, max_row=sheet.max_row, min_col=8, max_col=8)):
                for time_cell, room_cell in zip(time_row, room_row):
                    # 获取时间和考场编号
                    time_data = time_cell.value
                    room_data = room_cell.value
                    if time_data and room_data:
                        # 创建对应的子文件夹
                        folder_name = time.strftime('%Y-%m-%d %H-%M-%S', time.strptime(time_data, '%Y-%m-%d %H:%M:%S'))
                        folder_path = os.path.join(r'C:\Users\Administrator\Desktop\照片', folder_name)
                        room_folder_path = os.path.join(folder_path, str(room_data))
                        if not os.path.exists(room_folder_path):
                            os.mkdir(room_folder_path)
                            
                        # 查找对应的照片并保存
                        photo_element = driver.find_element(By.XPATH, f'//img[@title="报名照片" and contains(@src, "{student_number}")]')
                        photo_url = photo_element.get_attribute('src')
                        photo_extension = photo_url.split('.')[-1]
                        photo_name = f'{student_number}_{student_name}.{photo_extension}'
                        photo_path = os.path.join(room_folder_path, photo_name)
                        driver.get(photo_url)
                        time.sleep(1)  # 等待图片加载完成
                        driver.save_screenshot(photo_path)
                        print('保存照片成功:', photo_path)