..
[ncclient] / ncclient / operations / edit.py
index afae257..9794bb4 100644 (file)
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from ncclient.rpc import RPC
-from ncclient.content import iselement
+from ncclient.xml_ import *
+
+from rpc import RPC
 
 import util
 
+import logging
+
+logger = logging.getLogger("ncclient.operations.edit")
+
+"Operations related to changing device configuration"
+
 class EditConfig(RPC):
-    
-    SPEC = {
-        'tag': 'edit-config',
-        'children': [ ]
-    }
-    
-    def request(self, target=None, target_url=None, config=None,
-                default_operation=None, test_option=None, error_option=None):
-        util.one_of(target, target_url)
-        spec = EditConfig.SPEC.copy()
-        params = spec['children']
-        params.append({'tag': 'target', 'children': util.store_or_url(target, target_url)})
-        params.append({'tag': 'config', 'children': config})
-        if default_operation is not None:
-            params.append({'tag': 'default-operation', 'text': default_operation})
-        if test_option is not None:
-            params.append({'tag': 'test-option', 'text': test_option})
+    "`edit-config` RPC"
+
+    def request(self, target, config, default_operation=None, test_option=None, error_option=None):
+        """Loads all or part of the specified *config* to the *target* configuration datastore.
+
+        *target* is the name of the configuration datastore being edited
+
+        *config* is the configuration, which must be rooted in the `config` element. It can be specified either as a string or an :class:`~xml.etree.ElementTree.Element`.
+
+        *default_operation* if specified must be one of { `"merge"`, `"replace"`, or `"none"` }
+
+        *test_option* if specified must be one of { `"test_then_set"`, `"set"` }
+
+        *error_option* if specified must be one of { `"stop-on-error"`, `"continue-on-error"`, `"rollback-on-error"` }
+
+        The `"rollback-on-error"` *error_option* depends on the `:rollback-on-error` capability.
+        """
+        node = new_ele("edit-config")
+        node.append(util.datastore_or_url("target", target, self._assert))
         if error_option is not None:
-            params.append({'tag': 'test-option', 'text': test_option})
+            if error_option == "rollback-on-error":
+                self._assert(":rollback-on-error")
+            sub_ele(node, "error-option").text = error_option
+        if test_option is not None:
+            self._assert(':validate')
+            sub_ele(node, "test-option").text = test_option
+        if default_operation is not None:
+        # TODO: check if it is a valid default-operation
+            sub_ele(node, "default-operation").text = default_operation
+        node.append(validated_element(config, ("config", qualify("config"))))
+        return self._request(node)
+
 
 class DeleteConfig(RPC):
-    
-    SPEC = {
-        'tag': 'delete-config',
-        'children': [ { 'tag': 'target', 'children': None } ]
-    }
-    
-    def request(self, target=None, target_url=None):
-        spec = DeleteConfig.SPEC.copy()
-        spec['children'][0]['children'] = util.store_or_url(target, target_url)
-        return self._request(spec)
+    "`delete-config` RPC"
+
+    def request(self, target):
+        """Delete a configuration datastore.
+
+        :param target: name or URL of configuration datastore to delete
+        :type: :ref:`srctarget_params`"""
+        node = new_ele("delete-config")
+        node.append(util.datastore_or_url("target", target, self._assert))
+        return self._request(node)
 
 
 class CopyConfig(RPC):
-    
-    SPEC = {
-        'tag': 'copy-config',
-        'children': [
-            { 'tag': 'source', 'children': {'tag': None } },
-            { 'tag': 'target', 'children': {'tag': None } }
-        ]
-    }
-    
-    def request(self, source=None, source_url=None, target=None, target_url=None):
-        spec = CopyConfig.SPEC.copy()
-        spec['children'][0]['children'] = util.store_or_url(source, source_url)
-        spec['children'][1]['children'] = util.store_or_url(target, target_url)
-        return self._request(spec)
+    "`copy-config` RPC"
+
+    def request(self, source, target):
+        """Create or replace an entire configuration datastore with the contents of another complete
+        configuration datastore.
+
+        :param source: configuration datastore to use as the source of the copy operation or `config` element containing the configuration subtree to copy
+        :type source: :ref:`srctarget_params`
+
+        :param target: configuration datastore to use as the destination of the copy operation
+        :type target: :ref:`srctarget_params`"""
+        node = new_ele("copy-config")
+        node.append(util.datastore_or_url("target", target, self._assert))
+        node.append(util.datastore_or_url("source", source, self._assert))
+        return self._request(node)
 
 
 class Validate(RPC):
-    
-    'config attr shd not include <config> root'
-    
+    "`validate` RPC. Depends on the `:validate` capability."
+
     DEPENDS = [':validate']
-    
-    SPEC = {
-        'tag': 'validate',
-        'children': []
-    }
-    
-    def request(self, source=None, config=None):
-        util.one_of(source, capability)
-        spec = SPEC.copy()
-        if source is not None:
-            spec['children'].append({
-                'tag': 'source', 'children': {'tag': source}
-                })
-        else:
-            spec['children'].append({'tag': 'config', 'children': config})
-        return self._request(spec)
+
+    def request(self, source):
+        """Validate the contents of the specified configuration.
+
+        :param source: name of the configuration datastore being validated or `config` element containing the configuration subtree to be validated
+        :type source: :ref:`srctarget_params`"""
+        node = new_ele("validate")
+        try:
+            src = validated_element(source, ("config", qualify("config")))
+        except Exception as e:
+            logger.debug(e)
+            src = util.datastore_or_url("source", source, self._assert)
+        (node if src.tag == "source" else sub_ele(node, "source")).append(src)
+        return self._request(node)
 
 
 class Commit(RPC):
-    
+    "`commit` RPC. Depends on the `:candidate` capability, and the `:confirmed-commit`."
+
     DEPENDS = [':candidate']
-    
-    SPEC = {'tag': 'commit', 'children': [] }
-    
-    def _parse_hook(self):
-        pass
-    
+
     def request(self, confirmed=False, timeout=None):
-        spec = SPEC.copy()
+        """Commit the candidate configuration as the device's new current configuration. Depends on the `:candidate` capability.
+
+        A confirmed commit (i.e. if *confirmed* is `True`) is reverted if there is no followup commit within the *timeout* interval. If no timeout is specified the confirm timeout defaults to 600 seconds (10 minutes). A confirming commit may have the *confirmed* parameter but this is not required. Depends on the `:confirmed-commit` capability.
+
+        :param confirmed: whether this is a confirmed commit
+        :type confirmed: bool
+
+        :param timeout: confirm timeout in seconds
+        :type timeout: int"""
+        node = new_ele("commit")
         if confirmed:
-            self._assert(':confirmed-commit')
-            children = spec['children']
-            children.append({'tag': 'confirmed'})
+            self._assert(":confirmed-commit")
+            sub_ele(node, "confirmed")
             if timeout is not None:
-                children.append({
-                    'tag': 'confirm-timeout',
-                    'text': timeout
-                })
-        return self._request(Commit.SPEC)
+                sub_ele(node, "confirm-timeout").text = timeout
+        return self._request(node)
 
 
 class DiscardChanges(RPC):
-    
-    DEPENDS = [':candidate']
-    
-    SPEC = {'tag': 'discard-changes'}
-    
-    request = lambda self: self._request(DiscardChanges.SPEC)
+    "`discard-changes` RPC. Depends on the `:candidate` capability."
+
+    DEPENDS = [":candidate"]
+
+    def request(self):
+        """Revert the candidate configuration to the currently running configuration. Any uncommitted changes are discarded."""
+        return self._request(new_ele("discard-changes"))
\ No newline at end of file