python-3.x - 捕获 url 的屏幕截图

Question

我正在通过 phantomjs 捕获 URL 的屏幕截图。

我使用 phantomjs 驱动程序执行它，但捕获一个屏幕截图大约需要 2 秒。

driver = webdriver.PhantomJS('path/to/phantomjs.exe')
def driver_initialize(width,height):
  driver.set_script_timeout(30)
  if width and height:
        driver.set_window_size(width, height)


def driver_screenshot(urls):
  green_color=[]
  for i in urls:
    print(i)
    filename = i.split("/")[4] + ".png"
    screen_path =  os.path.join(path, filename)
    images.append(screen_path)
    driver.get(i)
    driver.save_screenshot(screen_path)


def driver_quit():
  driver.quit()

driver_initialize(1024,768)
driver_screenshot(links)
driver_quit()

使用此代码，每个屏幕截图捕获和保存过程大约需要 3-4 秒。我需要处理 100 万个网址。如何优化此代码？

标签： python-3.xseleniumphantomjs