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

Question

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

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

用户首先输入目录是在之后的另一种方法，他们会说

éTXT或E .TXT（两部作品），它会根据你使用什么扩展返回的文件。我不确定我的代码有什么问题。

Answer 1

在你的代码，

for file in directory 

肯定是不对的，因为目录是在它的路径的字符串。它应该像

for file in os.listdir(directory): 

获得目录