您的位置:

快速创建多层文件夹的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快速创建多层文件夹的函数,并且对该函数进行了扩展,使其更加健壮和实用,具体扩展包括给创建目录加上时间戳或其他前缀或后缀,封装成类进行调用,加上错误处理机制等。该功能可以在日常开发中提高效率和准确度,希望对大家有所帮助。