在 python 截取屏幕截图之前让 selenium 等待

Make selenium wait before taking a screenshot in python

我做了以下事情:

from selenium import webdriver
#Save a screenshot from spotify.com in current directory
DRIVER = 'chromedriver'
driver = webdriver.Chrome(DRIVER)
driver.get('https://www.youtube.com/watch?v=xzLu9AnuDAw&start=40&hd=0')
driver.save_screenshot('my_screenshot.png')
driver.quit()

问题是当视频仍在该帧(在本例中为 40 秒)加载时截屏。我需要特定时间段的 youtube 屏幕截图。

我知道这种方法会截取网站的屏幕截图,并且在加载视频之前以某种方式完成了 youtube 网站的加载。

有什么方法可以让 selenium 在捕获屏幕截图之前等待,比如 2 秒?

import time
time.sleep(2) # sleep for 2 seconds
driver.save_screenshot('my_screenshot.png')