删除指定文件和目录的Python函数

Python是一门强大的编程语言，它不仅能够完成各种任务，还能够实现文件和目录的删除。本文将向大家介绍如何使用Python编写函数来删除指定文件和目录。

一、删除文件

Python提供了os模块，它包含了用于与操作系统进行交互的函数。使用os.remove()函数可以删除指定的文件。示例代码如下：

import os

def remove_file(file_path):
    try:
        os.remove(file_path)
        print("文件删除成功！")
    except Exception as e:
        print("文件删除失败：", e)

remove_file("test.txt")

在上面的示例代码中，remove_file()函数接收一个参数file_path，该参数表示要删除的文件的路径。os.remove()函数将删除该文件，并在控制台输出结果。

二、删除空目录

使用Python删除空目录很简单，只需使用os.rmdir()函数即可。示例代码如下：

import os

def remove_empty_dir(dir_path):
    try:
        os.rmdir(dir_path)
        print("空目录删除成功！")
    except Exception as e:
        print("空目录删除失败：", e)

remove_empty_dir("empty_dir")

在上面的示例代码中，remove_empty_dir()函数接收一个参数dir_path，该参数表示要删除的空目录的路径。os.rmdir()函数将删除该空目录，并在控制台输出结果。

三、删除非空目录

使用Python删除非空目录有两种方法，一种是使用shutil模块中的shutil.rmtree()函数，另一种是使用os模块中的os.walk()函数。下面分别演示这两种方法。

使用shutil模块

shutil模块是Python标准库中的一个模块，它提供了一些高级的文件和文件夹操作函数。使用shutil.rmtree()函数可以删除非空目录。示例代码如下：

import shutil

def remove_dir_shutil(dir_path):
    try:
        shutil.rmtree(dir_path)
        print("非空目录删除成功！")
    except Exception as e:
        print("非空目录删除失败：", e)

remove_dir_shutil("non_empty_dir")

在上面的示例代码中，remove_dir_shutil()函数接收一个参数dir_path，该参数表示要删除的非空目录的路径。shutil.rmtree()函数将删除该非空目录，并在控制台输出结果。

使用os模块

os模块还提供了一个函数os.walk()，它可以遍历指定目录下的所有子目录和文件，并返回它们的路径。使用os模块可以手动删除非空目录。示例代码如下：

import os

def remove_dir_os(dir_path):
    try:
        for root, dirs, files in os.walk(dir_path, topdown=False):
            for dir in dirs:
                os.rmdir(os.path.join(root, dir))
            for file in files:
                os.remove(os.path.join(root, file))
        os.rmdir(dir_path)
        print("非空目录删除成功！")
    except Exception as e:
        print("非空目录删除失败：", e)

remove_dir_os("non_empty_dir")

在上面的示例代码中，remove_dir_os()函数接收一个参数dir_path，该参数表示要删除的非空目录的路径。os.walk()函数用于遍历指定目录下的所有子目录和文件，其中topdown=False表示从子目录到父目录遍历。接着，在循环中使用os.rmdir()函数删除子目录，使用os.remove()函数删除文件。最后，使用os.rmdir()函数删除父目录，并在控制台输出结果。

四、总结

本文介绍了如何使用Python编写函数来删除指定文件和目录，包括删除文件、删除空目录和删除非空目录。Python提供了os模块和shutil模块来支持文件和目录的删除操作，开发者可以根据需要选择合适的方法来实现。