#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time    : 2023/4/15 11:05
# @Author  : old tom
# @File    : graph_helper.py
# @Project : futool-tiny-datahub
# @Desc    : 血缘关系上图，关系查询
from py2neo import Node, Relationship, Graph, Subgraph
from datahub.metadata.constant.metadata_constant import MetaDataObjType


class MetaDataGraphBuilder(object):
    def __init__(self, graph: Graph):
        self.graph = graph

    def add_view_relation(self, tables: list, view_name: str):
        """
        添加视图关系
        :param tables: 创建视图的数据名称及类型[(名称大写,类型)]
        :param view_name: 视图名
        :return:
        """
        source_nodes = []
        for t in tables:
            # 只添加表名和属性
            source_nodes.append(Node(t[1], name=t[0]))
        source_nodes.append(Node(MetaDataObjType.View.value, name=view_name))
        self._add_one_to_many_relation(source_nodes, 'from')

    def add_table_relation(self):
        pass

    def add_procedure_relation(self, tables: list, procedure_name: str):
        source_nodes = []
        for t in tables:
            # 只添加表名和属性
            source_nodes.append(Node(t[1], name=t[0]))
        source_nodes.append(Node(MetaDataObjType.View.value, name=procedure_name))
        self._add_one_to_many_relation(source_nodes, 'into')

    def _add_one_to_many_relation(self, nodes: [], relation):
        """
        添加一对多关系
        :param one_node: 单节点
        :param other_nodes: 多节点
        :param relation: 关系描述
        :return:
        """
        # 批量创建relation
        relation_ls = []
        for other in nodes[0:-1]:
            relation_ls.append(Relationship(other, relation, nodes[-1]))
        subgraph = Subgraph(nodes=nodes, relationships=relation_ls)
        # 开启事务
        tx = self.graph.begin()
        try:
            tx.create(subgraph)
            self.graph.commit(tx)
        except Exception as e:
            self.graph.rollback(tx)
            print('图创建失败', e)