fromaipimportAipOcrfromPILimportImagefromosimportlistdirfromosimportrenamefromosimportremovefromosimportsystemfromwin32com.clientimportDispatchfromosimportgetcwdNAME_LIST=[]EXCEL_PATH=[]defget_name_by_pic(img):img.save('test_tmp.jpg')APP_ID='你的百度文字识别ID'API_KEY='你的KEY'SECRECT_KEY='你的SECRECT_KEY'client=AipOcr(APP_ID,API_KEY,SECRECT_KEY)i=open('test_tmp.jpg','rb')img=i.read()message=client.basicGeneral(img)name=message['words_result'][0]['words']returnnamedefget_pic(pic,which_size):img=Image.open(pic)region=img.crop(which_size)returnregiondefexcel(WHICH_SHEET):#WHICH_SHEET='Sheet1'COL=10excel=Dispatch('Excel.Application')excel.Visible=TruemyExcel=excel.Workbooks.Open(getcwd()+'\\'+EXCEL_PATH[0])mySheet=myExcel.Worksheets(WHICH_SHEET)#mySheet.Cells(9,9).Interior.ColorIndex=6#表格背景6为黄5位蓝4位绿3为红2为白1为黑flag=0len_name_list=int(len(NAME_LIST))print(len(NAME_LIST))forROWinrange(4,mySheet.usedrange.rows.count+1):foriinNAME_LIST:ifi[2:5]==mySheet.Cells(ROW,COL).Value[2:5]:NAME_LIST.remove(i)flag+=1print(flag,mySheet.Cells(ROW,COL).Value)mySheet.Cells(ROW,COL).Interior.ColorIndex=6rename(i,mySheet.Cells(ROW,COL).Value+'.jpg')breakprint('\n-----未匹配成功：-------\n',len_name_list-int(flag),NAME_LIST)myExcel.savemyExcel.closeexcel.Visible=0defrm_space(str):str=str.strip()str=str.replace('','')str=str.replace('','')returnstrdefmain():old_file_name=listdir(r'./')forjpg_excelinrange(0,int(len(old_file_name))):print(old_file_name[jpg_excel].split('.')[1])ifold_file_name[jpg_excel].split('.')[1]=='jpg'orold_file_name[jpg_excel].split('.')[1]=='png':NAME_LIST.append(old_file_name[jpg_excel])#elifold_file_name[jpg_excel].split('.')[1]=='xlsx'orold_file_name[jpg_excel].split('.')[1]=='xls'orold_file_name[jpg_excel].split('.')[1]=='xl*':elifold_file_name[jpg_excel].split('.')[1]=='xlsx':EXCEL_PATH.append(old_file_name[jpg_excel])elifold_file_name[jpg_excel].split('.')[1]=='xls':EXCEL_PATH.append(old_file_name[jpg_excel])#else:#continueforiinNAME_LIST:print('获取切割后的图片',i)new_img=get_pic(i,(200,675,600,710))print('获取成功！\n百度识图中.............')new_name=get_name_by_pic(new_img)print('成功识别',new_name,'命名中......')new_name=rm_space(new_name)rename(i,new_name+'.jpg')print('命名成功！即将进行第',i,'张')remove('test_tmp.jpg')main()print('图片改名完成，即将对比excel')excel('Sheet1')print('excel对比完成！')system('pause')