python 读取单文件夹中的图片文件信息保存到csv文件中

# -*- coding: utf-8 -*-
# @Time    : 2019-09-17 10:21
# @Author  : scyllake
import os
import csv

#要读取的文件的根目录
root_path=r'C:\Users\zjk\Desktop\整理后的图片'

#将所有目录下的文件信息放到列表中
def get_Write_file_infos(path):
    # 文件信息列表
    file_infos_list=[]

    # 遍历并写入文件信息
    for root, dirnames, filenames in os.walk(path):
        for filename in filenames:
            file_infos = {}
            dirname=root
            filename1 = filename.split('.jpg')[0]
            file_infos["商品名称"] = filename1
            file_infos["尺寸"]=''
            file_infos["图片"]=''
            
            #将数据追加字典到列表中
            file_infos_list.append(file_infos)
    return file_infos_list


#写入csv文件
def write_csv(file_infos_list):

    with open('1.csv','a+',newline='') as csv_file:
        csv_writer = csv.DictWriter(csv_file,fieldnames=['商品名称','尺寸','图片'])
        csv_writer.writeheader()
        for each in file_infos_list:
            csv_writer.writerow(each)


#主函数
def main():
#调用获取文件信息的函数
    file_infos_list=get_Write_file_infos(root_path)
#执行写入程序
    write_csv(file_infos_list)

#主程序入口

if __name__ == '__main__':
    main()