通过摄像头拍照后识别文字,并将有用的信息存放到excel
import cv2
import requests
import base64
import json
import re
import openpyxl,random
import os
import win32api,win32con
import glob
if os.path.lexists("数据库.xlsx"):
#os.remove("数据库.xlsx")
#print("已经存在数据表")
wb=openpyxl.load_workbook("数据库.xlsx")
ws=wb.active
ws["A1"].value="图片名称"
ws["B1"].value="付款金额"
ws["C1"].value="转账-交易单号"
else:
wb=openpyxl.Workbook()
ws=wb.active
ws["A1"].value="图片名称"
ws["B1"].value="付款金额"
ws["C1"].value="转账-交易单号"
#wb.save("./数据库.xlsx")
litq=[]
lith=[]
litn=[]
stu_num=[]
stu_num2=[]
jinge=""
dh=""
# 获取图像
def get_file_content(file):
with open(file, "rb") as f:
return f.read()
# 获取access_token
def gettoken():
# client_id 为官网获取的AK, client_secret 为官网获取的SK
host = "https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id=PPKWbCWxkIh5kPhaYtoaZ5jr&client_secret=6f2adCZkDY6EOr8Z8FKhjceDwPRr89kQ"
response = requests.get(host)
# if response:
# print(response.json())
return response.json()
# 把图片里的文字识别出来
def img_to_str(filename):
# 通用文字识别(标准版)
# request_url = "https://aip.baidubce.com/rest/2.0/ocr/v1/general_basic"
# 通用文字识别(高精度版)
request_url = "https://aip.baidubce.com/rest/2.0/ocr/v1/accurate_basic"
# 网络图片识别
#request_url = "https://aip.baidubce.com/rest/2.0/ocr/v1/webimage"
# 二进制方式打开图片文件
#request_url ="https://aip.baidubce.com/rest/2.0/realtime_search/v1/logo/add"
f = get_file_content(filename)
img = base64.b64encode(f)
params = {"image": img}
access_token = gettoken()
request_url = request_url + "?access_token=" + access_token["access_token"]
headers = {"content-type": "application/x-www-form-urlencoded"}
response = requests.post(request_url, data=params, headers=headers)
if response:
#print(response.json())
result_list = response.json()
return result_list
def main():
# 创建窗口
cv2.namedWindow("camera", 1)
# IP摄像头
# capture = cv2.VideoCapture("http://admin:admin@192.168.0.100:8081/video")
# 电脑本地摄像头
capture = cv2.VideoCapture(0)
while True:
success, img = capture.read()
cv2.imshow("camera", img)
# 按键
key = cv2.waitKey(10)
if key == 27: # 对应esc键,可以自定义Key值决定键盘退出
break
if key == 32: # 对应空格键,可以自定义键值触发拍照
filename = "frames.jpg" #拍照存放照片的路径
cv2.imwrite(filename, img)
s = img_to_str(filename)
sxtou(s)
print(s)
# 释放摄像头
capture.release()
# 关闭窗口
cv2.destroyWindow("camera")
def sxtou(ss):
for m in range(1,1000):
if ss["words_result"][m]["words"]=="当前状态":
jinge=ss["words_result"][m-1]["words"]
#print(jinge)
if (ss["words_result"][m]["words"]=="交易单号") or (ss["words_result"][m]["words"]=="转账单号"):
dh=ss["words_result"][m+1]["words"]
chuli(jinge,dh)
#print(dh)
#wb.save("./数据库.xlsx")
break
#return jinge,dh
def tupian(tex):
for i in range(1,1000):
if tex["words_result"][i]["words"]=="当前状态":
jine=tex["words_result"][i-1]["words"]
litq.append(jine)
if (tex["words_result"][i]["words"]=="交易单号") or (tex["words_result"][i]["words"]=="转账单号"):
danhao=tex["words_result"][i+1]["words"]
lith.append(danhao)
#wb.save("./数据库.xlsx")
break
#return litq,lith
def to_excel(n,y,m):
for i in range(0,len(n)):
ws.cell(row=i+2,column=1,value=m[i])
ws.cell(row=i+2,column=2,value=n[i])
ws.cell(row=i+2,column=3,value=y[i])
wb.save("./数据库.xlsx")
#遍历邮箱存放地址的函数
def get_filelist(dir):
for home, dirs, files in os.walk(dir):
for dir in dirs:
print(dir)
for filename in files:
print(filename)
litn.append(filename)
fullname = os.path.join(home, filename)
text = img_to_str(fullname)
#print(type(text))
tupian(text)
to_excel(litq,lith,litn)
#print(text)
print(fullname)
def chuli(str_a,str_b):
workbook1=openpyxl.load_workbook("数据库.xlsx")
#选定目标sheet
worksheet1 = workbook1.active
py2=False
for cell in worksheet1["B"]:
#print(cell.value)
stu_num.append(cell.value)#这里用循环把A列每个cell的值写入开始定义的空列表
for cell2 in worksheet1["C"]:
#print(cell2.value)
stu_num2.append(cell2.value)#这里用循环把A列每个cell的值写入开始定义的空列表
for i in range(0,len(stu_num)):
if (str_a==stu_num[i]) and (str_b==stu_num2[i]):
py2=True
if py2:
win32api.MessageBox(0, "数据库里有完全相同的数据,检验通过!!", "提示", win32con.MB_OK)
#("数据库里有完全相同的数据,检验通过!!")
else:
win32api.MessageBox(0, "数据库没有完全相同的数据,检验不通过!!", "提示", win32con.MB_OK)
if __name__ == "__main__":
path_file_number=glob.glob("./邮箱图片存放")
if len(stu_num)-1!=path_file_number:
get_filelist("D:修改邮箱图片存放")
main()