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