# -*- 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()
本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。