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="660000000066"
                #接收行号
                RECV1234RECV="880000000088"
                #删除正则匹配的一行
                data = re.sub(r'<!--.*-->
', "", 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+'
')
                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读取文件写入新文件