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 Sheet3100                     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 data116 117 if __name__ == '__main__':118     DataTypeList = ['HMS','BPI']119     GetProjectData(DataTypeList)