如何通过通配符参数，如* .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)