updated session.add_node to use better type hinting and force usage of classes directly, instead of indirectly through NodeTypes

daemon/examples/python/switch_inject.py

@@ -7,8 +7,11 @@ same CoreEmu instance the GUI is using.
 
 import logging
 
+from core.emulator.coreemu import CoreEmu
 from core.emulator.emudata import IpPrefixes
-from core.emulator.enumerations import EventTypes, NodeTypes
+from core.emulator.enumerations import EventTypes
+from core.nodes.base import CoreNode
+from core.nodes.network import SwitchNode
 
 NODES = 2
 
@@ -18,18 +21,18 @@ def main():
     prefixes = IpPrefixes("10.83.0.0/16")
 
     # create emulator instance for creating sessions and utility methods
-    coreemu = globals()["coreemu"]
+    coreemu: CoreEmu = globals()["coreemu"]
     session = coreemu.create_session()
 
     # must be in configuration state for nodes to start, when using "node_add" below
     session.set_state(EventTypes.CONFIGURATION_STATE)
 
     # create switch network node
-    switch = session.add_node(_type=NodeTypes.SWITCH)
+    switch = session.add_node(SwitchNode)
 
     # create nodes
     for _ in range(NODES):
-        node = session.add_node()
+        node = session.add_node(CoreNode)
         interface = prefixes.create_interface(node)
         session.add_link(node.id, switch.id, interface_one=interface)