当前位置：物联沃-IOTWORD物联网 > 技术教程 > Python3，这个方法，应该是读取文件天花板了。
代码收藏家代码收藏家 技术教程

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

fileinput介绍与实战

  • 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、总结

    • 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:
        print(line, end="")

    代码整合：

    # -*- 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知识。

    来源：Carl_奕然

    您可能感兴趣的内容:

    物联沃分享整理
    物联沃-IOTWORD物联网 » Python3，这个方法，应该是读取文件天花板了。

    代码收藏家 普通

    分享到：

    相关推荐

    发表评论