python处理excel

142536a

　　需求：本人最近做一个project，从log里查找对应文字段，并将之写入excel，并画图，将图保存成jpg文件，如果手工一个个处理会很麻烦，于是，python进入了视线。本人使用python的win32com库来做，最好是先将此过程录制成excel宏，然后根据宏来编写python程序。





  1 # $Revision: #1 $
  2 # $DateTime: 2011/10/13 14:08:58 $
  3 # $Author: yabi.c.chen $
  4 import win32com.client
  5 import re,os,glob,Image
  6 win32c = win32com.client.constants
  7 class GetProjectData:
  8     def __init__(self,DataTypeList = []):
  9         self.app = win32com.client.Dispatch('Excel.Application')
10         self.app.Visible = True
11         self.app.DisplayAlerts = False
12         self.app.AlertBeforeOverwriting = False
13         self.DataTypeList = DataTypeList
14         if self.DataTypeList:
15             for DataType in self.DataTypeList:
16                 self.DataType = DataType
17                 self.IsDrawChart = False
18                 if DataType == 'HMS':
19                     self.State = "WATERFALL"
20                     self.SheetName = ['VBAR_BY_ZONE','AFTER_ADJ_BPIC1','AFTER_ADJ_CLR','BEFORE_ADJ_BPIC2','AFTER_ADJ_BPIC2']
21                     self.pat = "HMS MEASUREMENTS RESULT:.*?HMS MEASUREMENTS RESULT END"
22                     self.IsDrawChart = True
23                 elif DataType == 'BPI':
24                     self.State = "WATERFALL"
25                     self.SheetName = ['VBAR_BY_ZONE','AFTER_ADJ_BPIC1','AFTER_ADJ_BPIC2']
26                     self.pat = "Hd     TPIC_Avg     TPIP_Avg     TPIM_Avg      BPI_Avg     BPIP_Avg     BPIM_Avg.*?EXECUTION INFO:"
27                 else:
28                     continue
29                 try:
30                     self.AddData2Xls()
31                 except:
32                     raise
33                 
34     def __del__(self):
35         self.app.Quit()
36                    
37     def DrawHMSChart(self,workbook,fileName,SerialNum,headCount):
38         AllJPGPath = os.path.dirname(fileName)+os.sep+'all.jpg'
39         Width = 536*(2000/536)
40         Height = 270*(len(self.SheetName)/(2000/536)+1)
41         AllJPG = Image.new('RGB',(Width,Height))
42         AllJPG.paste(0xFFFFFF,(0,0,Width,Height))
43         for index,Sheet in enumerate(workbook.Sheets):
44             workbook.Sheets(Sheet.Name).Activate()
45             workbook.Sheets(Sheet.Name).Cells(1,1).Activate()
46             chart = Sheet.ChartObjects().Add(200,100,400,200).Chart      #chart = workbook.Charts.Add()
47             chart.ChartType = win32c.xlLineMarkers
48             chartTitle = SerialNum + '_' + Sheet.Name
49             chart.ApplyLayout(1)
50             chart.Axes(win32c.xlCategory).TickLabelPosition = win32c.xlLow
51             chart.ChartTitle.Text = chartTitle
52             chart.Axes(win32c.xlValue, win32c.xlPrimary).AxisTitle.Text = "HMS"
53             #chart.Axes(win32c.xlCategory).TickLabelSpacing = 5
54             chart.Axes(win32c.xlValue).MinimumScale = 0
55             chart.Axes(win32c.xlValue).MaximumScale = 8
56             chart.Axes(win32c.xlValue).MajorUnit = 0.5
57             for hd in range(headCount):
58                 series = chart.SeriesCollection().NewSeries()
59                 chart.SeriesCollection(hd+1).Name = "Hd %s" % hd
60                 chart.SeriesCollection(hd+1).Values = "='%s'!$C$%d:$C$%d" % (Sheet.Name,2+24*hd,25+24*hd)
61             chart.Location(Where = win32c.xlLocationAsObject,Name = Sheet.Name)
62             #Save Chart as picture to local
63             JPGFile = os.path.dirname(fileName)+os.sep+chartTitle+'.jpg'
64             Sheet.ChartObjects(1).Chart.Export(JPGFile,'jpg')
65             if index < 3:
66                 AllJPG.paste(Image.open(JPGFile),(index*536,0))
67             else:
68                 AllJPG.paste(Image.open(JPGFile),((index-3)*536,270))
69         AllJPG.save(AllJPGPath)
70         workbook.Save()
71                        
72     def AddData2Xls(self):
73         filelist = glob.glob("*.txt")
74         for myfile in filelist:
75             SN = myfile[:8]
76             myfolder = os.path.splitext(os.path.basename(myfile))[0]
77             if not os.path.isdir(myfolder):
78                 os.mkdir(myfolder)
79             rawdata = self.GetRawData(myfile)
80             workbook = self.app.Workbooks.Add()
81             for index,data in enumerate(rawdata):
82                 worksheet = workbook.Sheets.Add(workbook.Sheets[workbook.Sheets.Count-1])
83                 worksheet.Name = self.SheetName[index]
84                 Data = []
85                 rowIndex = 0
86                 startIndex = int(1 and self.DataType != 'BPI' or 0)
87                 for linedata in data.splitlines()[startIndex:-1]:
88                     rowIndex +=1
89                     line = linedata.strip().split('\t')
90                     for colIndex in range(len(line)):
91                         try:
92                             line[colIndex] = float(line[colIndex])
93                         except:
94                             pass
95                     Data.append(line)
96                 HD =  int(Data[-1][0].strip()[-1]) + 1        #Get Head count
97                 worksheet.Range(worksheet.Cells(1,1),worksheet.Cells(rowIndex,len(line))).Value = Data
98             for sh in workbook.Sheets:
99                 if sh.Name not in self.SheetName:              #Delete Sheet1,Sheet2 and Sheet3
100                     workbook.Sheets(sh.Name).Delete()
101             filename = os.getcwd()+os.sep+myfolder+os.sep+myfolder+'_'+self.DataType+'.xls'
102             workbook.SaveAs(filename)
103             if self.IsDrawChart and self.DataType == 'HMS':
104                 self.DrawHMSChart(workbook,filename,SN,HD)
105             workbook.Close()
106     
107     def GetRawData(self,Filename):
108         f = open(Filename,'r')
109         rawdata = str(f.read())
110         #rawdata = rawdata[rawdata.find('- '*20 + self.State + ' : BEGIN   ' + ' -'*20):rawdata.find('- '*20 + self.State + ' : COMPLETE' + ' -'*20)]
111         rawdatapat = '- '*20 + self.State + ' : BEGIN   ' + ' -'*20 + '(.*?)' + '- '*20 + self.State + ' : COMPLETE' + ' -'*20      
112         rawdata = re.findall(rawdatapat,rawdata,re.DOTALL)
113         data = re.findall(self.pat,rawdata[-1],re.DOTALL)
114         f.close()
115         return data
116
117 if __name__ == '__main__':
118     DataTypeList = ['HMS','BPI']
119     GetProjectData(DataTypeList)