http_server.py

#!/usr/bin/env python
# -*- coding:utf-8 -*-
import json
from urllib.parse import unquote

import tornado.web
import tornado.ioloop

from utils.log_manager import bom_log
from predic_fac import PredictorFac
from utils.status import code2msg


class BaseHandler(tornado.web.RequestHandler):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.predictorfac = PredictorFac()

class KwHandler(BaseHandler):
    async def get(self):
        target = unquote(self.get_argument('keyword'))
        predict_type = unquote(self.get_argument('type', 'all'))
        bom_log.info(f'http收到识别关键词: {target}, 预测类型为{predict_type}')
        code, res = self.predictorfac.predict(target, predict_type)
        if code != 1:
            res = code2msg(code)
            bom_log.error(f'http关键词识别报错: ' + res['message'])
        else:
            res['status'] = code
        self.write(res)


class DictHandler(BaseHandler):
    async def post(self):
        try:
            bom_log.info(f'http收到识别体: {self.request.body}')
            dic_data = json.loads(self.request.body)
            code, res = self.predictorfac.predict(dic_data)
        except json.decoder.JSONDecodeError:
            code, res = self.predictorfac.predict(self.request.body)
        if code != 1:
            res = code2msg(code)
            bom_log.error(f'http识别报错: ' + res['message'])
        else:
            res['status'] = code
        self.write(res)

def gen_app():
    return tornado.web.Application(handlers=[(k, v) for k, v in register_tornado_handlers.items()])


register_tornado_handlers = {'/kw': KwHandler, '/dict': DictHandler}


def http_server(port):
    app = gen_app()
    app.listen(port)
    tornado.ioloop.IOLoop.current().start()


if __name__ == '__main__':
    port = 50052
    http_server(port)