• python代码
    1. # -*- coding: utf-8 -*-
    2. from selenium import webdriver
    3. from time import sleep
    4. from selenium.webdriver.common.keys import Keys
    5. import os
    7. def useselenium(url):
    8.     if 'HTTP_PROXY'in os.environ: del os.environ['HTTP_PROXY']
    9.     options=webdriver.ChromeOptions()
    10.     options.add_argument('--headless')
    11.     dr =  webdriver.Chrome(chrome_options=options)
    12.     dr.set_page_load_timeout(10)
    13.     dr.set_script_timeout(10)
    14.     try:
    15.         dr.get(url) ## 到达下一个页面
    16.         print "title of current page is %s" %(dr.title) ## 打印Title.
    17.         print "url of current page is %s" %(dr.current_url) ## 打印url.
    18.         dr.get_screenshot_as_file("./myscreen/" + base64.b64encode(filename) +".png")
    19.     except:
    20.         print "Error " + url
    21.     dr.quit()
    23. scan = open("domains.txt","r")
    24. scan_data = [line.strip() for line in scan.readlines()]
    25. scan.close()
    26. for line in scan_data:
    27.     try:
    28.         newline = line.split(",")[0]
    29.         useselenium(newline)
    30.     except:
    31.         pass
    • domains.txt的内容
    1. https://1688.com
    2. https://baidu.com
    • 之后就可以在myscreen目录下看到截图。