graphQa/actions/api/question_parser.py

class QuestionParser:

    '''构建实体节点'''
    def build_entitydict(self, args):
        entity_dict = {}
        for arg, types in args.items():
            for type in types:
                if type not in entity_dict:
                    entity_dict[type] = [arg]
                else:
                    entity_dict[type].append(arg)

        return entity_dict

    '''解析主函数'''
    def parser_main(self, res_classify):
        args = res_classify['args']                          # 解析问句中的实体
        entity_dict = self.build_entitydict(args)            # 创建一个实体的字典
        question_types = res_classify['question_types']      # 创建意图的字典并将一个问句中的意图放入到res_classify中去
        sqls = []
        for question_type in question_types:
            sql_ = {}
            sql_['question_type'] = question_type
            sql = []

            if question_type == 'query_des': # 故障时间
                sql = self.sql_transfer(question_type, entity_dict.get('fault_des'))

            elif question_type == 'query_hmc': # 故障时间
                sql = self.sql_transfer(question_type, entity_dict.get('fault_hmc'))

            elif question_type == 'query_obj': # 故障时间
                sql = self.sql_transfer(question_type, entity_dict.get('fault_obj'))

            elif question_type == 'query_x_obj': # 故障时间
                sql = self.sql_transfer(question_type, entity_dict.get('fault_x_obj'))

            elif question_type == 'query_s_sys': # 故障时间
                sql = self.sql_transfer(question_type, entity_dict.get('s_sys'))

            elif question_type == 'query_system': # 故障时间
                sql = self.sql_transfer(question_type, entity_dict.get('system'))

            elif question_type == 'fault_des':   # 故障现象
                sql = self.sql_transfer(question_type, entity_dict.get('fault_hmc'))

            elif question_type == 'fault_excluds': # 故障时间
                sql = self.sql_transfer(question_type, entity_dict.get('fault_hmc'))

            elif question_type == 'des_excluds':  # 故障所属系统
                sql = self.sql_transfer(question_type, entity_dict.get('fault_des'))

            elif question_type == 'fault_obj':  # 专业
                sql = self.sql_transfer(question_type, entity_dict.get('fault_hmc'))

            elif question_type == 'hmc_system':  # 故障产品或部位
                sql = self.sql_transfer(question_type, entity_dict.get('fault_hmc'))

            elif question_type == 'sys_s_sys':  # 故障原因
                sql = self.sql_transfer(question_type, entity_dict.get('system'))

            elif question_type == 'sys_fault_obj':  # 排除方法
                sql = self.sql_transfer(question_type, entity_dict.get('s_sys'))

            elif question_type == 'fault_obj_sys':  # 排除方法
                sql = self.sql_transfer(question_type, entity_dict.get('fault_obj'))

            elif question_type == 'fault_obj_fault_x_obj':  # 排除方法
                sql = self.sql_transfer(question_type, entity_dict.get('fault_obj'))


            if sql:
                sql_['sql'] = sql

                sqls.append(sql_)

        return sqls

    '''针对不同的问题，分开进行处理'''
    def sql_transfer(self, question_type, entities):
        if not entities:
            return []

        # 查询语句
        sql = []
        # 查询故障的故障现象
        if question_type == 'query_des':
            sql = ["MATCH (m:故障描述)-[r]->(n) WHERE m.name = '{0}' return m.name, r.name, n.name, n.doc_id, n.doc_name, n.filePage".format(i) for i in entities]

        elif question_type == 'query_hmc':
            sql = ["MATCH (m:HMC)-[r]->(n) WHERE m.name = {0} return m.name, r.name, n.name, n.doc_id, n.doc_name, n.filePage".format(i) for i in entities]

        elif question_type == 'query_obj':
            sql = ["MATCH (m:成品)-[r]->(n) WHERE m.name = '{0}' return m.name, r.name, n.name, n.doc_id, n.doc_name, n.filePage".format(i) for i in entities]

        elif question_type == 'query_x_obj':
            sql = ["MATCH (m)-[r]->(n:型号) WHERE n.name = '{0}' return m.name, r.name, n.name, n.doc_id, n.doc_name, n.filePage".format(i) for i in entities]

        elif question_type == 'query_s_sys':
            sql = ["MATCH (m:子系统)-[r]->(n) WHERE m.name = '{0}' return m.name, r.name, n.name, n.doc_id, n.doc_name, n.filePage".format(i) for i in entities]

        elif question_type == 'query_system':
            sql = ["MATCH (m:系统)-[r]->(n) WHERE m.name = '{0}' return m.name, r.name, n.name, n.doc_id, n.doc_name, n.filePage".format(i) for i in entities]

        elif question_type == 'fault_des':
            sql = ["MATCH (m:HMC)-[r:故障描述]->(n:故障描述) WHERE m.name = {0} return m.name, r.name, n.name, n.doc_id, n.doc_name, n.filePage".format(i) for i in entities]
            # MATCH (m:故障名称)-[r:原因]->(n:原因) where m.name = '转速指示器显示与规程规定不符' return m.name, r.name, n.name

        # 查询故障的发生时间
        elif question_type == 'fault_excluds':
            sql = ["MATCH (m:HMC)-[r:维修策略]->(n:维修策略) WHERE m.name = {0} " \
                   "OPTIONAL MATCH(a)-[b]->(c) WHERE a.type=n.name return m.name, n.name, r.name, a.name, b.name, c.name, n.doc_id, n.doc_name, n.filePage".format(i) for i in entities]

        # 查询故障的所属系统
        elif question_type == 'des_excluds':
            sql = ["MATCH (m:故障描述)-[r:维修策略]->(n:维修策略) WHERE m.name = '{0}' " \
                   "OPTIONAL MATCH(a)-[b]->(c) WHERE a.type=n.name return m.name, n.name, r.name, a.name, b.name, c.name, n.doc_id, n.doc_name, n.filePage".format(i) for i in entities]

        elif question_type == 'fault_obj':
            sql = ["MATCH (m:HMC)-[r:成品]->(n:成品) WHERE m.name = {0} return m.name, r.name, n.name, n.doc_id, n.doc_name, n.filePage".format(i) for i in entities]

        # 查询故障的产品部位
        elif question_type == 'hmc_system':
            sql = ["MATCH (m:HMC)-[r:系统]->(n:系统) WHERE m.name = {0} return m.name, r.name, n.name, n.doc_id, n.doc_name, n.filePage".format(i) for i in entities]

        # 查询故障的故障原因
        elif question_type == 'sys_s_sys':
            sql = ["MATCH (m:系统)-[r:子系统]->(n:子系统) WHERE m.name = '{0}' return m.name, r.name, n.name, n.doc_id, n.doc_name, n.filePage".format(i) for i in entities]

        # 查询故障的解决办法
        elif question_type == 'sys_fault_obj':
            # sql = ["MATCH (m:故障名称)-[r:排除方法]->(n:排除方法) WHERE m.name = '{0}' return m.name, r.name, n.name".format(i) for i in entities]
            sql = ["MATCH (m:子系统)-[r:成品]->(n:成品) WHERE m.name = '{0}' return m.name, r.name, n.name, n.doc_id, n.doc_name, n.filePage".format(i) for i in entities]

        elif question_type == 'fault_obj_sys':
            # sql = ["MATCH (m:故障名称)-[r:排除方法]->(n:排除方法) WHERE m.name = '{0}' return m.name, r.name, n.name".format(i) for i in entities]
            sql = ["MATCH (n:子系统)-[r:成品]->(m:成品) WHERE m.name = '{0}' return m.name, r.name, n.name, m.doc_id, m.doc_name, m.filePage".format(i) for i in entities]

        elif question_type == 'fault_obj_fault_x_obj':
            sql = ["MATCH (m:成品)-[r:型号]->(n:型号) WHERE m.name = '{0}' return m.name, r.name, n.name, n.doc_id, n.doc_name, n.filePage".format(i) for i in entities]

        return sql


# ["MATCH (m:{type:'{0}'})-[r:{type:'{0}'}]->(n:{type:'{0}')  return m.name, r.name, n.name".format(i) for i in entities]

if __name__ == '__main__':
    handler = QuestionParser()