标签：
完整代码如下，有时间再总结细节：
#coding=utf-8
import xlrd
import xlwt
from xlutils.copy import copy
from copy import deepcopy
class DataOutput_Helper:
'''It's a data structure about scenario.
there are many methods provioded to make output the excel data easily'''
def __init__(self,project,gvalue,lvalue,ip,mode,paraName,inputP,outputP,formula,scen):
self.__project=project
self.__gvalue=gvalue
self.__lvalue=lvalue
self.__ip=ip
self.__mode=mode
self.__paraName=paraName
self.__inputP=inputP
self.__outputP=outputP
self.__formula=formula
self.__scen=scen
print self.__scen
def get_scen_num(self):
return len(self.__scen.keys())
def get_scen_name(self):
return self.__scen.keys()
def get_rawdata_by_index(self,scenName,index):
return [self.__ip[index],self.__mode[index],self.__paraName[index],self.__inputP[index],self.__outputP[index],self.__formula[index], self.__scen[scenName]['param'][index],self.__scen[scenName]['period'][index]]
def get_data_num(self,scenName):
m=0
for n in range(len(self.__scen[scenName]['param'])):
if self.__scen[scenName]['param'][n]!='' or self.__scen[scenName]['period'][n]!='':
m+=1
return m
def get_project_name(self):
return self.__project
def get_data_by_index(self,scenName,index):
if self.__scen[scenName]['param'][index]=='' and self.__scen[scenName]['period'][index]=='':
return None
elif self.__mode[index]>1:
master=self.__ip[index]+str(self.__mode[index])
elif self.__mode[index]==1:
master=self.__ip[index]
paralist=[]
for n in self.__scen[scenName]['param'][index].split(','):
paralist.append(n.split('='))
formula_re=self.__formula[index]
for n in range(len(paralist)):
formula_re=formula_re.replace(paralist[n][0],str(paralist[n][1]))
for n in range(len(self.__gvalue)):
formula_re=formula_re.replace(self.__gvalue[n][0],str(self.__gvalue[n][1]))
for n in range(len(self.__lvalue)):
formula_re=formula_re.replace(self.__lvalue[n][0],str(self.__lvalue[n][1]))
return [master,formula_re,round(eval(formula_re),2)]
class readexcel:
def __init__(self,filename='scenario.xls'):
self.__filename__=filename
self.__book__=xlrd.open_workbook( filename,formatting_info=True)
self.__sheetnames__=[]#sheet名字
self.__VPparam__=[]
self.__VPperiod__=[]
self.__VRparam__=[]
self.__VRperiod__=[]
self.__gvalue=[]
self.__lvalue=[]
for i in self.__book__.sheet_names():
if i.lower()=='config'or i.lower()=='bw':
sheet=self.__book__.sheet_by_name(i)
for row in range(0,sheet.nrows):
for col in range(0,sheet.ncols):
###-----------读取config参数---------------------------
#-------读取变量定位符project，#key，#value，#formx x(0-~)
if sheet.cell_type(row,col)==xlrd.XL_CELL_TEXT and sheet.cell_value(row,col).lower()=='project':
projectName=sheet.cell_value(row,col+1)
elif sheet.cell_type(row,col)==xlrd.XL_CELL_TEXT and sheet.cell_value(row,col).lower()=='#key':
#--------读取全局参数key-value对
for key_row in range(row+1,sheet.nrows):
if sheet.cell_value(key_row,col)!='':
self.__gvalue.append([sheet.cell_value(key_row,col),sheet.cell_value(key_row,col+1)])
elif sheet.cell_type(row,col)==xlrd.XL_CELL_TEXT and sheet.cell_value(row,col).lower()=='#form':
#--------读取表格参数formx 自动判定宽度和高度
formx_width=1
formx_height=1
for formx_col in range(col,sheet.ncols):
if sheet.cell_value(row+2,formx_col)=='':
formx_width=formx_col-col
break
elif formx_col==sheet.ncols-1:
formx_width=formx_col-col+1
#print 'w',formx_width
for formx_row in range(row+1,sheet.nrows):
if sheet.cell_value(formx_row,col)=='':
formx_height=formx_row-row
break
elif formx_row==sheet.nrows-1:
formx_height=formx_row-row
#print 'h',formx_height
#----------将表格数据读入__formx__
if sheet.cell_value(formx_row,col+2)!='':
self.__lvalue.append([sheet.cell_value(formx_row,col+2),sheet.cell_value(formx_row,col+1)])
###-----------读取BW参数---------------------------
elif sheet.cell_type(row,col)==xlrd.XL_CELL_TEXT and sheet.cell_value(row,col).lower()=='ip':
ip=[]
i=0
for r in range(row+1,sheet.nrows):
if sheet.cell_value(r,col)=='':
ip.append(ip[i-1])
else:
ip.append(sheet.cell_value(r,col))
i+=1
elif sheet.cell_type(row,col)==xlrd.XL_CELL_TEXT and sheet.cell_value(row,col).lower()=='mode':
mode=[]
i=0
for r in range(row+1,sheet.nrows):
if sheet.cell_value(r,col)=='':
mode.append('N/A')
else:
mode.append(sheet.cell_value(r,col))
i+=1
elif sheet.cell_type(row,col)==xlrd.XL_CELL_TEXT and sheet.cell_value(row,col).lower()=='param'and sheet.cell_value(row,col-1)=='Mode':
paramName=[]
i=0
for r in range(row+1,sheet.nrows):
if sheet.cell_value(r,col)=='':
paramName.append('N/A')
else:
paramName.append(sheet.cell_value(r,col).split(','))
i+=1
elif sheet.cell_type(row,col)==xlrd.XL_CELL_TEXT and sheet.cell_value(row,col).lower()=='input':
inputP=[]
i=0
for r in range(row+1,sheet.nrows):
if sheet.cell_value(r,col)=='':
inputP.append('N/A')
else:
inputP.append(sheet.cell_value(r,col))
i+=1
elif sheet.cell_type(row,col)==xlrd.XL_CELL_TEXT and sheet.cell_value(row,col).lower()=='output':
outputP=[]
i=0
for r in range(row+1,sheet.nrows):
if sheet.cell_value(r,col)=='':
outputP.append('N/A')
else:
outputP.append(sheet.cell_value(r,col))
i+=1
elif sheet.cell_type(row,col)==xlrd.XL_CELL_TEXT and sheet.cell_value(row,col).lower()=='bw formula':
formula=[]
i=0
for r in range(row+1,sheet.nrows):
if sheet.cell_value(r,col)=='':
formula.append('N/A')
else:
formula.append(sheet.cell_value(r,col))
i+=1
##***********************************Scen数据结构***************************************************##
## 存储IP的结构： 1.Scen名字 2.参数 3.时间 {'name':{'param':[xx,xx,xx],'period':[xx,xx,xx]},.....}
##Exp:{u'Video Recoding': {'period': [33.0, ], 'param': [u'arg1=3', ]}, }
##
##*************************************************************************************************##
elif sheet.cell_type(row,col)==xlrd.XL_CELL_TEXT and sheet.cell_value(row,col).lower()=='#sce':
scen={}
for c in range(col+1,sheet.ncols):
if sheet.cell_value(row,c)!='':
for c1 in range(c,c+2):
for r in range(row+2,sheet.nrows):
if c1==c:
self.__VPparam__.append(sheet.cell_value(r,c1))
elif c1==c+1:
self.__VPperiod__.append(sheet.cell_value(r,c1))
scen.setdefault(sheet.cell_value(row,c),{}).__setitem__('param',self.__VPparam__[:])
scen.setdefault(sheet.cell_value(row,c),{}).__setitem__('period',self.__VPperiod__[:])
del self.__VPparam__[:]
del self.__VPperiod__[:]
#******************************绘制表格**************************************************************#
self.__Wbook__=copy(self.__book__)
#self.__Wbook__=xlwt.Workbook()
xlwt.add_palette_colour('style1',22)
self.__Wbook__.set_colour_RGB(22,189,183,107)
xlwt.add_palette_colour('style2',23)
self.__Wbook__.set_colour_RGB(23,25,25,112)
xlwt.add_palette_colour('style3',24)
self.__Wbook__.set_colour_RGB(24,240,230,140)
style1 = xlwt.easyxf('pattern: pattern solid, fore_colour style1,back_colour black;'
'font: colour style2, bold True;'
'borders: left 0x0d , right 0x0d, top 0x0d, bottom 0x0d;'
'alignment: horz center,vert center')
style2 = xlwt.easyxf(
'font: colour style2, bold False;'
'borders: left 0x0d,left_colour black , right 0x0d, top 0x0d, bottom 0x0d;'
'alignment: horz center,vert center')
style3 = xlwt.easyxf('pattern: pattern solid, fore_colour style3,back_colour black;'
'font: colour style2, bold True;'
'borders: left 0x0d , right 0x0d, top 0x0d, bottom 0x0d;'
'alignment: horz center,vert center')
bw=DataOutput_Helper(projectName,self.__gvalue,self.__lvalue,ip,mode,paramName,inputP,outputP,formula,scen)
#print bw.get_scen_name()
for n in bw.get_scen_name():#scen
sheet=self.__Wbook__.add_sheet(n,cell_overwrite_ok=True)
#构建表格，构建固定位置内容
sheet.write(0,0,'Scenario\n'+n,style1)
sheet.row(0).height=550
sheet.col(0).width=4000
sheet.col(4).width=4000
sheet.write_merge(0,0,1,6,'BW Requirement',style1)
for k in range(bw.get_data_num(n)):
sheet.row(2+k).height=800
sheet.write(1,1,'Master',style1)
sheet.write_merge(1,1,2,3,'BW(MB/s)',style1)
sheet.write(1,5,'Master',style1)
sheet.write(1,6,'BW(MB/s)',style1)
#print n
total=0
for m in range(bw.get_data_num(n)):#ip number
#print m
master,BW,Result=bw.get_data_by_index(n,m)
print bw.get_data_by_index(n,m)
for l in [1,5]:
sheet.write(2+m,l,master,style3)
sheet.write(2+m,2,BW,style3)
sheet.write(2+m,6,'',style3)
sheet.write(2+m,3,Result,style3)
total=total+Result
sheet.write_merge(1,bw.get_data_num(n)+2,0,0,bw.get_project_name()+'\n estimate \nBW(MB/s)',style2)
sheet.write_merge(1,bw.get_data_num(n)+2,4,4,bw.get_project_name()+'\n simulation \nBW(MB/s)',style2)
#填写Total
for t in [1,5]:
sheet.write(bw.get_data_num(n)+2,t,'Total',style3)
sheet.write(bw.get_data_num(n)+2,t+1,'',style3)
sheet.write(bw.get_data_num(n)+2,3,total,style3)#计算总和
sheet.col(2).width=17000 #BW formula
self.__Wbook__.save(filename)
n=readexcel()Python处理Excel(四)：自定义类处理Excel数据
标签：
原文地址：http://blog.csdn.net/u014234300/article/details/51363576
