哈喽兄弟们,今天咱们来了解一下 fileinput 。说到fileinput,可能90%的码农表示没用过,甚至没有听说过。这不奇怪,因为在python界,既然open可以走天下,何必要fileinput呢?
但是,今天我还是要介绍fileinput这个方法,因为太奈斯了。不止是香。是真香!接下来,就跟着我,一起fileinput,对,就是这个feel。
正文
1、方法介绍
基本用法
先来看一下fileinput的基本功能:
- fileinput.filename():返回当前被读取的文件名。
—>在第一行被读取之前,返回 None。 - fileinput.fileno():返回以整数表示的当前文件“文件描述符”。
—>当未打开文件时(处在第一行和文件之间),返回 -1。 - fileinput.lineno():返回已被读取的累计行号。
—>在第一行被读取之前,返回 0。在最后一个文件的最后一行被读取之后,返回该行的行号。 - fileinput.filelineno():返回当前文件中的行号。
—>在第一行被读取之前,返回 0。
—>在最后一个文件的最后一行被读取之后,返回此文件中该行的行号。
进阶用法
- fileinput.isfirstline():如果刚读取的行是其所在文件的第一行则返回 True,否则返回 False。
- fileinput.isstdin():如果最后读取的行来自 sys.stdin 则返回 True,否则返回 False。
- fileinput.nextfile():关闭当前文件以使下次迭代将从下一个文件(如果存在)读取第一行;不是从该文件读取的行将不会被计入累计行数。直到下一个文件的第一行被读取之后文件名才会改变。
—>在第一行被读取之前,此函数将不会生效;它不能被用来跳过第一个文件。
—>在最后一个文件的最后一行被读取之后,此函数将不再生效。 - fileinput.close():关闭序列。
2、 默认读取
代码示例
import fileinput
'当 Python 脚本没有传入任何参数时,fileinput 默认会以 stdin 作为输入源'
for line in fileinput.input():
print(f'{line}')
运行结果
你输入的内容,程序都会读取并再输出。
俗称:复读机
3、处理一个文件
代码示例
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’ 模式。
4、处理批量文件
多文件序号连续排序
调用方法
- fileinput.lineno()方法
代码示例
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='')
运行结果
多文件序号单独排序
调用方法
- fileinput.filelineno()方法
代码示例
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='')
运行结果
与glob配合用法
在颜值的时代,上面的输出样式,已经无法满足我们的需要了,于是乎,我们就想到了glob。
代码示例
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='')
运行结果
就这颜值,哪个小姐姐能不喜欢呢。
5、读取与备份
调用方法
- fileinput.input 的backup 参数,可以指定备份的后缀名,比如 .bak
代码示例
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='')
运行结果
6、重定向替换
解析
- 上面的例子, 用到了 inplace参数,表示是否将标准输出的结果写回文件,默认不取代。
代码示例:
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 则没有变化。
7、进阶
openhook含义解析
- 在 fileinput.input() 中有一个 openhook 的参数,它支持用户传入自定义的对象读取方法;
- 如果没有传入任何勾子,fileinput 默认使用的是 open 函数;
方法介绍
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”))
示例实战
假如我想要使用 fileinput 来读取网络上的文件,思路:
- 先使用 requests 下载文件到本地
- 再使用 open 去读取它;
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 即可:
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="")
代码整合:
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="")
运行结果
总结
关于fileinput的介绍,也就介绍到这里,fileinput本身是对 open 函数的再次封装,所以在读取的cc部分,就比open显得更专业,更优雅,这也是仅限于读取的方面。在写的方面,相对于open,就不是那么的强悍。归根结底,fileinput还是一个不错的方法。值得你拥有。
这里给大家分享一份Python全套学习资料,包括学习路线、软件、源码、视频、面试题等等,都是我自己学习时整理的,希望可以对正在学习或者想要学习Python的朋友有帮助!
CSDN大礼包:全网最全《全套Python学习资料》免费分享🎁
😝有需要的小伙伴,可以点击下方链接免费领取或者V扫描下方二维码免费领取🆓
1️⃣零基础入门
① 学习路线
对于从来没有接触过Python的同学,我们帮你准备了详细的学习成长路线图。可以说是最科学最系统的学习路线,你可以按照上面的知识点去找对应的学习资源,保证自己学得较为全面。
② 路线对应学习视频
还有很多适合0基础入门的学习视频,有了这些视频,轻轻松松上手Python~
③练习题
每节视频课后,都有对应的练习题哦,可以检验学习成果哈哈!
因篇幅有限,仅展示部分资料
2️⃣国内外Python书籍、文档
① 文档和书籍资料
3️⃣Python工具包+项目源码合集
①Python工具包
学习Python常用的开发软件都在这里了!每个都有详细的安装教程,保证你可以安装成功哦!
②Python实战案例
光学理论是没用的,要学会跟着一起敲代码,动手实操,才能将自己的所学运用到实际当中去,这时候可以搞点实战案例来学习。100+实战案例源码等你来拿!
③Python小游戏源码
如果觉得上面的实战案例有点枯燥,可以试试自己用Python编写小游戏,让你的学习过程中增添一点趣味!
4️⃣Python面试题
我们学会了Python之后,有了技能就可以出去找工作啦!下面这些面试题是都来自阿里、腾讯、字节等一线互联网大厂,并且有阿里大佬给出了权威的解答,刷完这一套面试资料相信大家都能找到满意的工作。
5️⃣Python兼职渠道
而且学会Python以后,还可以在各大兼职平台接单赚钱,各种兼职渠道+兼职注意事项+如何和客户沟通,我都整理成文档了。
上述所有资料 ⚡️ ,朋友们如果有需要 📦《全套Python学习资料》的,可以扫描下方二维码免费领取 🆓
😝有需要的小伙伴,可以点击下方链接免费领取或者V扫描下方二维码免费领取🆓