1 import os 2 import sys 3 import csv 4 5 6 # 导出到csv文件 7 8 def export_to_csv(datas): 9 10 with open('export.csv', 'w', newline='') as datacsv: 11 csvwriter = csv.writer(datacsv) 12 csvwriter.writerow(['姓名', '身份证号码']) 13 csvwriter.writerows(datas) 14 # for data in datas: 15 # csvwriter.writerow(data, ' ') 16 17 18 # 遍历一个目录 19 20 def trav_dir(dir, datas): 21 22 file_list = os.listdir(dir) 23 for file in file_list: 24 # 获取文件完整路径 25 file_path = os.path.join(dir, file) 26 if os.path.isdir(file_path): 27 trav_dir(file_path) 28 else: 29 # # 文件名替换 30 # if str_del in file: 31 # new_file = file.replace(str_del, '') 32 # # 对文件改名 33 # new_file_path = os.path.join(dir, new_file) 34 # # 对文件重命名 35 # rename_file(file_path, new_file_path) 36 # 获取文件名 37 fls = os.path.splitext(file) 38 if fls[1] in '.jpg, .png, jpeg': 39 file = fls[0] 40 index = file.find('4') 41 name = file[0:index] 42 number = ' ' + file[index:] 43 datas.append((name, number)) 44 # print(file[0:index]) 45 # print(file[index:]) 46 47 48 if __name__ == '__main__': 49 # 获取目录 50 if len(sys.argv) >= 2: 51 wk_dir = sys.argv[1] 52 else: 53 print('需要指定一个要录入的文件夹') 54 # wk_dir = '/Users/linxmouse/Desktop/csv' 55 exit(-1) 56 57 datas = [] 58 # 遍历一个目录 59 trav_dir(wk_dir, datas) 60 export_to_csv(datas)