#!/usr/bin/python
import os,xlsxwriter,re

def get_data():
  workbook = xlsxwriter.Workbook('Test.xlsx')           #建立Excel
  worksheet = workbook.add_worksheet()               #添加sheet
  worksheet.write('A1', 'SN')                   #添加列標(biāo)題
  worksheet.write('B1', 'IMEI')
  worksheet.write('C1', 'ICCID')
  files=os.listdir(r'E:\T\工位三（PAS LOG)\PASS')          #獲取目錄中所有l(wèi)og名稱列表
  j= 1
  for i in files:                         #遍歷目錄中的log文件
    xpath = os.path.join('E:\T\工位三（PAS LOG)\PASS',i)      #拼接log文件路徑
    f = open(xpath, 'r',encoding='ISO-8859-1')          #打開log文件
    result = f.readlines()                    #讀取所有l(wèi)og文件內(nèi)容
    f.close()
    for line in result:                     #遍歷log文件內(nèi)容
      if re.search('WriteSN:(.*)', line.strip()):       #正則匹配
        sn=re.search('WriteSN:(.*)',line.strip()).group(1)
        worksheet.write('A' + str(j + 1), sn)        #將匹配到的SN號寫入Excel中
        print(sn)
      if re.search('IMEI:(.*)', line.strip()):
        imei= re.search('IMEI:(.*)', line.strip()).group(1)
        worksheet.write('B' + str(j + 1), imei)
        print(imei)
      if re.search('ICCID:(.*)', line.strip()):
        iccid =re.search('ICCID:(.*)', line.strip()).group(1)
        worksheet.write('C' + str(j + 1), iccid)
        print(iccid)
    j=j+1
  workbook.close()
  
if __name__ == '__main__':
  get_data()