转自：
http://linkyou.blog.51cto.com/1332494/751799/
最近把测试框架优化了一下，原本生成的报表时txt格式的，现在搞成html格式。主要步骤如下
首先找个html处理工具，这样处理html会简单很多
我网上找了一下发现phy停好用,在这里非常感谢pyh的作者
phy源码如下
# @file: pyh.py
# @purpose: a HTML tag generator
# @author: Emmanuel Turlay <>
__doc__ = """The pyh.py module is the core of the PyH package. PyH lets you
generate HTML tags from within your python code.
See for documentation.
"""
__author__ = "Emmanuel Turlay <>"
__version__ = '$Revision: 43 $'
__date__ = '$Date$'
from sys import _getframe, stdout, modules, version
nOpen={}
nl = '/n'
doctype = '
charset = '/n'
tags = ['html', 'body', 'head', 'link', 'meta', 'div', 'p', 'form', 'legend',
'input', 'select', 'span', 'b', 'i', 'option', 'img', 'script',
'table', 'tr', 'td', 'th', 'h1', 'h2', 'h3', 'h4', 'h5', 'h6',
'fieldset', 'a', 'title', 'body', 'head', 'title', 'script', 'br', 'table','font']
selfClose = ['input', 'img', 'link', 'br']
class Tag(list):
tagname = ''
def __init__(self, *arg, **kw):
self.attributes = kw
if self.tagname :
name = self.tagname
self.isSeq = False
else:
name = 'sequence'
self.isSeq = True
self.id = kw.get('id', name)
#self.extend(arg)
for a in arg: self.addObj(a)
def __iadd__(self, obj):
if isinstance(obj, Tag) and obj.isSeq:
for o in obj: self.addObj(o)
else: self.addObj(obj)
return self
def addObj(self, obj):
if not isinstance(obj, Tag): obj = str(obj)
id=self.setID(obj)
setattr(self, id, obj)
self.append(obj)
def setID(self, obj):
if isinstance(obj, Tag):
id = obj.id
n = len([t for t in self if isinstance(t, Tag) and t.id.startswith(id)])
else:
id = 'content'
n = len([t for t in self if not isinstance(t, Tag)])
if n: id = '%s_%03i' % (id, n)
if isinstance(obj, Tag): obj.id = id
return id
def __add__(self, obj):
if self.tagname: return Tag(self, obj)
self.addObj(obj)
return self
def __lshift__(self, obj):
self += obj
return obj
def render(self):
result = ''
if self.tagname:
result = '<%s%s%s>' % (self.tagname, self.renderAtt(), self.selfClose()*' /')
if not self.selfClose():
for c in self:
if isinstance(c, Tag):
result += c.render()
else: result += c
if self.tagname:
result += '' % self.tagname
result += '/n'
return result
def renderAtt(self):
result = ''
for n, v in self.attributes.iteritems():
if n != 'txt' and n != 'open':
if n == 'cl': n = 'class'
result += ' %s="%s"' % (n, v)
return result
def selfClose(self):
return self.tagname in selfClose
def TagFactory(name):
class f(Tag):
tagname = name
f.__name__ = name
return f
thisModule = modules[__name__]
for t in tags: setattr(thisModule, t, TagFactory(t))
def ValidW3C():
out = a(img(src='', alt='Valid XHTML 1.0 Strict'), href='')
return out
class PyH(Tag):
tagname = 'html'
def __init__(self, name='MyPyHPage'):
self += head()
self += body()
self.attributes = dict(xmlns='', lang='en')
self.head += title(name)
def __iadd__(self, obj):
if isinstance(obj, head) or isinstance(obj, body): self.addObj(obj)
elif isinstance(obj, meta) or isinstance(obj, link): self.head += obj
else:
self.body += obj
id=self.setID(obj)
setattr(self, id, obj)
return self
def addJS(self, *arg):
for f in arg: self.head += script(type='text/javascript', src=f)
def addCSS(self, *arg):
for f in arg: self.head += link(rel='stylesheet', type='text/css', href=f)
def printOut(self,file=''):
if file: f = open(file, 'w')
else: f = stdout
f.write(doctype)
f.write(self.render())
f.flush()
if file: f.close()
class TagCounter:
_count = {}
_lastOpen = []
for t in tags: _count[t] = 0
def __init__(self, name):
self._name = name
def open(self, tag):
if isLegal(tag):
self._count[tag] += 1
self._lastOpen += [tag]
def close(self, tag):
if isLegal(tag) and self._lastOpen[-1] == tag:
self._count[tag] -= 1
self._lastOpen.pop()
else:
print 'Cross tagging is wrong'
def isAllowed(self, tag, open):
if not open and self.isClosed(tag):
print 'TRYING TO CLOSE NON-OPEN TAG: %s' % tag
return False
return True
def isOpen(self, tag):
if isLegal(tag): return self._count[tag]
def isClosed(self, tag):
if isLegal(tag): return not self._count[tag]
def isLegal(tag):
if tag in tags: return True
else:
print 'ILLEGAL TAG: %s' % tag
return False
第二步，写和DB交互的代码，通常写报表都需要从数据库或其他数据元获取数据。我的比较简单，源代码如下
#coding=utf-8
import loadmylib
import dbunit_mysql
from pyh import *
class dbreport:
def __init__(self,dbinfo='report',project='aliuid',suitid=None):
self.db = dbunit_mysql.DBUnit(dbinfo,project)
self.suitid = suitid
if self.suitid <= 0:
print "suitid must not be none"
return
#
def get_taskresult_records(self):
try:
suitresult = self.db.read("SELECT * FROM ex_suit e where suitid =%d"%self.suitid)
taskid = suitresult[0]['taskid']
taskresult = self.db.read("select * from task where taskid = %d"%int(taskid))
return taskresult[0]
except:
print "getsuitreport error"
def get_groupresult_records(self):
try:
groupresult = self.db.read("SELECT * FROM ex_group where suitid =%d"%self.suitid)
return groupresult
except:
print "getgroupreport error"
def get_failedcase_records(self):
try:
sqlstr = "SELECT e.*,f.testgroupname FROM ex_case e left join ex_group f on e.testgroupid = f.testgroupid where f.suitid=%d and e.status =1 order by testcaseid"%self.suitid
groupresult = self.db.read(sqlstr)
return groupresult
except:
print "getgroupreport error"
def get_totalcase_records(self):
try:
sqlstr = "SELECT e.*,f.testgroupname FROM ex_case e left join ex_group f on e.testgroupid = f.testgroupid where f.suitid=%d order by testcaseid"%self.suitid
groupresult = self.db.read(sqlstr)
return groupresult
except:
print "getgroupreport error"
def getcaseresult():
mytab.attributes[''] = '';
pass
if __name__ == '__main__':
dbrp = dbreport(suitid=12)
result = dbrp.get_failedcase_records()
result1 = dbrp.get_totalcase_records()
print len(result)
print result
print result1
第三步，自然是生成报表的代码了。源代码如下
#coding=utf-8
import loadmylib
import dbunit_mysql
from pyh import *
import dbreport
import commonfunc
import config
class report:
def __init__(self,dbinfo='report',project='aliuid',suitid=None):
self.page = PyH('%s test report'%project)
self.page << ''
self.dbinfo = dbinfo
self.project = project
self.drp = dbreport.dbreport(dbinfo,project,suitid)
#设定表格的格式
def tablecss(self,table = None,width='90%'):
table.attributes['cellSpacing'] = 1;
table.attributes['cellPadding'] = 1;
table.attributes['border'] = 1;
table.attributes['borderColor'] = '#666666';
table.attributes['width'] = width;
#set colum title bgcolor
def tr_title_css(self,tr = None):
tr.attributes['bgcolor'] = '#CCCC00';
#get summary info
def gentask_html(self):
#get taskresult from mysqldb
taskresult = self.drp.get_taskresult_records()
sum_div = div(b('.摘要'), id='myCSSid1')
self.page << sum_div
interfaces = taskresult['modules']
num_interface = len(interfaces.split(','))
totalcase = int(taskresult['totalcase'])
failedcase = int(taskresult['failedcase'])
successcase = totalcase - failedcase
tab_task = self.page << table()
self.tablecss(tab_task)
title_tr_task = tab_task << tr()
self.tr_title_css(title_tr_task)
#add task column title
title_tr_task << td('测试模块数')
title_tr_task << td('通过用例数')
title_tr_task << td('失败用例数')
title_tr_task << td('总共用例数')
#add column value
value_tr_task = tab_task << tr()
value_tr_task << td('%d'%num_interface)
value_tr_task << td('%d'%successcase)
value_tr_task << td('%d'%failedcase)
value_tr_task << td('%d'%totalcase)
self.page.printOut('test.html')
self.page << br()
def gengroupreport(self):
testgroups = self.drp.get_groupresult_records()
sum_div = div(b('.模块测试报告'), id='mydiv2')
self.page << sum_div
tab_group = self.page << table()
self.tablecss(tab_group)
title_tr_group = tab_group << tr()
self.tr_title_css(title_tr_group)
#add task column title
title_tr_group << td('模块名称')
title_tr_group << td('测试结果')
title_tr_group << td('通过用例数')
title_tr_group << td('失败用例数')
title_tr_group << td('总共用例数')
for i in range(len(testgroups)):
testgroup = testgroups[i]
groupname = testgroup['testgroupname']
groupstatus = int(testgroup['status'])
totalcase = int(testgroup['totalcase'])
failedcase = int(testgroup['failedcase'])
successcase = totalcase - failedcase
#add column value
value_tr_group = tab_group << tr()
value_tr_group << td('%s'%groupname)
if groupstatus == 0:
value_tr_group << td('Pass')
else:
value_tr_group << td('Failed')
value_tr_group << td('%d'%successcase)
value_tr_group << td('%d'%failedcase)
value_tr_group << td('%d'%totalcase)
self.page << br()
def gen_failedcase_report(self):
failedcases = self.drp.get_failedcase_records()
sum_div = div(b('.失败用例列表'), id='mydiv2')
self.page << sum_div
tab_failed_case = self.page << table()
self.tablecss(tab_failed_case)
title_tr_failed_case = tab_failed_case << tr()
self.tr_title_css(title_tr_failed_case)
#add task column title
title_tr_failed_case << td('用例编号')
title_tr_failed_case << td('模块名称')
title_tr_failed_case << td('用例说明')
title_tr_failed_case << td('用例等级')
title_tr_failed_case << td('测试结果')
for i in range(len(failedcases)):
testcase = failedcases[i]
casecode = testcase['testcasename']
groupname = testcase['testgroupname']
description = testcase['description']
level = testcase['caselevel']
status = int(testcase['status'])
#add column value
value_tr_failed_case = tab_failed_case << tr()
value_tr_failed_case << td('%s'%casecode)
value_tr_failed_case << td('%s'%groupname)
value_tr_failed_case << td('%s'%description.encode('utf-8'))
value_tr_failed_case << td('%s'%level)
if status == 0:
value_tr_failed_case << td('Pass')
else:
value_tr_failed_case << td('Failed')
self.page << br()
def gen_totalcase_report(self):
totalcases = self.drp.get_totalcase_records()
sum_div = div(b('.所有用例列表'), id='mydiv4')
self.page << sum_div
tab_total_case = self.page << table()
self.tablecss(tab_total_case)
title_tr_total_case = tab_total_case << tr()
self.tr_title_css(title_tr_total_case)
#add task column title
title_tr_total_case << td('用例编号')
title_tr_total_case << td('模块名称')
title_tr_total_case << td('用例说明')
title_tr_total_case << td('用例等级')
title_tr_total_case << td('测试结果')
for i in range(len(totalcases)):
testcase = totalcases[i]
casecode = testcase['testcasename']
groupname = testcase['testgroupname']
description = testcase['description']
level = testcase['caselevel']
status = int(testcase['status'])
#add column value
value_tr_total_case = tab_total_case << tr()
value_tr_total_case << td('%s'%casecode)
value_tr_total_case << td('%s'%groupname)
value_tr_total_case << td('%s'%description.encode('utf-8'))
value_tr_total_case << td('%s'%level)
if status == 0:
value_tr_total_case << td('Pass')
else:
value_tr_total_case << td('Failed')
self.page << br()
def getdev_build_info(self,section='build_dev'):
import os
build_dev = section
self.build_dev_info = config.SysConfig(self.project).getdbconnectinfo(build_dev)
host = self.build_dev_info['host']
user = self.build_dev_info['user']
pwd = self.build_dev_info['password']
source = self.build_dev_info['source']
result = commonfunc.ssh_cmd(host,user,pwd,"svn info %s"%source)
lresult = result.split("/r/n")
url = lresult[2]
reversion = lresult[5]
last_author = lresult[8]
last_date = lresult[10]
return lresult
def genbuildreport(self):
dev_info = self.getdev_build_info('build_dev')
test_info = self.getdev_build_info('build_test')
build_div = div(b('.版本信息'), id='mydiv5')
self.page << build_div
tab_build = self.page << table()
self.tablecss(tab_build,'70%')
title_tr_build = tab_build << tr()
self.tr_title_css(title_tr_build)
#add task column title
title_td = title_tr_build << td('开发代码版本信息')
title_td.attributes['colspan'] = '1';
#add column value
value_tr_build = tab_build << tr()
td_dev = value_tr_build << td()
#td_dev.attributes['font'] = 'size =1'
td_dev = td_dev << font()
td_dev.attributes['size'] = '2'
pt = td_dev << p()
pt << dev_info[2].strip()
pt <
pt << dev_info[5].strip()
pt <
pt <
pt <
pt <
title_tr_build = tab_build << tr()
self.tr_title_css(title_tr_build)
#add task column title
title_td = title_tr_build << td('测试代码版本信息')
title_td.attributes['colspan'] = '1';
#add column value
value_tr_build = tab_build << tr()
td_test = value_tr_build << td()
td_test = td_test << font()
td_test.attributes['size'] = '2'
pt = td_test << p()
pt << test_info[2].strip()
pt <
pt << test_info[5].strip()
pt <
pt << test_info[8].strip()
pt <
pt << test_info[10].strip()
pt <
self.page << br()
def genreport(self,filename='test.html'):
self.genbuildreport()
self.gentask_html()
self.gengroupreport()
self.gen_failedcase_report()
self.gen_totalcase_report()
self.page.printOut('test.html')
if __name__ == '__main__':
file = 'first.html'
rp = report(suitid=12)
rp.genreport(file)
