快速創建多層文件夾的Python函數

一、為什麼需要快速創建多層文件夾

在日常開發中，我們經常需要創建多層文件夾以存放不同的數據，例如訓練數據、測試數據和驗證數據等。手動創建多層文件夾，不僅費時費力，而且容易出錯，而使用Python編寫函數來快速創建多層文件夾，可以極大地提高效率和準確度。

二、Python函數實現

下面是一個快速創建多層文件夾的Python函數：

import os

def mkdir(path):
    """
    遞歸創建多級目錄
    :param path: 需要創建的目錄路徑
    :return: 無
    """
    # 去除首位空格
    path = path.strip()
    # 去除尾部 \ 符號
    path = path.rstrip("\\")
 
    # 判斷路徑是否存在
    # 存在     True
    # 不存在   False
    is_exists = os.path.exists(path)
 
    # 判斷結果
    if not is_exists:
        # 如果不存在則創建目錄
        os.makedirs(path)

        print(path + ' 創建成功')
        return True
    else:
        # 如果目錄存在則不創建，並提示目錄已存在
        print(path + ' 目錄已存在')
        return False

三、函數使用說明

使用該函數非常簡單，只需要傳入需要創建的文件夾路徑即可。如果路徑不存在，則自動創建對應的目錄。

例如，我們需要在D盤下創建一個名為「test」的文件夾，其中包含三個子文件夾：data、model和result，我們可以按照以下方式來調用該函數：

mkdir("D:\\test\\data")
mkdir("D:\\test\\model")
mkdir("D:\\test\\result")

這樣就可以快速創建多層的文件夾了，如果文件夾已經存在，則不會再次創建。

四、函數擴展

上述代碼雖然已經可以快速創建多層文件夾了，但是仍有一些不足之處。為了進一步提高代碼的健壯性和可擴展性，我們可以對該函數進行以下擴展：

1. 給創建的目錄加上時間戳或其他前綴或後綴，方便管理和查找。

2. 將創建目錄的過程封裝成一個類，方便多處調用。

3. 加上錯誤處理機制，例如如果傳入的路徑不合法、無法創建等，則進行異常捕獲並輸出錯誤信息。

代碼示例：

import os
import time
 
class FolderCreator:
    def __init__(self, folder_path):
        """
        初始化
        :param folder_path: 文件夾路徑
        """
        self.folder_path = folder_path
 
    def create_folder(self):
        """
        創建文件夾
        """
        try:
            # 判斷路徑是否存在
            # 存在     True
            # 不存在   False
            is_exists = os.path.exists(self.folder_path)
 
            # 判斷結果
            if not is_exists:
                # 如果不存在則創建目錄
                os.makedirs(self.folder_path)
 
                print(self.folder_path + ' 創建成功')
                return True
            else:
                # 如果目錄存在則不創建，並提示目錄已存在
                print(self.folder_path + ' 目錄已存在')
                return False
        except Exception as e:
            print("創建文件夾出錯：{}".format(e))
    
    def create_folder_with_timestamp(self, prefix=True):
        """
        創建帶時間戳的文件夾
        :param prefix: 是否將時間戳作為前綴
        """
        now = time.strftime("%Y-%m-%d_%H-%M-%S", time.localtime())
 
        if prefix:
            self.folder_path = now + "_" + self.folder_path
        else:
            self.folder_path = self.folder_path + "_" + now
 
        self.create_folder()

五、總結

本文介紹了如何使用Python快速創建多層文件夾的函數，並且對該函數進行了擴展，使其更加健壯和實用，具體擴展包括給創建目錄加上時間戳或其他前綴或後綴，封裝成類進行調用，加上錯誤處理機制等。該功能可以在日常開發中提高效率和準確度，希望對大家有所幫助。