检查 Python 中是否存在 *type* 文件

Question

我意识到这看起来与有关检查文件是否存在的其他问题相似，但它是不同的。 我试图弄清楚如何检查某种类型的文件是否存在，如果不存在则退出。 我最初尝试的代码是这样的：

filenames = os.listdir(os.curdir)

for filename in filenames:

   if os.path.isfile(filename) and filename.endswith('.fna'):
        ##do stuff

这可以对以 .fna 结尾的文件“做一些事情”，但我需要它来检查并确保 .fna 文件在那里，如果没有，则完全退出程序。

我试过这个：

try:

    if os.path.isfile(filename) and filename.endswith('.fna'):
        ## Do stuff 
except: 

    sys.stderr.write (‘No database file found. Exiting program. /n’)
    sys.exit(-1)

但这不起作用，如果 .fna 文件不存在，它只会跳过整个函数，而不打印错误。

Answer 1

Python 中的for语句有一个鲜为人知的else子句：

for filename in filenames:
    if os.path.isfile(filename) and filename.endswith(".fna"):
        # do stuff
        break
else:
    sys.stderr.write ('No database file found. Exiting program. \n')
    sys.exit(-1)

else子句仅在for语句运行整个枚举而不在循环内执行break运行。

Answer 2

查看glob模块：

import glob
import os
import sys

databases = filter(os.path.isfile, glob.glob('./*.fna'))

if not databases:
    sys.stderr.write("No database found.\n\n")
    exit(1)

for database in databases:
    do_something_with(database)

Answer 3

filenames = os.listdir(os.curdir)
found = False
for filename in filenames:
    if os.path.isfile(filename) and filename.endswith('.fna'):
        found = True
if not found:
    sys.stderr.write ('No database file found. Exiting program. \n')
    sys.exit(-1)

Answer 4

查看 os.path.splitext(path) 函数，它说：

将路径名 path 拆分为一对 (root, ext)，使得 root + ext == path 和 ext 为空或以句点开头且最多包含一个句点。 基本名称上的前导句点将被忽略； splitext('.cshrc') 返回 ('.cshrc', '')。

下面是一个例子：

>>> os.path.splitext("./01The News from Lake Wobegon/AlbumArtSmall.jpg")
('./01The News from Lake Wobegon/AlbumArtSmall', '.jpg')
>>> 

Answer 5

如果您正在使用异常，请不要测试该文件是否存在。 这就是您使用异常开始的原因。

try:
    # No if statement doing the check
    # Start doing stuff assuming abc.fna exists
except:
    # Uh oh! Something went bad.
    # Display error messages, clean up, carry on

为了澄清，请考虑代码片段：

try:
    with open('hi.txt') as f:
        print f.readlines()
except:
    print 'There is no hi.txt!'

Answer 6

您的try / except块不起作用，因为os.path.isfile如果失败则不会抛出异常； 它只是返回False 。

for循环的else子句很奇怪且不直观。 使用break来表示循环成功而不是合法地打破它只是很奇怪，并且与 Python 的哲学相悖。

这是一个很好的 Pythonic 方式来做你想做的事：

want = lambda f: os.path.isfile(f) and f.endswith(".fna")
valid_files = [f for f in os.listdir(os.curdir) if want(f)]
if len(valid_files) == 0:
    print >>sys.stderr, "failed to find .fna files!"
    sys.exit(1)
for filename in valid_files:
    # do stuff