Python学习笔记五(读取提取写入文件)

#Python打开读取一个文件内容，然后写入一个新的文件中，并对某些字段进行提取，写入新的字段的脚本。这是在工作中，需拼装大量的8543报文，原报文只读，不可写入，且也要对原报文进行提取某字段、新增报文头字段，删除某字段等操作，希望大家有用的着的地方，与大家共同学习进步！

 import os
 import re

 def get_filelist(dir):
     os.getcwd() #get current work direction.
     os.chdir(path) #change direction
     Filelist = []
     for home, dirs, files in os.walk(path):
         for filename in files:
             #文件名列表，包含完整路径
             #Filelist1.append(os.path.join(home, filename))
             #文件名列表，只包含文件名
             Filelist.append(filename)
     return Filelist

 def main():
     n=0
     for file in filelist:
         os.getcwd() #get current work direction.
         os.chdir(path) #change direction
         with open(file,mode='r',encoding='utf-8') as f1:
             data=f1.read()
             f1.close()

             #切换到新的路径下
             os.getcwd() #get current work direction.
             os.chdir(newpath) #change directio

             #读取文件的内容写入到新file文件下
             with open(file,mode='w',encoding='utf-8') as f2:
                 n=n+1
                 ##提取字段 拼装写入##
                 #系统简称大写
                 SYST=file[file.find("")+0:file.find("")+4].upper()
                 #SYST=SYST.upper()
                 #print(SYST)
                 #发送行号
                 SEND1234SEND=""
                 #接收行号
                 RECV1234RECV=""
                 #删除正则匹配的一行
                 data = re.sub(r'<!--.*-->\n', "", data)
                 #报文类型
                 MSGTYPE0MSGTYPE=data[data.find("tech:xsd:")+9:data.find("tech:xsd:")+24]
                 #MsgID
                 MSGRID0000MSGRID=data[data.find("<MsgId>")+7:data.find("<MsgId>")+23]
                 #拼装的报文头
                 head="{H:02"+SEND1234SEND+"  "+SYST+RECV1234RECV+"  "+SYST+"00000000000001XML"+MSGTYPE0MSGTYPE+"     "+MSGRID0000MSGRID+"    "+MSGRID0000MSGRID+"    3D         }"

                 f2.write(head+'\n')
                 f2.write(data)
                 f2.close()

                 if os.path.isfile(file):
                     print(file,"转换完成！")
                 else:
                     print(file,"转换失败！")

     print("共转换个文件数：",n)

 if __name__ == '__main__':
     path='E:\\办公文档\\LR项目资料\\XML格式报文示例'
     newpath='E:\\Python_test\\读取文件\\newfiles'

     #path=input("请输入文件路径：")
     #newpath=input("请输入转换后新文件路径：")

     filelist = get_filelist(dir)
     #print(filelist)
     #不用加引号，如果是多级目录，只判断最后一级目录是否存在
     if os.path.isdir(newpath):
         #print ('dir exists')
         pass
     else:
         #print ('dir not exists')
         os.mkdir(newpath) 

     main()
     

可参考：shell读取文件写入新文件