天花板级别的Python读取文件的方法

发表于:2022-11-25 09:51

字体: | 上一篇 | 下一篇 | 我要投稿

 作者:佚名    来源:CSDN

#
Python
分享:
  序言
  哈喽兄弟们,今天咱们来了解一下 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 即可:
  # -*- 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="")
  运行结果:
  总结
  关于fileinput的介绍,也就介绍到这里。
  fileinput本身是对 open 函数的再次封装,所以在读取的cc部分,就比open显得更专业,更优雅,这也是仅限于读取的方面。
  在写的方面,相对于open,就不是那么的强悍。
  归根结底,fileinput还是一个不错的方法。值得你拥有。
  本文内容不用于商业目的,如涉及知识产权问题,请权利人联系51Testing小编(021-64471599-8017),我们将立即处理
《2023软件测试行业现状调查报告》独家发布~

关注51Testing

联系我们

快捷面板 站点地图 联系我们 广告服务 关于我们 站长统计 发展历程

法律顾问:上海兰迪律师事务所 项棋律师
版权所有 上海博为峰软件技术股份有限公司 Copyright©51testing.com 2003-2024
投诉及意见反馈:webmaster@51testing.com; 业务联系:service@51testing.com 021-64471599-8017

沪ICP备05003035号

沪公网安备 31010102002173号