#!/usr/bin/env python3
# coding: utf-8
# File: bot_chatbot_graph.py
# Author: lhy<lhy_in_blcu@126.com,https://huangyong.github.io>
# Date: 18-10-4

from bot_question_classifier import *
from bot_question_parser import *
from bot_answer_search import *

'''问答类'''
class ChatBotGraph:
    def __init__(self):
        self.classifier = QuestionClassifier()
        self.parser = QuestionPaser()
        self.searcher = AnswerSearcher()

    def chat_main(self, sent,user):
        answer = '抱歉，暂未找寻到您需要的信息，我将继续学习。'
        res_classify = self.classifier.classify(sent,user)
        if not res_classify:
            return answer
        res_sql = self.parser.parser_main(res_classify)
        final_answers = self.searcher.search_main(res_sql)
        if not final_answers:
            return answer
        else:
            return '\n'.join(final_answers)

    def chat_check(self,user,type):
        self.classifier.user_check(user,type)