python - 使用 Selenium 为不停止加载的页面拍摄快照

Question

我正在使用 Selenium 来捕获网页的屏幕截图。它在像 stackoverflow 这样的网站上运行得很好，但我试图在一个永远不会停止加载的页面上使用它。有没有办法在 x 秒后抓取屏幕截图，无论是否完成？

当前代码:

import os
from selenium import webdriver


def main():
    driver = webdriver.Chrome()


    with open('test.txt', 'r') as f:
        for url in f.readlines():

            driver.get('http://' + url)

            sn_name = os.path.join('Screenshots', url.strip().replace('/', '-') + '.png')

            print('Attempting to save:', sn_name)
            if not driver.save_screenshot(sn_name):
                raise Exception('Could not save screen shot: ' + sn_name)

    driver.quit()


if __name__ == '__main__':
    main()

Answer 1

我认为事情不是这样的。
Webdriver 将隐式等待页面加载直到超时。
它应该给你一个超时异常。

我认为你应该使用 try-except 来捕获它，然后截图。
否则，您应该进行多线程编程，让另一个线程进行截图。