# coding:utf-8
"""
 预测客户端
"""
import tornado.ioloop
import tornado.web
import json
from machine_models import predict_model
from loguru import logger

logger.add('./logs/predict_{time}.log', rotation='00:00')


class MainHandler(tornado.web.RequestHandler):
    def post(self):
        request_params = self.request.body.decode('utf-8')
        try:
            request_dict = json.loads(request_params)
            predict_result = predict_model(request_dict)
            response_data = json.dumps(predict_result)
        except Exception as e:
            logger.warning(e)
            response_data = json.dumps({"error_code": 0})
        self.write(response_data)


if __name__ == '__main__':
    application = tornado.web.Application([(r"/jy_machining/predict", MainHandler), ])
    application.listen(8686)
    print('server start')
    tornado.ioloop.IOLoop.instance().start()