from selenium import webdriver

from selenium.webdriver.remote.webelement import WebElement

from selenium.webdriver.common.action_chains import ActionChains

from selenium.webdriver.chrome.options import Options

import time

def take_screenshot(url, save_fn=”capture.png”):

chrome_options = Options()

chrome_options.add_argument(‘–headless’)

chrome_options.add_argument(‘–disable-gpu’)

path = ‘/Users/Mac/Downloads/chromedriver.exe’

browser = webdriver.Chrome(path,chrome_options=chrome_options) # Get local session of firefox

#browser.set_window_size(1200, 3900)

browser.get(url) # Load page

browser.execute_script(“””

(function () {

var y = 0;

var step = 100;

window.scroll(0, 0);

function f() {

if (y < document.body.scrollHeight) {

y += step;

window.scroll(0, y);

setTimeout(f, 50);

} else {

window.scroll(0, 0);

document.title += “scroll-done”;

}

}

setTimeout(f, 1000);

})();

“””)

for i in xrange(30):

if “scroll-done” in browser.title:

break

time.sleep(1)

size = browser.find_element_by_xpath(“//body”).size

browser.set_window_size(size.get(“width”), size.get(“height”))

browser.save_screenshot(save_fn)

browser.close()

if __name__ == “__main__”:

take_screenshot(“http://www.studclub.cn/”)