acex-devkit 1.8.0__tar.gz → 1.9.0__tar.gz

{acex_devkit-1.8.0 → acex_devkit-1.9.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: acex-devkit
-Version: 1.8.0
+Version: 1.9.0
 Summary: ACE-X DevKit - Development kit for building ACE-X drivers and plugins
 License: AGPL-3.0
 Keywords: automation,devkit,sdk,drivers,plugins

{acex_devkit-1.8.0 → acex_devkit-1.9.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "acex-devkit"
-version = "1.8.0"
+version = "1.9.0"
 description = "ACE-X DevKit - Development kit for building ACE-X drivers and plugins"
 authors = ["Johan Lahti <johan.lahti@acebit.se>"]
 readme = "README.md"

{acex_devkit-1.8.0 → acex_devkit-1.9.0}/src/acex_devkit/drivers/base.py

@@ -1,6 +1,7 @@
 """Driver base classes for ACE-X network element drivers."""
 
 from abc import ABC, abstractmethod
+from contextlib import contextmanager
 from typing import Any, Dict
 
 from acex_devkit.models.node_response import NodeListItem
@@ -43,9 +44,9 @@ class RendererBase(ABC):
 class TransportBase(ABC):
     """Base class for device transport/communication.
 
-    Each method is self-contained — the driver decides internally
-    whether to open/close sessions per call, pool connections, or
-    make stateless requests.
+    Each method is self-contained, but transports may also expose
+    `session()` as a context manager so a caller can run multiple
+    operations against the same physical connection.
 
     Args:
         node: The node instance (identity, hostname, vendor, os, ned_id)
@@ -53,6 +54,16 @@ class TransportBase(ABC):
         **kwargs: Future use (credentials, options, etc.)
     """
 
+    @contextmanager
+    def session(self, connection: ManagementConnection, **kwargs):
+        """Open a reusable session for multiple ops on one device.
+
+        Default is a no-op: methods continue to open/close per call.
+        Drivers that benefit from connection reuse override this to
+        hold a single connection open for the duration of the block.
+        """
+        yield self
+
     @abstractmethod
     def get_config(self, node: NodeListItem, connection: ManagementConnection, **kwargs) -> str:
         """Fetch the full running configuration from a device."""
@@ -67,6 +78,14 @@ class TransportBase(ABC):
         """Run arbitrary commands and return output per command. Opt-in per driver."""
         raise NotImplementedError(f"{self.__class__.__name__} does not implement execute()")
 
+    def get_lldp_neighbors(self, node: NodeListItem, connection: ManagementConnection, **kwargs) -> list[dict]:
+        """Fetch LLDP/CDP neighbor table from a device.
+
+        Returns list of dicts with keys:
+            local_interface, remote_device, remote_interface, discovery_protocol
+        """
+        raise NotImplementedError(f"{self.__class__.__name__} does not implement get_lldp_neighbors()")
+
 
 class NetworkElementDriver:
     """Base class for network element drivers.