Python删除有文件的目录

Python删除有文件的目录

在Python中，我们可以使用shutil模块来删除目录，但是这个模块只能删除空目录。如果目录中有文件，我们需要先删除目录中的所有文件，然后再删除目录本身。

下面我们将演示如何在Python中删除有文件的目录。

步骤1：删除目录中所有文件

首先，我们需要编写一个函数来删除目录中的所有文件。我们可以使用os模块来遍历目录中的所有文件，并使用os.remove()来删除文件。

import os

def delete_files_in_dir(directory):
    for root, dirs, files in os.walk(directory):
        for file in files:
            file_path = os.path.join(root, file)
            os.remove(file_path)

# 测试删除目录中的所有文件
delete_files_in_dir("test_dir")

在上面的代码中，我们定义了一个delete_files_in_dir()函数，该函数接受一个目录作为参数，并遍历该目录中的所有文件，然后使用os.remove()来删除文件。在测试中，我们删除了名为test_dir的目录中的所有文件。

步骤2：删除目录本身

接下来，我们可以使用shutil模块的rmdir()函数来删除目录本身。

import shutil

def delete_dir(directory):
    try:
        shutil.rmtree(directory)
        print(f"Directory '{directory}' deleted successfully.")
    except OSError as e:
        print(f"Error: {e.strerror}")

# 测试删除目录本身
delete_dir("test_dir")

在上面的代码中，我们定义了一个delete_dir()函数，该函数接受一个目录作为参数，并使用shutil.rmtree()来删除目录及其所有内容。在测试中，我们删除了名为test_dir的目录。

示例代码

下面是一个完整的示例代码，演示了如何删除有文件的目录：

import os
import shutil

def delete_files_in_dir(directory):
    for root, dirs, files in os.walk(directory):
        for file in files:
            file_path = os.path.join(root, file)
            os.remove(file_path)

def delete_dir(directory):
    try:
        shutil.rmtree(directory)
        print(f"Directory '{directory}' deleted successfully.")
    except OSError as e:
        print(f"Error: {e.strerror}")

# 测试删除目录中的所有文件
delete_files_in_dir("test_dir")

# 测试删除目录本身
delete_dir("test_dir")

运行结果

在运行上面的示例代码后，如果test_dir目录存在，并包含文件，则会输出类似如下的结果：

Directory 'test_dir' deleted successfully.

这样，我们就成功删除了有文件的目录。

总之，通过上面的步骤，我们可以在Python中删除具有文件的目录。首先删除目录中的所有文件，然后再删除目录本身。这样可以确保目录被完全清空和删除。