【Python】笔记简记--029：关于读写文件的一个应用项目

任务：将文件（record.txt）中的数据进行分割并按照以下规律保存起来：
小甲鱼的对话单独保存为boy_*.txt的文件（去掉“小甲鱼:”）
小客服的对话单独保存为girl_*.txt的文件（去掉“小客服:”）
文件中总共有三段对话，分别保存为boy_1.txt, girl_1.txt，boy_2.txt, girl_2.txt, boy_3.txt, gril_3.txt共6个文件（提示：文件中不同的对话间已经使用“==========”分割）

def save_file(boy,girl,count):
    #文件分别保存
    file_name_boy ='boy_'+str(count)+'.txt'      
    file_name_girl ='girl_'+str(count)+'.txt'
    boy_file = open(file_name_boy,'w')
    girl_file = open(file_name_girl,'w')
    boy_file.writelines(boy)
    girl_file.writelines(girl)
    boy_file.close()
    girl_file.close()


def split_file(file_name):
    f = open('record.txt')
    boy = []
    girl = []
    count = 1
    for each_line in f:
        if each_line[:6]!= '======':
            #这里分割
            (role,line_spoken) = each_line.split(':',1) #字符串分割

            if role == '小甲鱼':
                boy.append(line_spoken)
            if role == '小客服':
                girl.append(line_spoken)
        else:
            save_file(boy,girl,count)

            count+=1
            boy = []
            girl = []
    save_file(boy,girl,count)
    f.close()

split_file('record.txt')