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


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)

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值