python入门6——读取文档并修改新建

def save_file(dj,lili,count):
        file_name_dj = 'dj'+str(count)+'.txt'
        file_name_lili = 'lili'+str(count)+'.txt'
        #建立两个文件名
        dj_file=open(file_name_dj,'w')
        lili_file=open(file_name_lili,'w')
        #打开文件
        dj_file.writelines(dj)
        lili_file.writelines(lili)
        #向文件中写入
        dj_file.close()
        lili_file.close()
        #关闭文件
#这个函数，用来新建文本文档，count代表文件的个数
def split_file(file_name):
        f=open('record.txt')
        dj=[]
        lili=[]
        count = 1
        for each_line in f:
            if each_line[:6] != '======':
                #找关键字分隔
                (role,line_spoken)=each_line.split(':',1)
                #调用split函数分割成两部分
                if role == 'dj':
                    dj.append(line_spoken)
                if role == 'lili':
                    lili.append(line_spoken)
                #讲分割后的内容写入数组
            else :
                save_file(dj,lili,count)
                #调用上面的函数存盘
                dj=[]
                lili=[]
                count +=1
                #存完一次，就初始化dj，lili，数组，继续下一个文档
        save_file(dj,lili,count)
        #这是由于最后一个文本无分隔符
        f.close()
#这个函数用来分离修改文本文档 ，同时保存
split_file('record')