python - 防止 FileExistsError 的函数

Question

我有这个函数来防止FileExistsError。它创建一个文件夹，但如果有另一个同名文件夹，它会在该文件夹的名称中添加一个序号。

import os

def create_folder(folder_name):

      path = '/home/k/Desktop/' + folder_name
      folder_number = 1

      if os.path.exists(path):
            while path == os.path.exists(path):
                  folder_number += 1
            path = '/home/k/Desktop/' + folder_name + ' ({})'.format(folder_number)
            os.makedirs(path)

      else:
            os.makedirs(path)

我能够创建该文件夹，如果我创建另一个同名的文件夹，我会将名称设置为 myfoldername(1)，但如果再次创建它，则会引发 FileExistsError:

FileExistsError: [Errno 17] File exists: '/home/k/Desktop/test (1)'

我认为错误出现在 while 循环中，但我不知道如何修复它。

Answer 1

分配带有递增数字的新路径名应该在 while 循环内，以便可以更新 path，以及 while 的条件循环只需要返回 bool 值os.path.exists方法:

while os.path.exists(path):
    folder_number += 1
    path = '/home/k/Desktop/' + folder_name + ' ({})'.format(folder_number)