让硒等待，然后在python中截屏

Question

我已经完成以下工作：

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网站的加载。

有没有一种方法可以指示硒等待2秒钟，然后再捕获屏幕快照？

Answer 1

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