node_node.py

from node_graphics_node import customgraphics_node
from node_content_widget import customgraphics_nodeContentWdiget
from node_socket import Socket, LEFT_TOP, LEFT_BOTTOM, RIGHT_TOP, RIGHT_BOTTOM


class Node():
    def __init__(self, scene, title="Undefined Node", inputs=[], outputs=[]):
        self.scene = scene
        self.title = title

        self.content = customgraphics_nodeContentWdiget()
        self.graphicNode = customgraphics_node(self)

        self.scene.addNode(self)
        self.scene.graphicScene.addItem(self.graphicNode)

        # self.graphicNode.title = "Edited"
        self.socket_spacing = 22

        # Create sockets for inputs and outputs
        self.inputs     = []
        self.outputs    = []

        counter = 0
        for item in inputs:
            socket = Socket(node=self, index=counter, position=LEFT_BOTTOM)
            counter += 1
            self.inputs.append(socket)

        counter = 0
        for item in outputs:
            socket = Socket(node=self, index=counter, position=RIGHT_TOP)
            counter += 1 
            self.outputs.append(socket)
    
    @property
    def pos(self):
        return self.graphicNode.pos() # QPintF
    
    
    def setPos(self, x, y):
        self.graphicNode.setPos(x, y)

    def getSocketPosition(self, index, position):
        x = 0 if position in (LEFT_TOP, LEFT_BOTTOM) else self.graphicNode.width
        if position in (LEFT_BOTTOM, RIGHT_BOTTOM):
            # Start from bottom
            y = self.graphicNode.height - self.graphicNode.edge_size - self.graphicNode._padding - index * self.socket_spacing
        else:
            # Start from top
            y = self.graphicNode.title_height + self.graphicNode._padding + self.graphicNode.edge_size + index*self.socket_spacing

        return x,y