"""
@author: luojunhui
"""


class NLPClass(object):
    """
    NLP Dealing
    """

    def __init__(self, params):
        self.model = None
        self.method = None
        self.function = None
        self.params = params

    async def check_params(self):
        """
        校验参数
        :return:
        """
        try:
            self.function = self.params['function']
            self.method = self.params['method']
            self.model = self.params['model']
            return None
        except Exception as e:
            response = {
                "error": "params error",
                "msg": str(e)
            }
            return response

    async def nlper(self):
        """
        NLP process
        :return:
        """
        match self.function:
            case "keywords":
                print("keywords")
            case "sensitive":
                print("sensitive")
            case "similarity":
                print("similarity")
    async def deal(self):
        """
        处理代码
        :return:
        """
        param_error = await self.check_params()
        if param_error:
            return param_error
        else:
            return await self.nlper()