文件夹内图片生成csv-Click包初体验

需求：把文件夹的图片名和路径对应写入csv并生成随机手机号和邮箱

• 分析：
  faker库生成假数据将文件和文件名写入csv。
  使用click库，命令行输入文件路径和生成文件名。
  使用pyinstaller，打包成exe文件。

1.下面是代码

import os
from faker import Faker
import csv
import click

@click.command()
@click.option("--files",help="The file path")
@click.option("--fname",default="data_test",help="Generate file name")
def generate_data(files,fname):
    if not os.path.exists(files):
        click.echo("file does not exist")
    else:
        f = open(r'{}.csv'.format(fname),'w',encoding='utf-8',newline="")
        csv_writer = csv.writer(f)
        csv_writer.writerow(['email','phone_numeber','name','image'])
        faker = Faker("zh_CN")
        for file in os.listdir(files):
            csv_writer.writerow([faker.email(), faker.phone_number(),file.split('.',1)[0],file])

generate_data()

click包网上的介绍很多,这里就不多做说明:
https://blog.csdn.net/weixin_38278993/article/details/100052961

2.安装pyinstaller包

pip install pyinstaller
pyinstaller -D xxx.py

tips:因为只是一个脚本所以打成一个文件使用参数-D