TXTtoExcel

 

 

#-*- coding:utf-8 -*-
import sys
from openpyxl.workbook import Workbook
from openpyxl.cell import get_column_letter

__author__ = 'qiangwang'

def writeExcel(input, excel):
    if input == None or len(input) == 0 or excel == None:
        return
    wb = Workbook()
    ws = wb.get_sheet_by_name(name = wb.get_sheet_names()[0])
    #ws =wb.create_sheet()
    ws.title = 'parser result'

    row = 0
    for items in input:
        print items
        for i in range(0, len(items)):
            column = get_column_letter(i + 1)
            ws.cell(column + str(row + 1)).value = items[i].strip()
        row = row + 1

    wb.save(filename = excel)

#example:  python toexcel.py in.txt out.xls
if __name__ == '__main__':

    if len(sys.argv) < 2:
        print 'give txt file'
    else:
        txt = sys.argv[1]
        if len(sys.argv) > 2:
            excel = sys.argv[2]
        else:
            excel = 'result.xsl'

        input = []
        fp = open(txt, 'r')
        for line in fp.readlines():
            line = line.replace('\n', '')
            input.append(line.split(','))

        fp.close()

        writeExcel(input, excel)

        print 'done'