[PyQt5-Node-Editor][基础篇]从零开始使用Pyqt5制作节点编辑器(7)——写个socket(插口)

目标

在声明一个Node的时候，可以指定有几个输入socket和几个输出socket

实现

定义节点（改动）

node = Node(self.scene,"这是一个节点",inputs = [1,2,3],outputs = [1])

引入新的输入参数来指定socket的数目
每个socket都是一个Socket类

定义Socket类

node_socket.py

from node_graphics_socket import QDMGraphicsSocket

LEFT_TOP = 1
LEFT_BOTTON = 2
RIGHT_TOP = 3
RIGHT_BOTTON = 4

class Socket():
    def __init__(self, node, index=0,position=LEFT_TOP):

        self.node = node
        self.index = index
        self.position = LEFT_TOP

        self.grSocket = QDMGraphicsSocket(self.node.grNode)