p4net 0.2.0__py3-none-any.whl

p4net/__init__.py

@@ -0,0 +1,6 @@
+"""p4net: a P4Runtime-native SDN simulation framework."""
+
+from p4net.network import Network
+
+__all__ = ["Network", "__version__"]
+__version__ = "0.2.0"

p4net/__main__.py

@@ -0,0 +1,6 @@
+"""Allow `python -m p4net <args>` to run the same entry point as the console script."""
+
+from p4net.cli.main import main
+
+if __name__ == "__main__":
+    raise SystemExit(main())

p4net/cli/__init__.py

@@ -0,0 +1,13 @@
+"""Interactive p4net CLI: dispatcher, shell, completer, console script."""
+
+from p4net.cli.dispatcher import CommandDispatcher
+from p4net.cli.exceptions import CLIError, CLIExit, CLIUsageError
+from p4net.cli.shell import P4NetShell
+
+__all__ = [
+    "CLIError",
+    "CLIExit",
+    "CLIUsageError",
+    "CommandDispatcher",
+    "P4NetShell",
+]

p4net/cli/completers.py

@@ -0,0 +1,46 @@
+"""prompt_toolkit completer for the p4net shell."""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+from prompt_toolkit.completion import NestedCompleter
+
+if TYPE_CHECKING:
+    from p4net.cli.dispatcher import CommandDispatcher
+
+
+def build_network_completer(dispatcher: CommandDispatcher) -> NestedCompleter:
+    """Build a NestedCompleter that knows about commands, hosts, and switches."""
+    nested: dict[str, dict[str, None] | None] = {}
+    for cmd in dispatcher.command_names:
+        nested[cmd] = None
+    # `topology` has one sub-verb so far: `graph`.
+    if "topology" in nested:
+        nested["topology"] = {"graph": None}
+    # Host commands: <host> [ping|ping6|cmd|ifconfig|xterm]
+    host_subs = _host_subcommands_for(dispatcher)
+    for name in dispatcher.host_names:
+        nested[name] = dict.fromkeys(host_subs, None)
+    # Switch commands.
+    for name in dispatcher.switch_names:
+        nested[name] = dict.fromkeys(_switch_subcommands_for(dispatcher), None)
+    return NestedCompleter.from_nested_dict(nested)
+
+
+def _switch_subcommands_for(dispatcher: CommandDispatcher) -> list[str]:
+    """Read the dispatcher's switch handler keys at completer-build time.
+
+    Phase-2 ships an empty handler map; phase-3 fills it in. This helper
+    keeps the completer in sync without requiring two import paths.
+    """
+    # `_switch_handlers` is private but stable; the alternative is duplicating
+    # the verb list, which would drift.
+    handlers = getattr(dispatcher, "_switch_handlers", {})
+    return sorted(handlers.keys())
+
+
+def _host_subcommands_for(dispatcher: CommandDispatcher) -> list[str]:
+    """Same idea for host verbs."""
+    handlers = getattr(dispatcher, "_host_handlers", {})
+    return sorted(handlers.keys())