自写文件读写及修改

#！/usr/bin/env python
# -*- coding:utf-8 -*-
#Author:QiFeng Zhang
import time
#修改文件内容

aftertxt = [] #修改后文件内容
aftertxt1 = []
temptxt  = [] #存放临时数据
def add(vpath,vtent):  #增加文件内容 1.要修改的文件，2.要增加的内容
    oldfile = open(vpath,'r',encoding='utf-8')
    flag_mod = '1' #控制是否将读取数据存入临时数组
    for line in oldfile:
        #print(line.strip())
        if vtent in line:  #控制读取哪段数据存入临时数组中
            flag_mod = '2'
        elif '[' in line.strip()  and flag_mod == '2':
            flag_mod = '3'
        #判断应该存入哪个数组中
        if flag_mod == '1':
            aftertxt.append(line)
        elif flag_mod == '2':
            aftertxt.append(line)
            if line.strip() == '['+vtent+']':
                temptxt.append(line.replace(vtent,vtent+"new"))
            else:
                temptxt.append(line)
        else:
            aftertxt1.append(line)
    oldfile.close()
    newfile = open("zyadd.ini",'w',encoding='utf-8') #以写模式新建文件
    aftertxt.extend(temptxt)  #创建新文件数组
    aftertxt.extend(aftertxt1)  #创建新文件数组
    newfile.writelines(aftertxt) #写文件
    newfile.close()
def minus(vpath,vtent): #删除文件内容 1.要修改的文件，2.要增加的内容
    oldfile = open(vpath,'r',encoding='utf-8')
    flag_mod = '1' #控制是否将读取数据存入临时数组
    for line in oldfile:
        #print(line.strip())
        if vtent in line:  #控制读取哪段数据存入临时数组中
            flag_mod = '2'
        elif '[' in line.strip()  and flag_mod == '2':
            flag_mod = '3'
        #判断应该存入哪个数组中
        if flag_mod == '1':
            aftertxt.append(line)
        elif flag_mod == '2':    #删除部分舍去
            pass
        #     aftertxt.append(line)
        #     if line.strip() == '[mycalc]':
        #         temptxt.append(line.replace("mycalc","mycalcnew"))
        #     else:
        #         temptxt.append(line)
        else:
            aftertxt1.append(line)
    oldfile.close()
    newfile = open("zyminus.ini",'w',encoding='utf-8')
    aftertxt.extend(temptxt)
    aftertxt.extend(aftertxt1)
    newfile.writelines(aftertxt)
    newfile.close()
def modify(vpath,vtent,vnew): #修改文件内容 1.要修改的文件，2.要增加的内容
    oldfile = open(vpath,'r',encoding='utf-8')
    newfile = open('zymodify.ini','w',encoding='utf-8')
    for line in oldfile:
        #print(line.strip())
        if vtent in line:  #控制读取哪段数据存入临时数组中
            newfile.write(line.replace(vtent,vnew))
            print('2>>>>', line.replace(vtent,vnew))
        else:
            newfile.write(line)
            print('1>>>>',line)
            newfile.flush()
    oldfile.close()
    newfile.close()

# oldfile = open("zy.ini",'r',encoding='utf-8')
# for line in oldfile:
#     print(line.strip())
add("zy.ini",'mycalc')
minus("zy.ini",'mycalc')
modify("zy.ini",'mycalc','mycalc这是个鬼')