如何拍摄selenium中发生的事情的视频

import cv2
import numpy as np
from selenium import webdriver

# 创建一个浏览器实例
driver = webdriver.Chrome()

# 打开网页或执行其他操作

# 获取屏幕分辨率
screen_size = driver.execute_script("return [window.screen.availWidth, window.screen.availHeight];")

# 创建一个VideoWriter对象，用于保存视频
fourcc = cv2.VideoWriter_fourcc(*"XVID")
video_writer = cv2.VideoWriter("selenium_video.avi", fourcc, 20.0, (screen_size[0], screen_size[1]))

while True:
    # 获取屏幕截图
    screenshot = driver.get_screenshot_as_png()
    
    # 将截图转换为OpenCV图像格式
    img_np = np.frombuffer(screenshot, np.uint8)
    img = cv2.imdecode(img_np, cv2.IMREAD_COLOR)
    
    # 将图像写入视频文件
    video_writer.write(img)
    
    # 按下'q'键退出循环
    if cv2.waitKey(1) == ord('q'):
        break

# 释放资源
video_writer.release()
cv2.destroyAllWindows()
driver.quit()

ffmpeg -f x11grab -s 1366x768 -i :0.0 -r 25 -c:v libx264 -preset ultrafast selenium_video.mp4