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

时间:2017-10-17 08:51:59

标签: python directory

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

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

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

2 个答案:

答案 0 :(得分:3)

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

import glob

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

答案 1 :(得分:1)

import glob

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