python文件读写操作

对文件操作的流程：

1、打开文件，得到文件句柄

2、通过句柄对文件进行操作

3、关闭文件

我们拿/etc/passwd文件进行测试

1、打开文件，"f"变量得到的是文件句柄

>>> f = open("/tmp/passwd","r",encoding="utf-8")       //“r”为只读模式

>>> data = f.read()

>>> print(data)

root:x:0:0:root:/root:/bin/bash

bin:x:1:1:bin:/bin:/sbin/nologin

daemon:x:2:2:daemon:/sbin:/sbin/nologin

adm:x:3:4:adm:/var/adm:/sbin/nologin

以上输出没有问题，那么我们此时再读一次文件内容呢？看一下效果

>>> data2 = f.read()

>>> print(data2)

>>>                //结果是什么都没有输出，原因是：文件指针跑到了最后，再继续读的时候光标是不重新开始的

以上是一次性全部读出文件的内容，那么要想读出文件的前5行怎么办？

>>> f = open("/tmp/passwd","r",encoding="utf-8")

>>> for i in range(5):

...     print(f.readline().strip())         //readline()为读一行

... 

root:x:0:0:root:/root:/bin/bash

bin:x:1:1:bin:/bin:/sbin/nologin

daemon:x:2:2:daemon:/sbin:/sbin/nologin

adm:x:3:4:adm:/var/adm:/sbin/nologin

lp:x:4:7:lp:/var/spool/lpd:/sbin/nologin

还有一个要求，就是文件的第4行不打印，插入一行标识符（两种办法）

>>> f = open("/tmp/passwd","r",encoding="utf-8")

>>> for index,line in enumerate(f.readlines()):             //enumerate可以取列表的下标，readlines()可以把读取的文件转换成一个列表，每一行为一个元素

...     if index == 3:

...         print("##########################")

...         continue

...     print(line.strip())

... 

root:x:0:0:root:/root:/bin/bash

bin:x:1:1:bin:/bin:/sbin/nologin

daemon:x:2:2:daemon:/sbin:/sbin/nologin

##########################

lp:x:4:7:lp:/var/spool/lpd:/sbin/nologin

sync:x:5:0:sync:/sbin:/bin/sync

shutdown:x:6:0:shutdown:/sbin:/sbin/shutdown     //这种方法比较低效，如果要读大文件不要使用此方法

接下来看一下第二种高效方式：

>>> f = open("/tmp/passwd","r",encoding="utf-8")

>>> count = 0

>>> for line in f:

...   if count == 3:

...     print("################")

...     count += 1

...     continue

...   print(line)

...   count += 1

2、对文件进行写操作

>>> f = open("/tmp/passwd","w",encoding="utf-8")   //"w"为写模式，注意：这是重新创建此文件，会清空源文件的所有内容，注意使用场景

>>> f.write("我是小明")

4

>>> f.write("我是小绿")

4

>>> f.write("我是小绿\n")            //"\n"为换行符

5

>>> f.write("我是小蓝")

5

>>> f.close()                    //关闭文件

>>> f = open("/tmp/passwd","r",encoding="utf-8")      //再次以读模式打开的时候会发现原有的内容已经没有了，只有现在新添加的新内容

>>> data = f.read()

>>> print(data)

我是小明我是小绿我是小绿

我是小蓝

3、文件指针

>>>print(f.tell())              //刚打开文件的时候文件指针是在开始的位置，所以打印"0"

0

>>>print(f.readline())

root:x:0:0:root:/root:/bin/bash

>>>print(f.tell())                 //当读取文件后再打印文件指针，已经从0移动到了当前位置

33

如果想把光标移动回去的话，应该用seek

>>>f.seek(0)
>>>print(f.tell())

0

4、文件内容的截断

>>>f = open("passwd","r+",encoding="utf-8")                //"r+"是读写权限，"rb"是读二进制文件，比如视频文件
>>>f.truncate(20)                                                               //truncate为截断文件，这段代码的意思是阶段到文件的第二十个字符为止
>>>print(f.read())

root:x:0:0:root:/roo

5、文件的修改

import os

f = open("passwd","r",encoding="utf-8")                                 //

f_new = open("passwd.bak","w",encoding="utf-8")

for line in f:

    if "root" in line:
        line = line.replace("root","xiaoming")
    f_new.write(line)
f.close()
f_new.close()
os.remove("passwd")                                  //删除原来的文件
os.rename("passwd.bak","passwd")            //把修改过的文件进行重命名

6、实现一个简单的文件修改的小脚本（只当练习使用，如果在工作中用，还需完善）

import os

import sys

file_name,old_line,new_line = sys.argv[1],sys.argv[2],sys.argv[3]

f = open(file_name,"r",encoding="utf-8")

f_new = open(file_name+".bak","w",encoding="utf-8")

for line in f:

  if old_line in line:

    line = line.replace(old_line,new_line)

  f_new.write(line)

f.close()

f_new.close()

os.remove(file_name)

os.rename(file_name+".bak",file_name)