1. 程式人生 > >用python + selenium 做網頁截圖

用python + selenium 做網頁截圖

本意是要把出差的Uber行程記錄從gmail裡匯出來,然後批量截圖用於報銷。
先把批量截圖程式碼貼上來,整理一下這段裡面的點,之後再補下載郵件的程式碼
1、selenium預設不帶chromdriver,需要手動安裝,Windows上的安裝方式
pip install chromdriver

2、安裝chromdriver後需要配置,簡單的做法就是在程式碼中指明chromedriver-Windows.exe這個檔案的絕對路徑
*值得注意的是,我的chromedriver-Windows版安裝後,沒有.exe檔案,需要手動把
C:\Python27\chromedriver-Windows
複製到
C:\Python27\Lib\site-packages\chromedriver\
路徑下,然後手動新增.exe副檔名。
#-*- encoding: utf-8 -*-
#author : yunxch97
# 行程截圖
#CreateDate : 2017-01-16

from selenium import webdriver
import os

#臨時使用
# thisDir="D:/email/uber/"
uberM=100000
i=uberM+1

chromedriver = "C:\Python27\Lib\site-packages\chromedriver\chromedriver-Windows.exe"

browser=webdriver.Chrome(chromedriver)
	
def makeFN(i):
	fn=thisDir+str(i)+".html"
	return fn

# 讀取網頁檔案,寫入csv

while os.path.isfile(makeFN(i)):
	webFN=makeFN(i)
	
	url = "file:///%s"%(webFN)
	browser.set_window_size(720, 1200)
	browser.get(url)

	browser.save_screenshot("%s.png"%(webFN))
	print i
	i+=1

browser.close()