pip install selenium
下载 phantomjs 对应平台的文件【当然也可以使用非phantomjs的方式,请自行查询,文末的github库中有对应的文件,觉得官网下载慢的同学可以使用对应的文件】
#! python3
# coding=utf-8
# 作者博客 http://coderknock.com
# 中文为方框的话需要安装字体
# 在centos中执行:yum install bitmap-fonts bitmap-fonts-cjk
# 在ubuntu中执行:sudo apt-get install xfonts-wqy
# 如果要截取 html 文件需要使用 file:///D:/WebstormProjects/ZuiBlog/index.html 类似这样的方式
from selenium import webdriver
import os
import time
import sys
url = sys.argv[1]
execName = 'phantomjs'
if os.name == 'nt':
execName = execName + '.exe'
driver = webdriver.PhantomJS(executable_path='./phantomjs/' + os.name + '/bin/' + execName)
# 设置宽高
driver.set_window_size(1280, 720)
# 这里的executable_path填你phantomJS的路径
driver.get(url)
time.sleep(2)
driver.save_screenshot("shot.png")
driver.quit()