說到fileinput，可能90%的碼農表示沒用過，甚至沒有聽說過。
這不奇怪，因為在python界，既然open可以走天下，何必要fileinput呢？。
其為open方法的高級封裝：
fileinput模塊可以對一個或多個文件中的內容進行迭代、遍歷等操作。
該模塊的input()函數有點類似文件readlines()方法，區別在於:
前者是一個迭代對象，即每次只生成一行，需要用for循環迭代。
後者是一次性讀取所有行。在碰到大文件的讀取時，前者無疑效率更高效。
用fileinput對文件進行循環遍歷，格式化輸出，查找、替換等操作，非常方便。
fileinput.input() # 進行讀取文件對象的初始化，返回能夠用於for循環遍歷的對象
# return FileInput(files, inplace, backup, mode=mode, openhook=openhook)
fileinput.filename() # 返回當前文件的名稱
# return fileInput.filename()
fileinput.lineno() # 返回當前已經讀取的行的數量（或者序號）
# return fileInput.lineno()
fileinput.filelineno() # 返回當前讀取的行的行號
# return fileInput.filelineno()
fileinput.isfirstline() # 檢查當前行是否是文件的第一行
# return fileInput.isfirstline()
fileinput.isstdin() # 判斷最後一行是否從stdin中讀取
# return fileInput.isstdin()
fileinput.close() # 關閉隊列，也是關閉文件
fileinput.nextfile() # 關閉當前文件以使下次迭代將從下一個文件（如果存在）讀取第一行；不是從該文件讀取的行將不會被計入累計行數。直到下一個文件的第一行被讀取之後文件名才會改變。
# —>在第一行被讀取之前，此函數將不會生效；它不能被用來跳過第一個文件。
# —>在最後一個文件的最後一行被讀取之後，此函數將不再生效。
#!/usr/bin/env python
# -*- coding: UTF-8 -*-
__author__ = "A.L.Kun"
__file__ = "demo02.py"
__email__ = "[email protected]"
import fileinput
# 典型用法
for line in fileinput.input(): # 如果input函數裡面不傳入參數，則，我們需要從命令行傳入文件路徑
print(line)
python demo02.py 1.txt 2.txt
我們使用input方法打開文件
fileinput.input(files=None, inplace=False, backup="", bufsize=0, *, mode="r", openhook=None)
參數：
files：默認是stdin方式 ，傳入要打開的文件，可以傳入一個文件列表，打開多個文件inplace：是否將標准輸出的結果寫回文件，默認不取代backup：備份文件的擴展名，只指定擴展名，如.bak。如果該文件的備份文件已存在，則會自動覆蓋。bufsize：緩沖區大小，默認為0，如果文件很大，可以修改此參數，一般默認即可mode： 讀寫模式，默認為只讀openhook：鉤子函數， 該鉤子用於控制打開的所有文件，比如說編碼方式等，返回一個文件對象
#!/usr/bin/env python
# -*- coding: UTF-8 -*-
__author__ = "A.L.Kun"
__file__ = "demo02.py"
__email__ = "[email protected]"
import fileinput
from glob import glob # 使用正則匹配文件
with fileinput.input(files=glob("*.txt"), openhook=fileinput.hook_encoded("utf-8")) as f: # 讀取當前目錄下，所有的txt文件
for line in f:
print(f"{
fileinput.filename()} -> {
line.strip()} | 第{
fileinput.filelineno()}行 | 共讀取{
fileinput.lineno()}行")
#!/usr/bin/env python
# -*- coding: UTF-8 -*-
__author__ = "A.L.Kun"
__file__ = "demo02.py"
__email__ = "[email protected]"
import fileinput
from glob import glob # 使用正則匹配文件
with fileinput.input(files=glob("*.txt"), inplace=True, backup=".bak") as f: # 讀取當前目錄下，所有的txt文件，同時新建一個備份文件文件，可以不新建備份文件
for line in f:
print(f"{
fileinput.filename()} -> {
line.strip()} | 第{
fileinput.filelineno()}行 | 共讀取{
fileinput.lineno()}行") # 在for循環裡面的print內容不會輸出到控制台上，而是直接輸出到文件對應的行裡面
在 fileinput.input() 中有一個 openhook 的參數，它支持用戶傳入自定義的對象讀取方法；
fileinput 內置了兩種勾子
fileinput.hook_compressed(filename, mode)
使用 gzip 和 bz2 模塊透明地打開 gzip 和 bzip2 壓縮的文件（通過擴展名 ‘.gz’ 和 ‘.bz2’ 來識別）；
如果文件擴展名不是 ‘.gz’ 或 ‘.bz2’，文件會以正常方式打開（即使用 open() 並且不帶任何解壓操作）；
使用示例: fi = fileinput.FileInput(openhook=fileinput.hook_compressed)
fileinput.hook_encoded(encoding, errors=None)
返回一個通過 open() 打開每個文件的鉤子，使用給定的 encoding 和 errors 來讀取文件。
使用示例: fi = fileinput.FileInput(openhook=fileinput.hook_encoded(“utf-8”, “surrogateescape”))
我們定義一個鉤子函數，來實現讀取網絡資源
#!/usr/bin/env python
# -*- coding: UTF-8 -*-
__author__ = "A.L.Kun"
__file__ = "demo02.py"
__email__ = "[email protected]"
import fileinput
from glob import glob # 使用正則匹配文件
from io import StringIO
from requests import get
def getOnlineSource(url, *args, **kwargs):
resp = get(url)
resp.encoding = resp.apparent_encoding
return StringIO(resp.text) # 文件對象即為字符串流
with fileinput.input("http://www.baidu.com", openhook=getOnlineSource) as f: # 獲取對應url中的資源，並輸出
for line in f:
print(line) # 輸出每一行內容，即，等於是輸出io流中的所有內偶然
