使用os.path基于扩展名搜索文件

Question

我想知道如何根据扩展名搜索文件。如果你要搜索py或txt或其他扩展名，用户会输入它然后搜索它。我不允许使用glob或os.walk

这是我的方法。如有必要，我会发布整个代码。

用户首先输入一个目录，然后用另一种方法，他们会说

E txt或E .txt（两者都有效），它会根据您使用的扩展名返回文件。我不确定我的代码有什么问题。

Answer 1

在您的代码中，

 for file in directory 

肯定是错误的，因为directory是一个包含路径的字符串。它应该像

for file in os.listdir(directory):

获取目录中的文件列表