Module Manager: Support augmenting models

This is very early code, but seems to work.

1 files changed, 28 insertions, 0 deletions

diff --git a/ncserver/base/modman.py b/ncserver/base/modman.py
index d8659a2..5033afa 100644
--- a/ncserver/base/modman.py
+++ b/ncserver/base/modman.py
@@ -146,6 +146,25 @@ class ModuleManager:
 
         return None
 
+    def _augments_for_ns(self, namespace):
+        """Find modules that augment the specified namespace.
+
+        :param str namespace:
+            The namespace of the model whose augments are desired.
+
+        :returns:
+            An Iterable of modules that augment the namespace.
+            The Iterable may be empty.
+        """
+        augments = list()
+
+        for mod in self.modules.values():
+            if getattr(mod, 'M_AUGMENTS', None) is not None and \
+               namespace in mod.M_AUGMENTS:
+                augments.append(mod)
+
+        return augments
+
     def load_module(self, name):
         """Load a module.
 
@@ -325,6 +344,15 @@ class ModuleManager:
                 raise error.OperationNotSupportedAppError(rpc)
             self.logger.debug('Dispatching edit-config to %s', module.M_NAME)
             module.edit(session, rpc, child, def_op)
+            for augment in self._augments_for_ns(namespace):
+                try:
+                    self.logger.debug(
+                        'Augment: Dispatching edit-config for %s to %s',
+                        namespace, augment.M_NAME
+                    )
+                    augment.edit(session, rpc, child, def_op)
+                except error.OperationNotSupportedAppError:
+                    continue
 
 
 MODMAN = ModuleManager()