刪除指定文件和目錄的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模塊來支持文件和目錄的刪除操作，開發者可以根據需要選擇合適的方法來實現。