【Python学习笔记Day29】5.5 Python文件的使用（二）

【Python学习笔记Day29】 文件——一个任务
任务：将文件（record.txt）中的数据进行分割并按照以下规律保存起来

record.txt:
小客服:小甲鱼，今天有客户问你有没有女朋友？
小甲鱼:咦？？
小客服:我跟她说你有女朋友了！
小甲鱼:。。。。。。
小客服:她让你分手后考虑下她！然后我说:"您要买个优盘，我就帮您留意下~"
小甲鱼:然后呢？
小客服:她买了两个，说发一个货就好~
小甲鱼:呃。。。。。。你真牛！
小客服:那是，谁让我是鱼C最可爱小客服嘛~
小甲鱼:下次有人想调戏你我不阻止~
小客服:滚！！！
================================================================================
小客服:小甲鱼，有个好评很好笑哈。
小甲鱼:哦？
小客服:"有了小甲鱼，以后妈妈再也不用担心我的学习了~"
小甲鱼:哈哈哈，我看到丫，我还发微博了呢~
小客服:嗯嗯，我看了你的微博丫~
小甲鱼:哟西~
小客服:那个有条回复“左手拿著小甲魚，右手拿著打火機，哪裡不會點哪裡，so easy ^_^”
小甲鱼:T_T
================================================================================
小客服:小甲鱼，今天一个会员想找你
小甲鱼:哦？什么事？
小客服:他说你一个学生月薪已经超过12k了！！
小甲鱼:哪里的？
小客服:上海的
小甲鱼:那正常，哪家公司？
小客服:他没说呀。
小甲鱼:哦
小客服:老大，为什么我工资那么低啊？？是时候涨涨工资了！！
小甲鱼:啊，你说什么？我在外边呢，这里好吵吖。。。。。。
小客服:滚！！！

1）小甲鱼的对话单独保存为boy_.txt文件（去掉“小甲鱼：”）
2)小客服的对话单独保存为girl_.txt的文件（去掉小客服：）
3)文件中总共有三段对话，分别保存为boy_1.txt,girl_1.txt,boy_2.txt,girl_2.txt,boy_3.txt,girl_3.txt
共6个文件（文件中不同的对话间已经使用“======”分割）

f = open('record.txt',encoding='utf-8')
print(f.read())
f.close()

f = open('record.txt',encoding='utf-8')
boy = []
girl = []
count = 1

for each_line in f:
    if str(each_line[:6]) != "======":
        (name,content) = each_line.split(':',1) #https://www.cnblogs.com/jsdd/p/11613121.html
        if name == "小甲鱼":
            boy.append(content)
        else:
            girl.append(content)
    else:
        file_boy_name = "boy_" + str(count) + ".txt"
        file_girl_name = "girl_" + str(count) + ".txt"  #文件名

        boy_file = open(file_boy_name,'w')
        girl_file = open(file_girl_name,'w')    #以打开方式创建文件，不存在创建，存在覆盖

        boy_file.writelines(boy)
        girl_file.writelines(girl)  #将每一行写入列表里

        boy_file.close()
        girl_file.close()   #关闭文件
        
        boy = []
        girl = []   #每次初始化列表

        count += 1  #计数加一
        
file_boy_name = "boy_" + str(count) + ".txt"
file_girl_name = "girl_" + str(count) + ".txt"  

boy_file = open(file_boy_name,'w')
girl_file = open(file_girl_name,'w')   

boy_file.writelines(boy)
girl_file.writelines(girl)  

boy_file.close()
girl_file.close()   
        

f.close()

用函数优化程序，把经常出现的用函数封装

def div(count,boy,girl):
    file_boy_name = "boy_" + str(count) + ".txt"
    file_girl_name = "girl_" + str(count) + ".txt"  

    boy_file = open(file_boy_name,'w')
    girl_file = open(file_girl_name,'w')   

    boy_file.writelines(boy)
    girl_file.writelines(girl)  

    boy_file.close()
    girl_file.close()


"""
f = open('record.txt',encoding='utf-8')
boy = []
girl = []
count = 1

for each_line in f:
    if str(each_line[:6]) != "======":
        (name,content) = each_line.split(':',1) #https://www.cnblogs.com/jsdd/p/11613121.html
        if name == "小甲鱼":
            boy.append(content)
        else:
            girl.append(content)
    else:
        div(count,boy,girl)
        count += 1  
        boy = []
        girl = []
        
div(count,boy,girl)  
        

f.close()
"""

可以继续对该程序进程封装：

def split_file(file_name):
    f = open(file_name,encoding='utf-8')
    boy = []
    girl = []
    count = 1

    for each_line in f:
        if str(each_line[:6]) != "======":
            (name,content) = each_line.split(':',1) #https://www.cnblogs.com/jsdd/p/11613121.html
            if name == "小甲鱼":
                boy.append(content)
            else:
                girl.append(content)
        else:
            div(count,boy,girl)
            count += 1  
            boy = []
            girl = []
        
    div(count,boy,girl)  
        
    f.close()

split_file('record.txt')