1

昨晚我写了一个程序，它把所有注释的行从一个文件中拉出来，然后输出到一个新文件中。这是我们在我的编程课上经常要做的事情，并且通过文件复制和粘贴来挑选很快就会变老。

我检查了我的输入文件和输出文件以查看它们是否存在。如果输入文件确实存在，那么一切都会继续——如果不存在，请询问另一个文件名。如果输出文件确实存在，询问用户是否要覆盖——如果它不存在，一切都很好。

我遇到的问题是，当 file = open(fileName, 'r') 找不到文件时，一项检查正确地引发 IOError，另一项检查创建一个空白文件而不是给出 IOError。

这尤其令人困惑，因为这两位代码几乎相同。相同的过程，只是不同的文件名变量......

代码如下。第二部分是创建空白文件的部分。首先按预期给出错误。

# Try to open the input file
inFileOpen = False
while not inFileOpen and userTrying:
    # If it opens, all good
    try:
        inFile = open(inFileName, 'r')
        inFileOpen = True
    # If it doesn't open, ask user to try a different file
    except IOError:
        ...

# Try to open the output file
toFileOpen = False
while not toFileOpen and userTrying:
    # If the file opens in r mode, that means it exists already, so ask user if they
    # want to overwrite the existing file, if not ask for a new file name
    try:
        # ********************
        # For some reason, 'r' mode is creating a file... no clue...
        # ********************
        toFile = open(toFileName)
        toFile.close() # I have tried removing this just guessing at solutions. Didn't work.

        # ... Ask if user wants to overwrite

    # If the file can't be opened, all good, that means it doesn't exist yet
    except IOError:
        toFileOpen = False
        toFile = open(toFileName, 'w')
4

2 回答 2

2

如果您使用的是 Python 3.3，open则具有与“w”相同的“x”模式，但FileExistsException如果文件已经存在，它将引发 a。这比单独检查要好，因为它是对现有文件的测试和打开它以进行写入之间的原子和免疫竞争条件。虽然这对于您当前的脚本可能不是问题，但在更安全的关键程序中可能很重要。

在旧版本的 Python（2.x 和 3.0-3.2）中，它有点棘手。一种选择是打开文件以进行附加（“a”模式），然后检查当前流位置（使用tell）以查看它是否已包含某些内容。如果是这样，但用户想要替换旧内容，您可以调用truncate()然后seek(0)擦除当前数据并返回到文件的开头。这仍然可能会在没有通知的情况下覆盖零长度文件，但我怀疑这不是一个大问题。另一种方法是将os.O_EXCL|os.O_CREAT标志显式传递给os.open，然后将返回的文件描述符传递给os.fdopen以获取常规 Python 文件对象。这本质上是“x”模式在幕后所做的。

编辑：这是一个漂亮的上下文管理器解决方案，用于创建一个输出文件，给定合适的get_filename和confirm_overwrite与用户进行实际交互的功能。此代码适用于 Python 3.3，但可以修改open调用和except语句以使其适用于早期版本。

from contextlib import contextmanager

@contextmanager
def create_output_file():
    while True:
        out_filename = get_filename()

        try:
            with open(out_filename, "x") as f:
                yield f
            break

        except FileExistsException:
            if confirm_overwrite(out_filename):
                with open(out_filename, "w") as f:
                    yield f
                break

将其与with语句一起使用：

with create_output_file() as f:
    f.write("whatever")
于 2012-11-11T22:48:01.050 回答
1

我认为 os.path.exists(filename) 将是检查文件的好方法：

例如。

 while not os.path.exists(infile):
     infile = askfilename()
 with open(infile) as f:
     for line in f:
         ...

您可以对 outfile 执行类似操作

于 2012-11-11T22:51:22.647 回答