Add more type hints.

Author: Poikilos

openlcb/node.py

@@ -15,6 +15,7 @@
 
 from enum import Enum
 from typing import Set
+from openlcb.nodeid import NodeID
 from openlcb.pip import PIP
 from openlcb.snip import SNIP
 from openlcb.localeventstore import LocalEventStore
@@ -40,11 +41,12 @@ class Node:
         events (LocalEventStore): The store for local events associated
             with the node.
     """
-    def __init__(self, nodeID, snip: SNIP = None, pipSet: Set[PIP] = None):
-        self.id = nodeID
-        self.snip = snip
+    def __init__(self, nodeID: NodeID, snip: SNIP = None,
+                 pipSet: Set[PIP] = None):
+        self.id: NodeID = nodeID
+        self.snip: SNIP = snip
         if snip is None : self.snip = SNIP()
-        self.pipSet = pipSet
+        self.pipSet: Set[PIP] = pipSet
         if pipSet is None : self.pipSet = set([])
         self.state = Node.State.Uninitialized
         self.events = LocalEventStore()

openlcb/nodestore.py

@@ -1,4 +1,5 @@
 from typing import (
+    Dict,
     List,  # in case list doesn't support `[` in this Python version
     Union,  # in case `|` doesn't support 'type' in this Python version
 )
@@ -18,7 +19,7 @@ class NodeStore :
     '''
 
     def __init__(self) :
-        self.byIdMap: dict = {}
+        self.byIdMap: Dict[NodeID, Node] = {}
         self.nodes: List[Node] = []
         self.processors: List[Processor] = []
 
@@ -34,10 +35,10 @@ def store(self, node: Node) :
         self.nodes.sort(key=lambda x: x.snip.userProvidedNodeName,
                         reverse=True)
 
-    def isPresent(self, nodeID) :
+    def isPresent(self, nodeID: NodeID) -> bool:
         return self.byIdMap.get(nodeID) is not None
 
-    def asArray(self) :
+    def asArray(self) -> List[Node]:
         return [self.byIdMap[i] for i in self.byIdMap]
 
     # Retrieve a Node's content from the store