Python中安全删除目录的rmtree_safe()函数

在Python中，可以使用shutil.rmtree()函数来删除目录及其内容。但是这个函数并没有提供安全删除的选项，如果目录内容很重要，可能会因为误操作而导致数据丢失。为了解决这个问题，可以自己编写一个rmtree_safe()函数，先将要删除的目录和内容复制到一个临时目录中，再删除原目录，最后将临时目录移动到原目录的位置。下面是一个实现这个功能的rmtree_safe()函数的例子。

import shutil
import os
import tempfile

def rmtree_safe(directory):
    # 创建一个临时目录
    temp_dir = tempfile.mkdtemp()

    try:
        # 将要删除的目录及其内容复制到临时目录中
        shutil.copytree(directory, temp_dir)

        # 删除原目录及其内容
        shutil.rmtree(directory)

        # 将临时目录移动到原目录的位置
        shutil.move(temp_dir, directory)
    
    except Exception as e:
        # 发生任何异常时，删除临时目录
        shutil.rmtree(temp_dir)
        raise e

# 使用例子
directory = 'path/to/directory'

try:
    rmtree_safe(directory)
except Exception as e:
    print(f"删除目录失败：{e}")
else:
    print(f"成功删除目录：{directory}")

在上述例子中，首先使用tempfile.mkdtemp()函数创建一个临时目录，然后使用shutil.copytree()函数将要删除的目录及其内容复制到临时目录中。接下来使用shutil.rmtree()函数删除原目录及其内容，然后使用shutil.move()函数将临时目录移动到原目录的位置。如果在这个过程中发生任何异常，会使用shutil.rmtree()函数删除临时目录，并将异常继续抛出。

这样就实现了一个安全删除目录的rmtree_safe()函数。在使用这个函数时，注意替换directory变量为实际要删除的目录路径。