1、引言

小鱼：小屌丝，这段代码为什么要开两个线程？

小屌丝：因为我要读写文件，还要备份文件，所以就开两个线程了。

小鱼：嗯，想法是对的，但是，还有一种简便的方法，不需要开两个线程就能搞得定的。

小屌丝：额…难道是with open?

小鱼：不是。

小屌丝：那还有啥呢？我咋想不起来了。

小鱼：嗯，这个方法很奈斯，但是很少人使用，因为大部分码农都是直接open 文件的。

小屌丝：那你就别藏着掖着了，赶紧展示一下，让我也长长见识。

2、 fileinput

说到fileinput，可能90%的码农表示没用过，甚至没有听说过。

这不奇怪，因为在python界，既然open可以走天下，何必要fileinput呢？。

但是，今天小鱼还是要介绍fileinput这个方法，因为太奈斯了。

不止是香。是真香！

接下来，就跟着小鱼，一起fileinput，对，就是这个feel。

2.1 方法介绍

2.1.1 基本用法

先来看一下fileinput的基本功能：

fileinput.filename()：返回当前被读取的文件名。

—>在第一行被读取之前，返回 None。

fileinput.fileno()：返回以整数表示的当前文件“文件描述符”。

—>当未打开文件时（处在第一行和文件之间），返回 -1。

fileinput.lineno()：返回已被读取的累计行号。

—>在第一行被读取之前，返回 0。在最后一个文件的最后一行被读取之后，返回该行的行号。

fileinput.filelineno()：返回当前文件中的行号。

—>在第一行被读取之前，返回 0。

—>在最后一个文件的最后一行被读取之后，返回此文件中该行的行号。

2.1.2 进阶用法

fileinput.isfirstline()：如果刚读取的行是其所在文件的第一行则返回 True，否则返回 False。

fileinput.isstdin()：如果最后读取的行来自 sys.stdin 则返回 True，否则返回 False。

fileinput.nextfile()：关闭当前文件以使下次迭代将从下一个文件（如果存在）读取第一行；不是从该文件读取的行将不会被计入累计行数。直到下一个文件的第一行被读取之后文件名才会改变。

—>在第一行被读取之前，此函数将不会生效；它不能被用来跳过第一个文件。

—>在最后一个文件的最后一行被读取之后，此函数将不再生效。

fileinput.close()：关闭序列。

2.2 默认读取

代码示例

# -*- coding:utf-8 -*-
# @Time   : 2022-07-23
# @Author : carl_DJ
import fileinput
'当 Python 脚本没有传入任何参数时，fileinput 默认会以 stdin 作为输入源'
for line in fileinput.input():
    print(f'{line}')

运行结果

你输入的内容，程序都会读取并再输出。

俗称：复读机

2.3 处理一个文件

代码示例

# -*- coding:utf-8 -*-
# @Time   : 2022-07-23
# @Author : carl_DJ
import fileinput
'files 输入打开文件的名称即可'
with fileinput.input(files=('output.txt',)) as file:
    for line in file:
        print(f'{fileinput.filename()} 第{fileinput.lineno()}行：{line}',end='')

运行结果

解析：

fileinput 有且仅有这两种读取模式：‘r’，‘rb’;

fileinput.input() 默认使用 mode=‘r’ 的模式读取文件，如果你的文件是二进制的，可以使用mode=‘rb’ 模式。

2.4 处理批量文件

2.4.1 多文件序号连续排序

调用方法

fileinput.lineno()方法

代码示例

# -*- coding:utf-8 -*-
# @Time   : 2022-07-23
# @Author : carl_DJ
import fileinput
'files 输入打开文件的名称即可'
with fileinput.input(files=('output.txt','input.txt')) as file:
    for line in file:
        #fileinput.lineno() 把两个文件的整合陈一个文件对象file，需要排序输出
        print(f'{fileinput.filename()} 第{fileinput.lineno()}行: {line}', end='')
        # fileinput.filelineno()两个文件单独读取，需要单独排序
        print(f'{fileinput.filename()} 第{fileinput.filelineno()}行: {line}', end='')

运行结果

2.4.2 多文件序号单独排序

调用方法

fileinput.filelineno()方法

代码示例

# -*- coding:utf-8 -*-
# @Time   : 2022-07-23
# @Author : carl_DJ
import fileinput
'files 输入打开文件的名称即可'
with fileinput.input(files=('test1.txt','test2.txt')) as file:
    for line in file:       
        # fileinput.filelineno()两个文件单独读取，需要单独排序
        print(f'{fileinput.filename()} 第{fileinput.filelineno()}行: {line}', end='')

运行结果

2.4.3 与glob配合用法

在颜值的时代，上面的输出样式，已经无法满足我们的需要了，

于是乎，我们就想到了glob。

代码示例

# -*- coding:utf-8 -*-
# @Time   : 2022-07-23
# @Author : carl_DJ
import fileinput
import glob
#glob 匹配te开头的txt文件
for line in fileinput.input(glob.glob("te*.txt")):
    if fileinput.isfirstline():
        #输出读取文件
        print('='*10,f'读取文件{fileinput.filename()}','='*10)
        #fileinput.filelineno()方法读取
    print(str(fileinput.filelineno())+ ':'+line.upper(),end='')

运行结果

就这颜值，哪个小姐姐能不喜欢呢。

2.5 读取与备份

调用方法

fileinput.input 的backup 参数，可以指定备份的后缀名，比如 .bak

代码示例

# -*- coding:utf-8 -*-
# @Time   : 2022-07-23
# @Author : carl_DJ
import fileinput
#触发backup的动作，源文件内容被修改，对源文件进行backup
with fileinput.input(files=("test1.txt",), backup=".bak",inplace=1) as file:
    for line in file:
        print(line.rstrip().replace('111111', '222222'))
        print(f'{fileinput.filename()} 第{fileinput.lineno()}行: {line}', end='')

运行结果

2.5 重定向替换

解析

上面的例子，用到了 inplace参数，表示是否将标准输出的结果写回文件，默认不取代

代码示例：

# -*- coding:utf-8 -*-
# @Time   : 2022-07-23
# @Author : carl_DJ
import fileinput
#触发backup的动作，源文件内容被修改，对源文件进行backup
with fileinput.input(files=("test2.txt",), inplace=True) as file:
    print("[INFO] task is started...")
    for line in file:
        print(f'{fileinput.filename()} 第{fileinput.lineno()}行: {line}', end='')
    print("[INFO] task is closed...")

运行结果

注

通过运行结果，可以看到：

- 在 for 循环体内的 print 内容会写回到原文件中了。

- 而在 for 循环体外的 print 则没有变化。

2.6 进阶

2.6.1 openhook含义解析

在 fileinput.input() 中有一个 openhook 的参数，它支持用户传入自定义的对象读取方法；

-如果没有传入任何勾子，fileinput 默认使用的是 open 函数；

2.6.2 方法介绍

fileinput 内置了两种勾子

1、fileinput.hook_compressed(filename, mode)

使用 gzip 和 bz2 模块透明地打开 gzip 和 bzip2 压缩的文件（通过扩展名 ‘.gz’ 和 ‘.bz2’ 来识别）；

如果文件扩展名不是 ‘.gz’ 或 ‘.bz2’，文件会以正常方式打开（即使用 open() 并且不带任何解压操作）；

使用示例: fi = fileinput.FileInput(openhook=fileinput.hook_compressed)

2、fileinput.hook_encoded(encoding, errors=None)

返回一个通过 open() 打开每个文件的钩子，使用给定的 encoding 和 errors 来读取文件。

使用示例: fi = fileinput.FileInput(openhook=fileinput.hook_encoded(“utf-8”, “surrogateescape”))

2.6.3 示例实战

假如我想要使用 fileinput 来读取网络上的文件，思路：

先使用 requests 下载文件到本地

再使用 open 去读取它；

# -*- coding:utf-8 -*-
# @Time   : 2022-07-23
# @Author : carl_DJ
def online_open(url, mode):
    import requests
    r = requests.get(url) 
    filename = url.split("/")[-1]
    with open(filename,'w') as f1:
        f1.write(r.content.decode("utf-8"))
    f2 = open(filename,'r')
    return f2

直接将这个函数传给 openhook 即可：

# -*- coding:utf-8 -*-
# @Time   : 2022-07-23
# @Author : carl_DJ
import fileinput
file_url = 'https://www.csdn.net/robots.txt'
with fileinput.input(files=(file_url,), openhook=online_open) as file:
    for line in file:

代码整合：

# -*- coding:utf-8 -*-
# @Time   : 2022-07-23
# @Author : carl_DJ
def online_open(url, mode):
    import requests
    r = requests.get(url)
    filename = url.split("/")[-1]
    with open(filename,'w') as f1:
        f1.write(r.content.decode("utf-8"))
    f2 = open(filename,'r')
    return f2
import fileinput
file_url = 'https://www.csdn.net/robots.txt'
with fileinput.input(files=(file_url,), openhook=online_open) as file:
    for line in file:
        print(line, end="")

运行结果

3、总结

看到这里，今天的分享差不多就要结束了。

关于fileinput的介绍，也就介绍到这里。

fileinput本身是对 open 函数的再次封装，所以在读取的部分，就比open显得更专业，更优雅，这也是仅限于读取的方面。

在写的方面，相对于open，就不是那么的强悍。

归根结底，fileinput还是一个不错的方法。值得你拥有。

最后，再唠叨一句:

关注小鱼，带你学习更多更有趣的python知识。

Python3，这个方法，应该是读取文件天花板了。

1、引言

2、 fileinput

2.1 方法介绍

2.1.1 基本用法

2.1.2 进阶用法

2.2 默认读取

2.3 处理一个文件

2.4 处理批量文件

2.4.1 多文件序号连续排序

2.4.2 多文件序号单独排序

2.4.3 与glob配合用法

2.5 读取与备份

2.5 重定向替换

2.6 进阶

2.6.1 openhook含义解析

2.6.2 方法介绍

2.6.3 示例实战

3、总结

热门文章

最新文章

相关课程

相关电子书

相关实验场景