在我们编写程序时,经常会遇到需要导入一个文件夹下面的多个文件或者是多个子目录下的所有文件。希望能实现将指定路径及其子目录下的所有图片文件,并将它们的路径保存在一个列表中返回的操作,以便于我们后续导入文件。
为了实现上述操作,我们需要使用到操作系统中的OS文件操作第三方库,使用import OS导入此库,通过导入os
模块,你可以使用一些函数来执行与文件和目录相关的操作。一些常用的os
模块的函数包括os.listdir()
、os.path.join()
、os.path.isdir()
和os.path.splitext()
。你可以通过调用这些函数来完成对文件和目录的操作。接下来,我将详细解释这些函数的用法。
image_extensions = {'.jpg', '.jpeg', '.JPG', '.JPEG'}
def genlist(image_dir):
# 遍历指定目录及其子目录下的所有图片文件,并将它们的路径保存在一个列表中返回
image_list = []
for filename in os.listdir(image_dir): # 列出指定目录(image_dir)中的所有文件和文件夹的名称,返回一个包含这些名称的列表
path = os.path.join(image_dir,filename) # 将路径名(path)和文件名(filename)组合成完整路径的函数调用
if os.path.isdir(path): # 用于判断给定路径(path)是否为一个目录
image_list = image_list + genlist(path)
else:
ext = os.path.splitext(filename)[1] # 将一个路径分割为基本路径和扩展名两部分,并以元组的形式返回
if ext in image_extensions:
image_list.append(os.path.join(image_dir, filename)) # 将路径名(path)和文件名(filename)组合成完整路径
return image_list # 返回的所有文件的路径的列表
args = parse_args()
if not os.path.exists(args.output_dir): # 判断文件夹路径是否存在
os.makedirs(args.output_dir) # 路径不存在,则创建它
flist = genlist(args.input_dir) # 获取指定路径下的所有文件的路径的列表
for i,p in enumerate(flist): # 遍历flist列表中的元素和索引。在每次迭代中,索引被赋值给变量i,而对应的元素则赋值给变量p
if os.path.getsize(p) != 0: # os.path.getsize(p)函数获取文件p的大小,并将其与0进行比较。这个条件判断语句的作用是判断文件是否为空
os.rename(p,os.path.join(args.output_dir,str(i)+'.jpg')) # 重命名为args.output_dir目录下的一个新文件名
shutil.rmtree(args.input_dir) # 删除整个args.input_dir目录及其所有子目录和文件
print('done')