update

ieemoo-ai-predict.py

@@ -24,6 +24,22 @@ from flask import request, Flask, jsonify, request
 #from ocr_system.predict_all import *
 import logging
 
+import logging.config
+from skywalking import agent, config
+SW_SERVER = os.environ.get('SW_AGENT_COLLECTOR_BACKEND_SERVICES')
+SW_SERVICE_NAME = os.environ.get('SW_AGENT_NAME')
+if SW_SERVER and SW_SERVICE_NAME:
+    config.init() #采集服务的地址，给自己的服务起个名称
+    #config.init(collector="123.60.56.51:11800", service='ieemoo-ai-search') #采集服务的地址，给自己的服务起个名称
+    agent.start()
+def setup_logging(path):
+    if os.path.exists(path):
+        with open(path, 'r') as f:
+            config = json.load(f)
+            logging.config.dictConfig(config)
+            logger = logging.getLogger("root")
+    return logger
+logger = setup_logging('utils/logging.json')
 
 app = Flask(__name__)
 # 日志系统配置
@@ -178,7 +194,7 @@ def predict():
     if request.method == 'POST':
         try:
            data = request.get_data()
-           print(data)
+           #print(data)
            data_str = data.decode("utf-8")
            json_data = json.loads(data_str)
            pic = json_data.get("file")
@@ -204,10 +220,12 @@ def predict():
               ocr_res = {"Ocr": {"barcode_rst": "", "barcode_success": "false", "ocr_rst": "", "ocr_success": False}}
               ocr_res.update(tar_res)
            os.remove(os.sep.join(['../module/ieemoo-ai-predict/imgs',code+'.jpg']))
+           logger.info(ocr_res)
            return ocr_res
-        except:
-           return {"Ocr": {"barcode_rst": "", "barcode_success": "false", "ocr_rst": "", "ocr_success": False}, "TargetDetect": {""}, "tag": False}  
+        except Exception as e:
+            logger.warning(e)
+            return {"Ocr": {"barcode_rst": "", "barcode_success": "false", "ocr_rst": "", "ocr_success": False}, "TargetDetect": {""}, "tag": False}  
 if __name__ == '__main__':
         #http_server = WSGIServer(('0.0.0.0', 8000), app)
         #http_server.serve_forever()
-        app.run()
+        app.run(host='192.168.1.142', port=8000)