自动化Google表单提交：构建数据驱动的程序化解决方案

from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
import pandas as pd
import time

# --- 配置参数 ---
# 替换为您的浏览器驱动路径
# 例如：'C:/path/to/chromedriver.exe' 或 '/usr/local/bin/chromedriver'
DRIVER_PATH = 'path/to/your/chromedriver' 
# 替换为您的Google表单URL
FORM_URL = 'YOUR_GOOGLE_FORM_URL_HERE'
# 您的数据文件路径
DATA_FILE = 'data.xlsx'

def setup_driver():
    """初始化WebDriver并返回"""
    options = webdriver.ChromeOptions()
    # 可以选择开启无头模式，即不显示浏览器界面，提高运行效率
    # options.add_argument('--headless')
    # options.add_argument('--disable-gpu') # 无头模式下建议添加
    # options.add_argument('--no-sandbox') # Linux环境下可能需要

    try:
        driver = webdriver.Chrome(executable_path=DRIVER_PATH, options=options)
        return driver
    except Exception as e:
        print(f"Error initializing WebDriver: {e}")
        print("Please ensure ChromeDriver path is correct and matches your Chrome browser version.")
        exit()

def fill_form_with_data(driver, data_row):
    """
    根据给定的数据行填充Google表单。
    您需要根据您的Google表单实际结构修改这里的元素定位器和填充逻辑。
    """
    try:
        # 导航到表单URL
        driver.get(FORM_URL)
        # 显式等待表单加载完成，例如等待某个关键输入框出现
        WebDriverWait(driver, 10).until(
            EC.presence_of_element_located((By.XPATH, '//*[@id="mG61Hd"]/div[2]/div/div[2]/div[1]/div/div/div[2]/div/div[1]/div/div[1]/input'))
        )

        print(f"开始填充数据：{data_row['姓名']}")

        # 1. 填充文本字段 - 姓名 (示例XPath，请根据实际表单调整)
        # 假设表单中第一个文本输入框是姓名
        name_input_xpath = '//*[@id="mG61Hd"]/div[2]/div/div[2]/div[1]/div/div/div[2]/div/div[1]/div/div[1]/input'
        name_input = driver.find_element(By.XPATH, name_input_xpath)
        name_input.send_keys(data_row['姓名'])
        time.sleep(0.5) # 短暂等待

        # 2. 填充文本字段 - 年龄 (示例XPath)
        # 假设表单中第二个文本输入框是年龄
        age_input_xpath = '//*[@id="mG61Hd"]/div[2]/div/div[2]/div[2]/div/div/div[2]/div/div[1]/div/div[1]/input'
        age_input = driver.find_element(By.XPATH, age_input_xpath)
        age_input.send_keys(str(data_row['年龄'])) # 确保年龄是字符串
        time.sleep(0.5)

        # 3. 填充文本字段 - 邮箱 (示例XPath)
        # 假设表单中第三个文本输入框是邮箱
        email_input_xpath = '//*[@id="mG61Hd"]/div[2]/div/div[2]/div[3]/div/div/div[2]/div/div[1]/div/div[1]/input'
        email_input = driver.find_element(By.XPATH, email_input_xpath)
        email_input.send_keys(data_row['邮箱'])
        time.sleep(0.5)

        # 4. 选择单选按钮 - 满意度 (示例XPath，需要根据实际选项值定位)
        satisfaction_level = data_row['满意度']
        if satisfaction_level == '非常满意':
            # 示例：定位“非常满意”选项的XPath
            driver.find_element(By.XPATH, '//*[@id="mG61Hd"]/div[2]/div/div[2]/div[4]/div/div/div[2]/div[1]/div/span/div/div[1]/label/div/div[1]/div/div[3]/div').click()
        elif satisfaction_level == '满意':
            # 示例：定位“满意”选项的XPath
            driver.find_element(By.XPATH, '//*[@id="mG61Hd"]/div[2]/div/div[2]/div[4]/div/div/div[2]/div[1]/div/span/div/div[2]/label/div/div[1]/div/div[3]/div').click()
        elif satisfaction_level == '一般':
            # 示例：定位“一般”选项的XPath
            driver.find_element(By.XPATH, '//*[@id="mG61Hd"]/div[2]/div/div[2]/div[4]/div/div/div[2]/div[1]/div/span/div/div[3]/label/div/div[1]/div/div[3]/div').click()
        time.sleep(0.5)

        # 5. 填充长文本字段 - 建议 (示例XPath)
        # 假设表单中最后一个文本区域是建议
        suggestion_textarea_xpath = '//*[@id="mG61Hd"]/div[2]/div/div[2]/div[5]/div/div/div[2]/div/div[1]/div[2]/textarea'
        suggestion_textarea = driver.find_element(By.XPATH, suggestion_textarea_xpath)
        suggestion_textarea.send_keys(data_row['建议'])
        time.sleep(0.5)

        # 6. 点击提交按钮 (示例XPath)
        submit_button_xpath = '//*[@id="mG61Hd"]/div[2]/div/div[3]/div[1]/div[1]/div/span/span'
        submit_button = driver.find_element(By.XPATH, submit_button_xpath)
        submit_button.click()

        # 提交后等待一段时间，确保表单提交成功并加载确认页面
        time.sleep(3) 
        print(f"成功提交数据：{data_row['姓名']}")

    except Exception as e:
        print(f"提交数据 {data_row['姓名']} 失败：{e}")
        # 可以在这里添加错误截图或更详细的日志记录
        # driver.save_screenshot(f"error_{data_row['姓名']}.png")

def main():
    driver = setup_driver()

    try:
        # 读取Excel数据
        df = pd.read_excel(DATA_FILE)

        # 遍历每一行数据并提交表单
        for index, row in df.iterrows():
            fill_form_with_data(driver, row)
            time.sleep(2) # 每次提交之间添加间隔，避免过快操作被检测

    except FileNotFoundError:
        print(f"Error: Data file '{DATA_FILE}' not found. Please ensure it exists.")
    except Exception as e:
        print(f"An unexpected error occurred in main: {e}")
    finally:
        if driver:
            driver.quit() # 关闭浏览器
            print("所有数据提交完成，浏览器已关闭。")

if __name__ == "__main__":
    main()