From a113d11811f7eb0932d5790d67394a283b6cd6d7 Mon Sep 17 00:00:00 2001
From: hjdhnx <hjd124579>
Date: Fri, 30 Sep 2022 19:37:09 +0800
Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0base64=E9=AA=8C=E8=AF=81?=
 =?UTF-8?q?=E7=A0=81=E8=AF=86=E5=88=AB=E6=8E=A5=E5=8F=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 controllers/parse.py | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/controllers/parse.py b/controllers/parse.py
index eee5d68..15223fb 100644
--- a/controllers/parse.py
+++ b/controllers/parse.py
@@ -6,6 +6,7 @@
 from flask import Blueprint, jsonify,redirect
 from utils.web import getParmas,get_interval
 import os
+from utils.cfg import cfg
 from utils.log import logger
 from utils.encode import OcrApi
 from utils.pyctx import py_ctx,getPreJs,runJScode,JsObjectWrapper,PyJsString,parseText,jsoup,time
@@ -95,16 +96,23 @@ def parse_home(filename):
 
 @parse.route('/ocr',methods=['POST'])
 def base64_ocr():
+    ocr_api = cfg.OCR_API
+    # print(ocr_api)
     # print('params:',getParmas())
     img = getParmas('img')
     # print(img)
-    img_bytes = base64.b64decode(img)
+    if not img:
+        return R.failed('识别失败:缺少img参数')
+    try:
+        img_bytes = base64.b64decode(img)
+    except:
+        return R.failed('识别失败:img参数不是正确的base64格式')
     # print(img_bytes)
     img_path = 'txt/pluto'
     os.makedirs(img_path,exist_ok=True)
     with open(f'{img_path}/yzm.png','wb+') as f:
         f.write(img_bytes)
-    ocr = OcrApi('http://dm.mudery.com:10000')
+    ocr = OcrApi(ocr_api)
     code = ocr.classification(img_bytes)
     resp = R.success('识别成功',code)
     print(resp.json)
-- 
GitLab