如何实现WebDriver从GUI切换为Headless？Selenium手动登录后切换方案咨询

这是个很实用的需求——毕竟手动登录一次，后续用headless跑自动化流程能省不少事。我有两种靠谱的方案，你可以根据自己的场景选择：

这种方法的核心是让headless模式加载你手动登录过的用户配置文件，因为Profile里不仅保存了登录Cookies，还包含本地存储、会话存储甚至浏览器偏好设置，能完美还原登录状态，适合复杂的登录场景（比如需要记住设备、依赖本地存储Token的应用）。

操作步骤： #

1. 打开Firefox浏览器，在地址栏输入 about:profiles 并回车
2. 点击「创建新配置文件」，给它起个好记的名字（比如AutomationLoginProfile），选择存储路径后完成创建
3. 点击「启动新配置文件」，用这个新打开的浏览器手动完成登录操作，确认登录状态后关闭浏览器
4. 用代码加载这个Profile并启动headless模式

Python代码示例： #

from selenium import webdriver
from selenium.webdriver.firefox.options import Options
from selenium.webdriver.firefox.service import Service

# 配置Firefox选项
firefox_options = Options()
firefox_options.headless = True  # 开启headless模式

# 关键：指定你创建的Profile路径，替换成自己的实际路径
firefox_options.add_argument("-profile")
firefox_options.add_argument("/Users/yourname/Library/Application Support/Firefox/Profiles/xxxxxx.AutomationLoginProfile")

# 启动geckodriver（替换成你的geckodriver路径）
driver_service = Service("/path/to/your/geckodriver")
driver = webdriver.Firefox(service=driver_service, options=firefox_options)

# 现在访问需要登录的页面，已经是登录状态了
driver.get("https://your-app-url.com/upload")

# 后续执行上传自动化流程...

如果不想维护单独的Profile，可以先启动正常浏览器让用户手动登录，然后提取Cookies，再注入到headless浏览器中。这种方法更轻量，但要注意：部分SPA应用会把登录Token存在LocalStorage里，这种情况需要额外处理LocalStorage数据。

操作步骤： #

1. 启动正常Firefox浏览器，引导用户完成登录
2. 提取当前会话的所有Cookies
3. 关闭正常浏览器，启动headless浏览器并注入Cookies
4. 刷新页面恢复登录状态

Python代码示例： #

from selenium import webdriver
from selenium.webdriver.firefox.options import Options
from selenium.webdriver.firefox.service import Service

# 第一步：启动正常浏览器，让用户手动登录
normal_options = Options()
normal_service = Service("/path/to/your/geckodriver")
normal_driver = webdriver.Firefox(service=normal_service, options=normal_options)
normal_driver.get("https://your-app-url.com/login")

# 提示用户完成登录后继续
input("请手动完成登录，确认状态后按回车键继续...")

# 获取登录后的所有Cookies
login_cookies = normal_driver.get_cookies()

# 关闭正常浏览器
normal_driver.quit()

# 第二步：启动headless浏览器并注入Cookies
headless_options = Options()
headless_options.headless = True
headless_service = Service("/path/to/your/geckodriver")
headless_driver = webdriver.Firefox(service=headless_service, options=headless_options)

# 注意：必须先访问目标域名，否则无法添加对应域名的Cookies
headless_driver.get("https://your-app-url.com/login")

# 逐个注入Cookies（部分Cookies可能因属性限制添加失败，加个异常捕获更稳妥）
for cookie in login_cookies:
    try:
        headless_driver.add_cookie(cookie)
    except Exception as e:
        print(f"添加Cookie {cookie.get('name')} 失败: {str(e)}")

# 刷新页面，登录状态即可恢复
headless_driver.refresh()

# 现在可以执行上传流程了
headless_driver.get("https://your-app-url.com/upload")
# ... 你的上传自动化代码 ...

额外注意事项： #

• 确保geckodriver版本和Firefox版本匹配，否则可能出现兼容性问题
• 如果应用依赖LocalStorage存储登录信息，方案2需要额外提取并注入LocalStorage数据（可以用normal_driver.execute_script("return window.localStorage;")获取，再用headless_driver.execute_script()注入）
• 方案1的Profile可以重复使用，下次直接启动headless模式即可，无需再次手动登录

内容的提问来源于stack exchange，提问作者user10312643