python对文件增删改查

# coding:utf8
__author__ = 'bluesli'



'''
.readline()每只读取文件的一行,
通常也是读取到的一行内容放到一个字符串变量中,
返回str类型,如下图;
2:
.readlines()每次按行读取整个文件内容,
将读取到的内容放到一个列表中,
返回list类型
'''

#把文件内容封装成字典
def convert_to_dict(data_source):
    pass



#获取
def select():
    input_str = input('请输入你要查询的记录')
    f = open('test.conf','r',encoding='utf-8')
    datas = f.readlines()
    for index,line in enumerate(datas):
        if(line.strip()==input_str):
            print('记录已经找到')
            print(datas[index+1].strip())
        print('该条记录是%s,不匹配'%line)
#增加
def add():
    input_str = input("请输入你想增加的记录")
    f = open('test.conf','a',encoding='utf-8')
    f.writelines('\n'+input_str)



#删除
'''
我们可以使用 open() 方法把需要修改的文件打开为两个文件,然后逐行读入内存,找到需要删除的行时,用后面的行逐一覆盖。实现方式见以下代码。
with open('file.txt', 'r') as old_file:
  with open('file.txt', 'r+') as new_file:
 
    current_line = 0
 
    # 定位到需要删除的行
    while current_line < (del_line - 1):
      old_file.readline()
      current_line += 1
 
    # 当前光标在被删除行的行首,记录该位置
    seek_point = old_file.tell()
 
    # 设置光标位置
    new_file.seek(seek_point, 0)
 
    # 读需要删除的行,光标移到下一行行首
    old_file.readline()
     
    # 被删除行的下一行读给 next_line
    next_line = old_file.readline()
 
    # 连续覆盖剩余行,后面所有行上移一行
    while next_line:
      new_file.write(next_line)
      next_line = old_file.readline()
 
    # 写完最后一行后截断文件,因为删除操作,文件整体少了一行,原文件最后一行需要去掉
    new_file.truncate()
'''
def delete():
    input_str = input("请输入你想删除的记录")
    f = open('test.conf', 'r', encoding='utf-8')
    f1 = open('delete_test.conf','a',encoding='utf-8')
    datas = f.readlines()
    for index, line in enumerate(datas):
        if (line.strip() == input_str):
            print('删除。。。')
            continue
        f1.writelines(line)
    f.close()




if __name__ == '__main__':
    print('1:获取\n2:增加\n3:删除')
    input_num = int(input('请输入数字选择你想要的操作'))
    if(input_num==1):
        select()
    if(input_num==2):
        add()
    if(input_num==3):
        delete()

猜你喜欢

转载自www.cnblogs.com/bluesl/p/9079683.html