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

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网站的加载会以某种方式完成。

有没有一种方法可以指示selenium在捕获屏幕截图之前等待2秒钟？

Answer 1

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