Connect actions of top-level sim components

This commit is contained in:
Marek Wolan
2023-08-16 16:45:52 +01:00
parent d972fcc07e
commit ced45d4275
3 changed files with 72 additions and 1 deletions

View File

@@ -0,0 +1,23 @@
from typing import Dict
from primaite.simulator.core import Action, ActionManager, AllowAllValidator, SimComponent
from primaite.simulator.network.hardware.base import Link, Node
class NetworkContainer(SimComponent):
"""TODO."""
nodes: Dict[str, Node] = {}
links: Dict[str, Link] = {}
def __init__(self, **kwargs):
super().__init__(**kwargs)
self.action_manager = ActionManager()
self.action_manager.add_action(
"node",
Action(
func=lambda request, context: self.nodes[request.pop(0)].apply_action(request, context),
validator=AllowAllValidator(),
),
)