python 读取发票信息；python 电子发票；python 纸质发票；python 发票；

利用百度免费的api提取发票信息。可以提取pdf或者拍照的png图片。

  1. import requests
  2. import base64
  3. # pip install PyMuPDF
  4. # 安装PyMuPDF才能用fitz
  5. import fitz
  8. # fileName : string
  9. # 例如："test.png"，也可以是pdf
  10. class InvoiceParser:
  11.     def __init__(self, fileName):
  12.         self.fileName = fileName
  13.         self.response = {}
  14.         self.invoiceInfo = {"发票号码": -1,
  15.                             "开票日期": -1,
  16.                             "发票金额(含税)": -1,
  17.                             "销售方名称": -1,
  18.                             "购买物品": -1}
  19.         self._set_up()
  21.     def _set_up(self):
  22.         self.get_response()
  23.         # 录入二维码发票信息
  24.         self.invoiceInfo["发票号码"] = self.response["InvoiceNum"]
  25.         self.invoiceInfo["开票日期"] = self.response["InvoiceDate"]
  26.         self.invoiceInfo["发票金额(含税)"] = self.response["AmountInFiguers"]
  27.         self.invoiceInfo["销售方名称"] = self.response["SellerName"]
  28.         self.invoiceInfo["购买物品"] = self.response["CommodityName"]
  30.     # 通过百度云识别发票信息
  31.     def get_response(self):
  32.         # 发票文件转换为二进制
  33.         f = open(self.fileName, 'rb')
  34.         invoiceFile = base64.b64encode(f.read())
  35.         # 设置url
  36.         request_url = "https://aip.baidubce.com/rest/2.0/ocr/v1/vat_invoice"
  37.         access_token = "自己获取"
  38.         request_url = request_url + "?access_token=" + access_token
  39.         # 设置其他参数
  40.         headers = {'content-type': 'application/x-www-form-urlencoded'}
  41.         # 默认是图片
  42.         params = {"image": invoiceFile}
  43.         # 如果是pdf转pdf
  44.         if ".pdf" in self.fileName:
  45.             params = {"pdf_file": invoiceFile}
  46.         response = requests.post(request_url, data=params, headers=headers)
  47.         # 保存到self
  48.         self.response = response.json()['words_result']
  49.         return self.response
  52. def GetAccessToken():
  53.     apiKey = "自己获取"
  54.     secretKet = "自己获取"
  55.     url = "https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id={}&client_secret={}&".format(
  56.         apiKey, secretKet)
  57.     response = requests.post(url=url)
  58.     return eval(response.text)["access_token"]
  61. if __name__ == '__main__':
  62.     # 获取get_response()函数中的access_token
  63.     # 获取一次就可以注释掉了
  64.     # print(GetAccessToken())
  65.     tmp = InvoiceParser("u盘.png")
  66.     print(str(tmp.invoiceInfo))
  67.     # print(str(tmp.response))

参考链接

  1. 百度AI增值税发票识别
  2. 文字识别OCR文档