录入规则文件名到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('')
                 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)