Google表单自动化提交：利用编程实现动态数据填充

姓名,邮箱,选择项
张三,zhangsan@example.com,选项A
李四,lisi@example.com,选项B
王五,wangwu@example.com,选项A

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 = 'https://docs.google.com/forms/d/e/YOUR_FORM_ID/viewform'
# 您的数据文件路径
DATA_FILE = 'data.csv'

def automate_google_form():
    """
    自动化填充并提交Google表单的主函数。
    """
    options = webdriver.ChromeOptions()
    # 可以选择开启无头模式 (不显示浏览器界面，在后台运行)
    # options.add_argument('--headless')
    # options.add_argument('--disable-gpu') # 某些Linux系统需要

    driver = None
    try:
        # 初始化WebDriver
        driver = webdriver.Chrome(executable_path=DRIVER_PATH, options=options)
        print("浏览器已启动。")

        # 加载数据
        try:
            df = pd.read_csv(DATA_FILE)
            print(f"成功加载数据文件：{DATA_FILE}，共 {len(df)} 条记录。")
        except FileNotFoundError:
            print(f"错误：数据文件 '{DATA_FILE}' 未找到。请确保文件存在并位于脚本同一目录下。")
            return

        # 遍历数据并提交表单
        for index, row in df.iterrows():
            print(f"\n--- 正在处理第 {index + 1} 条数据 ---")
            driver.get(FORM_URL) # 每次提交前重新加载表单页面
            wait = WebDriverWait(driver, 15) # 设置最长等待时间为15秒

            try:
                # --- 元素定位与数据填充示例 ---
                # 请根据您的Google表单实际结构修改以下定位器 (By.ID, By.XPATH, By.CSS_SELECTOR 等)
                # 提示：Google表单的输入框ID通常是 'entry_XXXXXXXXX' 形式

                # 示例1：文本输入框 (假设CSV中有一列名为 '姓名')
                # 您需要替换为实际的元素定位器
                name_input = wait.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')))
                name_input.send_keys(row['姓名'])
                print(f"填充姓名: {row['姓名']}")

                # 示例2：另一个文本输入框 (假设CSV中有一列名为 '邮箱')
                email_input = wait.until(EC.presence_of_element_located((By.XPATH, '//*[@id="mG61Hd"]/div[2]/div/div[2]/div[2]/div/div/div[2]/div/div[1]/div/div[1]/input')))
                email_input.send_keys(row['邮箱'])
                print(f"填充邮箱: {row['邮箱']}")

                # 示例3：单选按钮 (假设CSV中有一列名为 '选择项')
                # Google表单的单选按钮通常通过其值来定位
                radio_button_value = row['选择项']
                radio_button_xpath = f"//div[@role='radio' and @data-value='{radio_button_value}']"
                radio_button = wait.until(EC.element_to_be_clickable((By.XPATH, radio_button_xpath)))
                radio_button.click()
                print(f"选择项: {radio_button_value}")

                # 示例4：复选框 (假设CSV中有一列名为 '兴趣'，值为 '编程,阅读')
                # 如果CSV中有多选值，用逗号分隔
                # if '兴趣' in row and pd.notna(row['兴趣']):
                #     checkbox_options = str(row['兴趣']).split(',')
                #     for option in checkbox_options:
                #         checkbox_xpath = f"//div[@role='checkbox' and @data-value='{option.strip()}']"
                #         checkbox = wait.until(EC.element_to_be_clickable((By.XPATH, checkbox_xpath)))
                #         if not checkbox.is_selected(): # 确保只点击未选中的
                #             checkbox.click()
                #         print(f"选择兴趣: {option.strip()}")

                # 示例5：下拉菜单 (需要导入 Select 类)
                # from selenium.webdriver.support.ui import Select
                # select_element = wait.until(EC.presence_of_element_located((By.ID, 'entry_YYYYYYY'))) # 替换为实际ID
                # select = Select(select_element)
                # select.select_by_visible_text(row['下拉问题'])
                # print(f"选择下拉项: {row['下拉问题']}")


                # 提交按钮
                # Google表单的提交按钮通常是一个带有特定文本的 div 元素
                submit_button = wait.until(EC.element_to_be_clickable((By.XPATH, '//div[@role="button" and .//span[text()="提交"]]')))
                submit_button.click()
                print(f"第 {index + 1} 条数据提交成功。")

                # 等待表单提交后的页面加载或跳转，避免过快提交
                time.sleep(3)

            except Exception as e:
                print(f"提交第 {index + 1} 条数据时发生错误: {e}")
                # 可以选择记录错误日志或跳过当前记录
                continue

    except Exception as e:
        print(f"自动化过程中发生全局错误: {e}")
    finally:
        if driver:
            driver.quit()
            print("浏览器已关闭。")
        print("所有数据提交任务完成。")

if __name__ == '__main__':
    automate_google_form()