python增删改查csv文件_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()