读取文件夹图片数据，写入CSV文件

最新推荐文章于 2022-12-11 14:59:24 发布

原创最新推荐文章于 2022-12-11 14:59:24 发布 · 605 阅读

5 ·

CC 4.0 BY-SA版权

该代码段实现从指定目录读取图片文件，批量生成包含这些信息的CSV文件，并将CSV文件与源图片一起压缩成ZIP档案。CSV文件包含personId、name、image路径等字段，用于数据管理和上传状态记录。


import os
import sys
import csv
import zipfile

file_csv = 'CSV_TEST.csv'

def create_csv(path):
    count = 0
    personid_prefix = 'CX'
    username_prefix = 'CXX'

    with open(file_csv, 'a',encoding='utf-8-sig', newline = '') as csv_f:
        row = ['personId', 'name', 'image', 'wechatUserID', 'icCard', 'idCard','qrCode', '展示内容', '设备组', '二次鉴权（2关闭 1打开）' ,'开始时间' ,'结束时间']
        csv_writer = csv.writer(csv_f, dialect = "excel")
        csv_writer.writerow(row)

        for f in os.listdir(path):
            count += 1
            personid = personid_prefix + str(count)
            username = username_prefix + str(count)
            row = [personid, username, str(f), '', '', '', '', '[]', '["uploading"]', '2', '2020-07-01 13:12', '2020-08-01 13:12']
            csv_writer.writerow(row)

def to_zip(img_path):
    zip_f = zipfile.ZipFile('test.zip','a',zipfile.ZIP_STORED)
    zip_f.write(file_csv)
    for img_f in os.listdir(img_path):
        zip_f.write(os.path.join(img_path, img_f))
    zip_f.close()


if __name__ == '__main__':
    #img_path = sys.argv[1]
    img_path = 'C:/Users/tangwei_tang/PycharmProjects/pythonProject/photo/images/'
    create_csv(img_path)
    #to_zip(img_path)