Захват скриншота URL - PullRequest
Новая
       6

Захват скриншота URL

0 голосов
/ 05 июня 2019

Я снимаю скриншот URL через phantomjs.

Я выполнил это с помощью драйвера 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 секунд. Мне нужно обработать 1 миллион URL. Как я могу оптимизировать этот код?

...