如何检查目录是否包含某种文件类型？

Question

我正在尝试创建一个程序，该程序遍历文件夹中的每个文件，处理找到的任何文本文件。 到目前为止，我有这样的事情：

import os
for file in os.listdir():
    if file.endswith(".txt"):
        open(file)

如果目录中不存在文本文件，如何让它返回错误？

Answer 1

如果您只想检查.txt个文件的计数是0，那么您可以改为使用glob.glob()：

import glob

if len(glob.glob("*.txt")) == 0:
    print "No TXT files"

Answer 2

import glob

if not glob.glob("*.txt"):
  print('No text files Found')