small node and port clean up.

Author: jchanvfx

.gitignore

@@ -12,5 +12,8 @@ docs/_build/
 # test stuff
 dist
 build
+
+tests/*
+
 *.egg-info
 *.json

NodeGraphQt/base/node.py

@@ -529,6 +529,15 @@ def inputs(self):
         """
         return {p.name(): p for p in self._inputs}
 
+    def input_ports(self):
+        """
+        Return all input ports.
+
+        Returns:
+            list[NodeGraphQt.Port]: node input ports.
+        """
+        return self._inputs
+
     def outputs(self):
         """
         Returns all the output ports from the node.
@@ -538,6 +547,15 @@ def outputs(self):
         """
         return {p.name(): p for p in self._outputs}
 
+    def output_ports(self):
+        """
+        Return all output ports.
+
+        Returns:
+            list[NodeGraphQt.Port]: node output ports.
+        """
+        return self._outputs
+
     def input(self, index):
         """
         Return the input port with the matching index.

NodeGraphQt/base/port.py

@@ -3,6 +3,7 @@
                                        PortDisconnectedCmd,
                                        PortVisibleCmd)
 from NodeGraphQt.base.model import PortModel
+from NodeGraphQt.constants import IN_PORT, OUT_PORT
 
 
 class Port(object):
@@ -118,9 +119,9 @@ def connected_ports(self):
         for node_id, port_names in self.model.connected_ports.items():
             for port_name in port_names:
                 node = graph.get_node_by_id(node_id)
-                if self.type_() == 'in':
+                if self.type_() == IN_PORT:
                     ports.append(node.outputs()[port_name])
-                elif self.type_() == 'out':
+                elif self.type_() == OUT_PORT:
                     ports.append(node.inputs()[port_name])
         return ports
 

example.py

@@ -1,7 +1,6 @@
 #!/usr/bin/python
 # -*- coding: utf-8 -*-
 import os
-import sys
 
 from NodeGraphQt import (NodeGraph,
                          BaseNode,
@@ -34,7 +33,7 @@ def __init__(self):
 
 
 if __name__ == '__main__':
-    app = QtWidgets.QApplication(sys.argv)
+    app = QtWidgets.QApplication([])
 
     # create node graph.
     graph = NodeGraph()

setup.py

@@ -15,9 +15,9 @@
     'supports PySide & PySide2'
 )
 classifiers = [
-    'Programming Language :: Python :: 3.7',
-    'License :: OSI Approved :: MIT License',
     'Operating System :: OS Independent',
+    'License :: OSI Approved :: MIT License',
+    'Programming Language :: Python :: 3.7',
 ]
 
 setuptools.setup(