Question

我要求用户输入一个名称，我想在文件的目录中搜索一个文件是否存在与输入名称相同的文件，如果是，则程序要求一个新名称，如果没有，那么它将创建一个具有该名称的文件。我正在努力，因为我编写的代码只检查目录中的第一个文件，而不是继续其余的。我遇到问题的代码看起来像这样：

import os
Name = input("Please enter name for new file")
if Name in os.listdir():
    print("Sorry this name already exists, please choose another one")
    break
else:
    NewFile = open("Name" + ".txt", "w+")
    break

Answer 1

您可以使用glob：

import glob
existing = glob.glob('*.txt')
name = '{}.txt'.format(input("Please enter name for new file"))
if name in existing:
    print("Sorry this name already exists, please choose another one")
else:
    newfile = open(name, "w+")

Python：逐个检查目录中的文件名

1 个答案: