Skip to content

lichenggang / bom_identify

Go to a project
Commit 0d5d0364 by lichenggang
Options

未识别到参数或数量列先这么写着

parent 79304ec4
Inline Side-by-side
Showing with 15 additions and 7 deletions
classify_server.py
...@@ -22,12 +22,17 @@ class Classify(classify_pb2_grpc.classifyServicer): ...@@ -22,12 +22,17 @@ class Classify(classify_pb2_grpc.classifyServicer):
res = self.predictorfac.predict(dic_data) res = self.predictorfac.predict(dic_data)
res['status'] = 1 res['status'] = 1
for i in ['参数', '数量']: if '数量' not in str(res):
if i not in str(res):
res = { res = {
'status': 0, 'status': 0,
'err_msg': f'未识别到{i}列' 'err_msg': f'未识别到数量列'
} }
if '参数' not in str(res):
res = {
'status': 0,
'err_msg': f'未识别到参数列'
}
except json.decoder.JSONDecodeError: except json.decoder.JSONDecodeError:
res = self.predictorfac.predict(request.keyword) res = self.predictorfac.predict(request.keyword)
res['status'] = 1 res['status'] = 1
......
http_server.py
...@@ -41,12 +41,15 @@ class DictHandler(BaseHandler): ...@@ -41,12 +41,15 @@ class DictHandler(BaseHandler):
dic_data = json.loads(self.request.body) dic_data = json.loads(self.request.body)
res = self.predictorfac.predict(dic_data) res = self.predictorfac.predict(dic_data)
res['status'] = 1 res['status'] = 1
if '数量' not in str(res):
for i in ['参数', '数量']: res = {
if i not in str(res): 'status': 0,
'err_msg': f'未识别到数量列'
}
if '参数' not in str(res):
res = { res = {
'status': 0, 'status': 0,
'err_msg': f'未识别到{i}列' 'err_msg': f'未识别到参数列'
} }
except json.decoder.JSONDecodeError: except json.decoder.JSONDecodeError:
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or sign in to comment