python脚本，如何要求用户输入他想要的扩展文件并列出现有文件

Question

我有这个脚本，它可以在不询问用户的情况下从选定的路径中选择选定类型的文件，并且运行良好。

我想要的是能够获得用户输入并列出请求的文件类型。

有人能帮我吗？

fileSearch.py

#!/usr/bin/python
import os
from fnmatch import fnmatch
root = 'C:PythonWorkspace'
extension = "*.py"
for path, subdirs, files in os.walk(root):
    for name in files:
        if fnmatch(name, extension):
            print os.path.join(path, name)

我写了另一个基于这个的脚本

newFileSearch.py

#!/usr/bin/python
import os
from fnmatch import fnmatch
root2 = 'C:Users'
request = raw_input("Select the extension you want!")
for path, subdirs, files in os.walk(root2):
    for name in files:
        if fnmatch(name, request):
            print os.path.join(path.name)

在第二个脚本中，什么都没有发生，它只是询问用户

Answer 1

print os.path.join(path.name)应为print os.path.join(path, name)，如上所述。

此外，如果您希望用户只输入扩展名（例如py），您可以使用：

if fnmatch(name, "*." + request):