如何通過通配符參數，如* .TXT在窗口在cmd

Question

我的目錄樹：

test/ 
|_____ 1.txt content: 1_line1\n1_line2 
|_____ 2.txt content: 2_line1\n2_line2 
|_____ test_fileinput.py 

我的Python腳本：

import fileinput 
import sys 

for line in fileinput.input(sys.argv[1:]): 
    print(fileinput.filename(), fileinput.filelineno(), line) 

首先我想它在Linux上，因爲你看到它完美的作品：

$ python3 test_fileinput.py *.txt 
1.txt 1 1_line1 

1.txt 2 1_line2 
2.txt 1 2_line1 

2.txt 2 2_line2 

但在Windows上：

我當然可以做python test_fileinput.py 1.txt 2.txt，但我不知道有沒有，我仍然可以在Windows上通過*.txt的方法嗎？謝謝。

Answer 1

您可以使用glob模塊，該模塊提供了執行通配符匹配的平臺無關方式。例如，glob('*.txt')會返回當前目錄中所有txt文件的列表。

import fileinput 
import sys 
from glob import glob 

for line in fileinput.input(glob(sys.argv[1]): 
    print(fileinput.filename(), fileinput.filelineno(), line) 

如果您想保留在那裏你可以指定多個輸入（這樣做python test_fileinput.py *.txt *.csv other.md將是有效的行爲）的行爲，您可以修改代碼如下：

import fileinput 
import sys 
from glob import glob 

all_files = [f for files in sys.argv[1:] for f in glob(files)] 

for line in fileinput.input(all_files): 
    print(fileinput.filename(), fileinput.filelineno(), line)