NMSA: Add rollback() and transaction() methods

1 files changed, 24 insertions, 0 deletions

diff --git a/ncserver/module/nms_ifupdownng.py b/ncserver/module/nms_ifupdownng.py
index 8c92717..0c03fd9 100644
--- a/ncserver/module/nms_ifupdownng.py
+++ b/ncserver/module/nms_ifupdownng.py
@@ -10,6 +10,8 @@ with this source distribution for more information.
 SPDX-License-Identifier: NCSA
 """
 
+from contextlib import contextmanager
+
 import logging
 import pathlib
 import socket
@@ -573,6 +575,28 @@ def commit():
     _TRANSACTION = False
 
 
+def rollback():
+    """Roll back outstanding operations."""
+    global _TRANSACTION  # pylint: disable=W0603
+
+    _load_config()
+    _load_sysctl_cfg()
+
+    _TRANSACTION = False
+
+
+@contextmanager
+def transaction():
+    begin_transaction()
+    try:
+        yield None
+    except:
+        rollback()
+        raise
+    else:
+        commit()
+
+
 def get_param(iface: str, parameter: str):
     """Retrieve the parameter for the specified interface."""
     _load_config()