Fix typing warnings

.isort.cfg

@@ -6,5 +6,5 @@ include_trailing_comma = True
 force_grid_wrap=0
 use_parentheses=True
 
-known_third_party = PIL,PyQt5,PySide2,coloredlogs,graphene,numpy,pandas,pylab,pyqtconsole,pyqtgraph,pytest,pytestqt,recommonmark,scipy,setuptools,yaml
+known_third_party = PIL,PyQt5,PySide2,black,coloredlogs,graphene,numpy,pandas,pylab,pyqtconsole,pyqtgraph,pytest,pytestqt,recommonmark,scipy,setuptools,yaml
 

nodedge/editor_widget.py

@@ -23,6 +23,7 @@
 from nodedge.graphics_view import GraphicsView
 from nodedge.node import Node
 from nodedge.scene import InvalidFile, Scene
+from nodedge.socket_type import SocketType
 from nodedge.utils import dumpException
 
 
@@ -269,13 +270,22 @@ def addNodes(self) -> None:
         :class:`~nodedge.edge.Edge`.
         """
         node1 = Node(
-            self.scene, "Node 1", inputSocketTypes=[1, 2, 3], outputSocketTypes=[1]
+            self.scene,
+            "Node 1",
+            inputSocketTypes=[SocketType.Any, SocketType.Number, SocketType.String],
+            outputSocketTypes=[SocketType.Any],
         )
         node2 = Node(
-            self.scene, "Node 2", inputSocketTypes=[1, 2, 3], outputSocketTypes=[1]
+            self.scene,
+            "Node 2",
+            inputSocketTypes=[SocketType.Any, SocketType.Number, SocketType.String],
+            outputSocketTypes=[SocketType.Any],
         )
         node3 = Node(
-            self.scene, "Node 3", inputSocketTypes=[1, 2, 3], outputSocketTypes=[1]
+            self.scene,
+            "Node 3",
+            inputSocketTypes=[SocketType.Any, SocketType.Number, SocketType.String],
+            outputSocketTypes=[SocketType.Any],
         )
 
         node1.pos = (-350, -250)

nodedge/graphics_socket.py

@@ -89,7 +89,8 @@ def socketType(self) -> SocketType:
         :return: socket type
         :rtype: ``SocketType``
         """
-        return self.socket.socketType
+
+        return self.socket.socketType  # type: ignore
 
     # noinspection PyAttributeOutsideInit
     def updateSocketType(self) -> None:

nodedge/node.py

@@ -48,25 +48,25 @@ def __init__(
         self,
         scene: "Scene",  # type: ignore
         title: str = "Undefined node",
-        inputSocketTypes: Collection[int] = (),
-        outputSocketTypes: Collection[int] = (),
+        inputSocketTypes: Collection[SocketType] = (),
+        outputSocketTypes: Collection[SocketType] = (),
     ):
         """
         :param scene: reference to the :class:`~nodedge.scene.Scene`
         :type scene: :class:`~nodedge.scene.Scene`
         :param title: node title shown in scene
         :type title: ``str``
         :param inputSocketTypes: list of types of the input `Sockets`
-        :type inputSocketTypes: ``Collection[int]``
+        :type inputSocketTypes: ``Collection[SocketType]``
         :param outputSocketTypes: list of types of the output `Sockets`
-        :type outputSocketTypes: ``Collection[int]``
+        :type outputSocketTypes: ``Collection[SocketType]``
         """
 
         super().__init__()
         self._title: str = title
         self.scene: "Scene" = scene  # type: ignore
-        self.inputSocketTypes = inputSocketTypes
-        self.outputSocketTypes = outputSocketTypes
+        self.inputSocketTypes: Collection[SocketType] = inputSocketTypes
+        self.outputSocketTypes: Collection[SocketType] = outputSocketTypes
 
         self.__logger = logging.getLogger(__file__)
         self.__logger.setLevel(logging.INFO)

nodedge/scene_item_detail_widget.py

@@ -15,7 +15,7 @@ class SceneItemDetailWidget(QWidget):
     def __init__(self, parent=None):
         super().__init__(parent)
         self.parent = parent
-        self.layout = QGridLayout()
+        self.layout: QGridLayout = QGridLayout()
         self.layout.setAlignment(Qt.AlignTop)
         self.setAutoFillBackground(True)
 
@@ -37,7 +37,10 @@ def addRow(self, title: str, edit: bool = False):
         stringLabel.setAlignment(Qt.AlignTop)
         stringLabel.setFixedHeight(30)
 
-        valueWidget: QWidget = QLineEdit("") if edit is True else QLabel("")
+        if edit is True:
+            valueWidget: QWidget = QLineEdit("")
+        else:
+            valueWidget = QLabel("")
 
         valueWidget.setSizePolicy(
             QSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)