使用自动化测试框架selenium，批量的进行截图

最近在做一个自动化的电子邮件系统，实现是在outlook客户端能够接受html的邮件。

但是，由于我做的报表使用fusioncharts做的，fusioncharts框架是一个js图表库，所有的图表都是通过javascript来渲染出来的，

由于outlook不能支持js，所以只能将做好的图表页面截图，放入HTML页面中来发送。

截图的代码如下：

from datetime import datetime
import json
import urllib2

from selenium import webdriver
from selenium.common.exceptions import TimeoutException
from selenium.webdriver.support.wait import WebDriverWait

def web_spider(logDebug,url="http://www.baidu.com",enable_proxy=True,proxy_address='http://baidu.com:8080'):
#     logDebug=open(logs_path,'a')
proxy_handler = urllib2.ProxyHandler({"http" : proxy_address})
null_proxy_handler = urllib2.ProxyHandler({})
if enable_proxy:
opener = urllib2.build_opener(proxy_handler)
else:
opener = urllib2.build_opener(null_proxy_handler)
content=""
try:
urllib2.install_opener(opener)
headers = {'User-Agent':'Mozilla/5.0 (Windows; U; Windows NT 6.1; en-US; rv:1.9.1.6) Gecko/20091201 Firefox/3.5.6'}
req = urllib2.Request(url =url ,headers = headers)
#     f=open("email.txt",'w+')
content=urllib2.urlopen(req).read()
content.decode("utf-8",'ignore').encode('utf-8')

#     f.write(content)
#     f.close()
except Exception,e:
logDebug.write("%s:web spider exception,%s\n" %(datetime.now().strftime('%Y%m%d-%H%M%S'),e))

pass
return content

def get_screenshot(browser,logDebug,url,filename='screenshot.png'):
flag=False
try:
browser.get(url)
wait=WebDriverWait(browser,60)
waitFlag=False
try:
waitFlag=wait.until(lambda browser:
'completed' in browser.find_element_by_id('load_flag').get_attribute("value"),
'wait timeout exception')
except TimeoutException,e:
print e
logDebug.write('%s:TimeoutException:%s\n' %(datetime.now().strftime('%Y%m%d-%H%M%S'),e))
print waitFlag
if waitFlag and filename:
browser.save_screenshot(filename)
flag=True
#             browser.get_screenshot_as_file('pngfile.png')
except Exception,e:
print e
logDebug.write("%s:Screenshot Exception,%s\n" %(datetime.now().strftime('%Y%m%d-%H%M%S'),e))
pass
return flag

def screenshot_batch(logs_path,screenshot_url):
logDebug=open(logs_path,'a')
count=0

report_list=[]
try:
content_json=web_spider(logDebug,screenshot_url)
decodejson=json.loads(content_json)
if 'report_list' in decodejson:
report_list=decodejson['report_list']
else:
return
except Exception,e:
print e
logDebug.write("%s:web spider or json decode exception,%s\n" %(datetime.now().strftime('%Y%m%d-%H%M%S'),e))
return count

try:
if decodejson:
browser = webdriver.Firefox()
browser.maximize_window()

for item in report_list:
report_url=item['report_url'].strip() if item['report_url'] else ''
img_path=item['img_path'].strip() if item['img_path'] else ''
screenshot_flag=False
if report_url!='' and img_path!='':
screenshot_flag=get_screenshot(browser,logDebug,report_url,img_path)
if screenshot_flag:
count+=1
browser.close()
browser.quit()

except Exception,e:
print e
logDebug.write("%s:open Firefox batch exception,%s\n" %(datetime.now().strftime('%Y%m%d-%H%M%S'),e))

logDebug.close()
return count