Python写入Excel文件的简单代码

import os
import sys
import re
import xlwt
import time

workbook = xlwt.Workbook()
sheet1 = workbook.add_sheet('sheet1', cell_overwrite_ok=True)

if __name__ == '__main__':
    if len(sys.argv) < 2:
        print ('Missing paramaters. Format should be: logtoexcel.py [logFileName]')
        sys.exit(1)
    else:
        logFileName = sys.argv[1]

    try:
        fp = open(logFileName, 'r')
    except Exception as err:
        print("File not exists:" + str(err))
        sys.exit(1)

    row = 0

    for fstr in fp.readlines():
        result1 = fstr[0:fstr.find(':')]

        result2 = re.search('\d{0,2}:\d{0,2}:\d{0,2}', fstr)
        if result2 != None:
            col = 0
            # 写入数据write(行,列,value)
            sheet1.write(row, col, result1)
            col = col + 1
            sheet1.write(row, col, result2.group())
            row = row + 1

    fp.close()

    xlsFileName = logFileName[0:logFileName.rfind('.')]

    if len(sys.argv) < 3:
        workbook.save('./' + xlsFileName + '.xls')
    else:
        workbook.save(sys.argv[2])

    print(logFileName)