录入规则文件名到CSV文件

import os
import sys
import csv


# 导出到csv文件

def export_to_csv(datas):

    with open('export.csv', 'w', newline='') as datacsv:
        csvwriter = csv.writer(datacsv)
        csvwriter.writerow(['姓名', '身份证号码'])
        csvwriter.writerows(datas)
        # for data in datas:
        #     csvwriter.writerow(data, '\t')


# 遍历一个目录

def trav_dir(dir, datas):

    file_list = os.listdir(dir)
    for file in file_list:
        # 获取文件完整路径
        file_path = os.path.join(dir, file)
        if os.path.isdir(file_path):
            trav_dir(file_path)
        else:
            # # 文件名替换
            # if str_del in file:
            #     new_file = file.replace(str_del, '')
            #     # 对文件改名
            #     new_file_path = os.path.join(dir, new_file)
            #     # 对文件重命名
            #     rename_file(file_path, new_file_path)
            # 获取文件名
            fls = os.path.splitext(file)
            if fls[1] in '.jpg, .png, jpeg':
                file = fls[0]
                index = file.find('4')
                name = file[0:index]
                number = '\t' + file[index:]
                datas.append((name, number))
                # print(file[0:index])
                # print(file[index:])


if __name__ == '__main__':
    # 获取目录
    if len(sys.argv) >= 2:
        wk_dir = sys.argv[1]
    else:
        print('需要指定一个要录入的文件夹')
        # wk_dir = '/Users/linxmouse/Desktop/csv'
        exit(-1)

    datas = []
    # 遍历一个目录
    trav_dir(wk_dir, datas)
    export_to_csv(datas)