如何从RemoteWebDriver服务器而不是本地FirefoxDriver获取屏幕截图？

pip install selenium

pip install pillow

from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.firefox.options import Options
from PIL import Image
import io
import base64

# 设置Firefox浏览器的配置选项
options = Options()
options.headless = True

# 创建一个RemoteWebDriver实例，指定Selenium Grid的URL和浏览器配置选项
driver = webdriver.Remote(
    command_executor='http://localhost:4444/wd/hub',
    desired_capabilities=options.to_capabilities()
)

# 导航到目标网站
driver.get("https://www.example.com")

# 获取屏幕截图并将其转换为base64编码的字符串
screenshot_base64 = driver.get_screenshot_as_base64()

# 将base64编码的字符串解码为图像数据
screenshot_data = base64.b64decode(screenshot_base64)

# 将图像数据保存为PNG格式的文件
with open("screenshot.png", "wb") as f:
    f.write(screenshot_data)

# 关闭RemoteWebDriver实例
driver.quit()