1 #!/usr/bin/env python 2 #-*- coding:utf-8 -*- 3 4 ####################################################### 5 # 用于批量删除excel的指定行 # 6 # 适用于所有office，前提需要安装pywin32和office软件 # 7 ####################################################### 8 9 import os 10 import sys 11 import time 12 import glob 13 import shutil 14 import string 15 import os.path 16 import traceback 17 import ConfigParser 18 import win32com.client 19 20 SPATH = "" #需处理的excel文件目录 21 DPATH = "" #处理后的excel存放目录 22 23 SKIP_FILE_LIST = [] #需要跳过的文件列表 24 MAX_SHEET_INDEX = 1 #每个excel文件的前几个表需要处理 25 DELETE_ROW_LIST = [] #需要删除的行号 26 27 def dealPath(pathname=''): 28 '''deal with windows file path''' 29 if pathname: 30 pathname = pathname.strip() 31 if pathname: 32 pathname = r'%s'%pathname 33 pathname = string.replace(pathname, r'/', '\') 34 pathname = os.path.abspath(pathname) 35 if pathname.find(":\") == -1: 36 pathname = os.path.join(os.getcwd(), pathname) 37 return pathname 38 39 class EasyExcel(object): 40 '''class of easy to deal with excel''' 41 42 def __init__(self): 43 '''initial excel application''' 44 self.m_filename = '' 45 self.m_exists = False 46 self.m_excel = win32com.client.DispatchEx('Excel.Application') #也可以用Dispatch，前者开启新进程，后者会复用进程中的excel进程 47 self.m_excel.DisplayAlerts = False #覆盖同名文件时不弹出确认框 48 49 def open(self, filename=''): 50 '''open excel file''' 51 if getattr(self, 'm_book', False): 52 self.m_book.Close() 53 self.m_filename = dealPath(filename) or '' 54 self.m_exists = os.path.isfile(self.m_filename) 55 if not self.m_filename or not self.m_exists: 56 self.m_book = self.m_excel.Workbooks.Add() 57 else: 58 self.m_book = self.m_excel.Workbooks.Open(self.m_filename) 59 60 def reset(self): 61 '''reset''' 62 self.m_excel = None 63 self.m_book = None 64 self.m_filename = '' 65 66 def save(self, newfile=''): 67 '''save the excel content''' 68 assert type(newfile) is str, 'filename must be type string' 69 newfile = dealPath(newfile) or self.m_filename 70 if not newfile or (self.m_exists and newfile == self.m_filename): 71 self.m_book.Save() 72 return 73 pathname = os.path.dirname(newfile) 74 if not os.path.isdir(pathname): 75 os.makedirs(pathname) 76 self.m_filename = newfile 77 self.m_book.SaveAs(newfile) 78 79 def close(self): 80 '''close the application''' 81 self.m_book.Close(SaveChanges=1) 82 self.m_excel.Quit() 83 time.sleep(2) 84 self.reset() 85 86 def addSheet(self, sheetname=None): 87 '''add new sheet, the name of sheet can be modify,but the workbook can't ''' 88 sht = self.m_book.Worksheets.Add() 89 sht.Name = sheetname if sheetname else sht.Name 90 return sht 91 92 def getSheet(self, sheet=1): 93 '''get the sheet object by the sheet index''' 94 assert sheet > 0, 'the sheet index must bigger then 0' 95 return self.m_book.Worksheets(sheet) 96 97 def getSheetByName(self, name): 98 '''get the sheet object by the sheet name''' 99 for i in xrange(1, self.getSheetCount()+1): 100 sheet = self.getSheet(i) 101 if name == sheet.Name: 102 return sheet 103 return None 104 105 def getCell(self, sheet=1, row=1, col=1): 106 '''get the cell object''' 107 assert row>0 and col>0, 'the row and column index must bigger then 0' 108 return self.getSheet(sheet).Cells(row, col) 109 110 def getRow(self, sheet=1, row=1): 111 '''get the row object''' 112 assert row>0, 'the row index must bigger then 0' 113 return self.getSheet(sheet).Rows(row) 114 115 def getCol(self, sheet, col): 116 '''get the column object''' 117 assert col>0, 'the column index must bigger then 0' 118 return self.getSheet(sheet).Columns(col) 119 120 def getRange(self, sheet, row1, col1, row2, col2): 121 '''get the range object''' 122 sht = self.getSheet(sheet) 123 return sht.Range(self.getCell(sheet, row1, col1), self.getCell(sheet, row2, col2)) 124 125 def getCellValue(self, sheet, row, col): 126 '''Get value of one cell''' 127 return self.getCell(sheet,row, col).Value 128 129 def setCellValue(self, sheet, row, col, value): 130 '''set value of one cell''' 131 self.getCell(sheet, row, col).Value = value 132 133 def getRowValue(self, sheet, row): 134 '''get the row values''' 135 return self.getRow(sheet, row).Value 136 137 def setRowValue(self, sheet, row, values): 138 '''set the row values''' 139 self.getRow(sheet, row).Value = values 140 141 def getColValue(self, sheet, col): 142 '''get the row values''' 143 return self.getCol(sheet, col).Value 144 145 def setColValue(self, sheet, col, values): 146 '''set the row values''' 147 self.getCol(sheet, col).Value = values 148 149 def getRangeValue(self, sheet, row1, col1, row2, col2): 150 '''return a tuples of tuple)''' 151 return self.getRange(sheet, row1, col1, row2, col2).Value 152 153 def setRangeValue(self, sheet, row1, col1, data): 154 '''set the range values''' 155 row2 = row1 + len(data) - 1 156 col2 = col1 + len(data[0]) - 1 157 range = self.getRange(sheet, row1, col1, row2, col2) 158 range.Clear() 159 range.Value = data 160 161 def getSheetCount(self): 162 '''get the number of sheet''' 163 return self.m_book.Worksheets.Count 164 165 def getMaxRow(self, sheet): 166 '''get the max row number, not the count of used row number''' 167 return self.getSheet(sheet).Rows.Count 168 169 def getMaxCol(self, sheet): 170 '''get the max col number, not the count of used col number''' 171 return self.getSheet(sheet).Columns.Count 172 173 def clearCell(self, sheet, row, col): 174 '''clear the content of the cell''' 175 self.getCell(sheet,row,col).Clear() 176 177 def deleteCell(self, sheet, row, col): 178 '''delete the cell''' 179 self.getCell(sheet, row, col).Delete() 180 181 def clearRow(self, sheet, row): 182 '''clear the content of the row''' 183 self.getRow(sheet, row).Clear() 184 185 def deleteRow(self, sheet, row): 186 '''delete the row''' 187 self.getRow(sheet, row).Delete() 188 189 def clearCol(self, sheet, col): 190 '''clear the col''' 191 self.getCol(sheet, col).Clear() 192 193 def deleteCol(self, sheet, col): 194 '''delete the col''' 195 self.getCol(sheet, col).Delete() 196 197 def clearSheet(self, sheet): 198 '''clear the hole sheet''' 199 self.getSheet(sheet).Clear() 200 201 def deleteSheet(self, sheet): 202 '''delete the hole sheet''' 203 self.getSheet(sheet).Delete() 204 205 def deleteRows(self, sheet, fromRow, count=1): 206 '''delete count rows of the sheet''' 207 maxRow = self.getMaxRow(sheet) 208 maxCol = self.getMaxCol(sheet) 209 endRow = fromRow+count-1 210 if fromRow > maxRow or endRow < 1: 211 return 212 self.getRange(sheet, fromRow, 1, endRow, maxCol).Delete() 213 214 def deleteCols(self, sheet, fromCol, count=1): 215 '''delete count cols of the sheet''' 216 maxRow = self.getMaxRow(sheet) 217 maxCol = self.getMaxCol(sheet) 218 endCol = fromCol + count - 1 219 if fromCol > maxCol or endCol < 1: 220 return 221 self.getRange(sheet, 1, fromCol, maxRow, endCol).Delete() 222 223 224 def echo(msg): 225 '''echo message''' 226 print msg 227 228 def dealSingle(excel, sfile, dfile): 229 '''deal with single excel file''' 230 echo("deal with %s"%sfile) 231 basefile = os.path.basename(sfile) 232 excel.open(sfile) 233 sheetcount = excel.getSheetCount() 234 if not (basefile in SKIP_FILE_LIST or file in SKIP_FILE_LIST): 235 for sheet in range(1, sheetcount+1): 236 if sheet > MAX_SHEET_INDEX: 237 continue 238 reduce = 0 239 for row in DELETE_ROW_LIST: 240 excel.deleteRow(sheet, row-reduce) 241 reduce += 1 242 #excel.deleteRows(sheet, 2, 2) 243 excel.save(dfile) 244 245 def dealExcel(spath, dpath): 246 '''deal with excel files''' 247 start = time.time() 248 #check source path exists or not 249 spath = dealPath(spath) 250 if not os.path.isdir(spath): 251 echo("No this directory :%s"%spath) 252 return 253 #check destination path exists or not 254 dpath = dealPath(dpath) 255 if not os.path.isdir(dpath): 256 os.makedirs(dpath) 257 shutil.rmtree(dpath) 258 #list the excel file 259 filelist = glob.glob(os.path.join(spath, '*.xlsx')) 260 if not filelist: 261 echo('The path of %s has no excel file'%spath) 262 return 263 #deal with excel file 264 excel = EasyExcel() 265 for file in filelist: 266 basefile = os.path.basename(file) 267 destfile = os.path.join(dpath, basefile) 268 dealSingle(excel, file, destfile) 269 echo('Use time:%s'%(time.time()-start)) 270 excel.close() 271 272 def loadConfig(configfile='./config.ini'): 273 '''parse config file''' 274 global SPATH 275 global DPATH 276 global SKIP_FILE_LIST 277 global MAX_SHEET_INDEX 278 global DELETE_ROW_LIST 279 280 file = dealPath(configfile) 281 if not os.path.isfile(file): 282 echo('Can not find the config.ini') 283 return False 284 parser = ConfigParser.ConfigParser() 285 parser.read(file) 286 SPATH = parser.get('pathconfig', 'spath').strip() 287 DPATH = parser.get('pathconfig', 'dpath').strip() 288 filelist = parser.get('otherconfig', 'filelist').strip() 289 index = parser.get('otherconfig', 'maxindex').strip() 290 rowlist = parser.get('otherconfig', 'deleterows').strip() 291 if filelist: 292 SKIP_FILE_LIST = filelist.split(";") 293 if rowlist: 294 DELETE_ROW_LIST = map(int, rowlist.split(";")) 295 MAX_SHEET_INDEX = int(index) if index else MAX_SHEET_INDEX 296 297 298 def main(): 299 '''main function''' 300 loadConfig() 301 if SPATH and DPATH and MAX_SHEET_INDEX: 302 dealExcel(SPATH, DPATH) 303 raw_input("Please press any key to exit!") 304 305 if __name__=="__main__": 306 main()