PipeGraphPy 3.0.5__py3-none-win_amd64.whl → 3.0.6__py3-none-win_amd64.whl

PipeGraphPy/__init__.py

@@ -1,6 +1,6 @@
 # coding:utf-8
 
-__version__ = "3.0.5"
+__version__ = "3.0.6"
 
 from PipeGraphPy.core.graph import (
     Graph,

PipeGraphPy/__main__.py

@@ -34,6 +34,7 @@ settings.RUN_MODEL_SAVE_PATH = RUN_MODEL_SAVE_PATH or os.path.join(HOME, "models
 settings.SDK_MODEL_SAVE_PATH = SDK_MODEL_SAVE_PATH or RUN_MODEL_SAVE_PATH
 settings.PREDICT_MODEL_SAVE_PATH = PREDICT_MODEL_SAVE_PATH or RUN_MODEL_SAVE_PATH
 settings.MODULES_CACHE_PATH = MODULES_CACHE_PATH or os.path.join(HOME, "modules")
+sys.path.append(os.path.dirname(settings.MODULES_CACHE_PATH))
 
 # 数据库配置
 USE_DB = os.environ.get("PGP_USE_DB") == "1"

PipeGraphPy/core/anchor.py

@@ -0,0 +1,74 @@
+"""
+锚点(Anchor)模块。
+
+锚点表示节点上的一个输入/输出接口。节点连接时把源节点的输出锚点连到
+目标节点的输入锚点, 边信息同时写入内存图(multi_graph)与数据库。
+"""
+
+from PipeGraphPy.common import multi_graph
+from PipeGraphPy.core.edge import Edge
+from PipeGraphPy.db.models import EdgesTB
+
+
+class Anchor(object):
+    """节点锚点: anc 为锚点编号, node_info 为所属节点信息。"""
+
+    def __init__(self, anc: int, node_info):
+        self.anc = anc
+        self.node_info = node_info
+
+    def __str__(self):
+        return str(self.anc)
+
+    def __repr__(self):
+        return "Anchor(Node(%s), %s)" % (str(self.node_info["id"]), str(self.anc))
+
+    def _add_edge(self, **kwargs):
+        """创建边并挂到所属图上(内存), 不做数据库持久化。
+
+        kwargs: source_id, target_id, source_anchor, target_anchor
+        """
+        if kwargs["source_id"] == kwargs["target_id"]:
+            raise Exception("源节点与目标节点不能是同一节点")
+        if not multi_graph.get(self.node_info["graph_id"]):
+            raise Exception("未找到对应graph信息")
+        multi_graph[self.node_info["graph_id"]]._edges.append(Edge.create(**kwargs))
+
+    def _connect_anchor(self, anchor):
+        """把本输出锚点连接到目标输入锚点(校验同图)。"""
+        if not isinstance(anchor, Anchor):
+            raise Exception("连接对像只能是Anchor对象")
+        if self.node_info["graph_id"] != anchor.node_info["graph_id"]:
+            raise Exception("节点所在图不一致")
+        self._add_edge(
+            source_id=self.node_info["id"],
+            target_id=anchor.node_info["id"],
+            source_anchor=self.anc,
+            target_anchor=anchor.anc,
+        )
+
+    def _break_anchor(self, anchor):
+        """删除与目标锚点之间的边(数据库)。"""
+        EdgesTB.rm(
+            graph_id=self.node_info["graph_id"],
+            source_id=self.node_info["id"],
+            target_id=anchor.node_info["id"],
+            source_anchor=self.anc,
+            target_anchor=anchor.anc,
+        )
+
+    def connect(self, anchors):
+        """连接一个或多个锚点, 任一失败则回滚已建立的连接。
+
+        anchors: Anchor 或 list[Anchor]
+        """
+        if not isinstance(anchors, list):
+            anchors = [anchors]
+        try:
+            for anchor in anchors:
+                self._connect_anchor(anchor)
+            return self
+        except Exception as e:
+            for anchor in anchors:
+                self._break_anchor(anchor)
+            raise e

PipeGraphPy/core/edge.py

@@ -0,0 +1,33 @@
+"""
+边(Edge)模块。
+
+边描述两个节点锚点之间的连接关系, info 中保存:
+    source_id/target_id/source_anchor/target_anchor/graph_id 等。
+"""
+
+from PipeGraphPy.db.models import EdgesTB
+from PipeGraphPy.utils.str_handle import tid
+
+
+class Edge(object):
+    def __init__(self, id, info=None):
+        self.id = id
+        self._info = info
+
+    @property
+    def info(self):
+        """边信息; 为空时从数据库按id加载。"""
+        if not self._info:
+            edge_info = EdgesTB.find_one(id=self.id)
+            if not edge_info:
+                raise ValueError("数据库不存在此边[%s]信息" % self.id)
+            self._info = edge_info
+        return self._info
+
+    @classmethod
+    def create(cls, **kwargs):
+        """创建一个内存边对象(不落库), 自动生成id。"""
+        id = "e%s" % tid()
+        info = {"id": id}
+        info.update(kwargs)
+        return cls(id, info)