一、为什么需要快速创建多层文件夹
在日常开发中,我们经常需要创建多层文件夹以存放不同的数据,例如训练数据、测试数据和验证数据等。手动创建多层文件夹,不仅费时费力,而且容易出错,而使用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快速创建多层文件夹的函数,并且对该函数进行了扩展,使其更加健壮和实用,具体扩展包括给创建目录加上时间戳或其他前缀或后缀,封装成类进行调用,加上错误处理机制等。该功能可以在日常开发中提高效率和准确度,希望对大家有所帮助。