From 7d4eae131f2f844f368aa5ff79c68191756775b6 Mon Sep 17 00:00:00 2001
From: Dan Prince <dan.prince@rackspace.com>
Date: Fri, 11 Mar 2011 14:33:12 -0500
Subject: Add config for osapi_extensions_path. Update the ExtensionManager so
 that it loads extensions in the osapi_extensions_path.

---
 nova/api/openstack/__init__.py   |  9 +++++----
 nova/api/openstack/extensions.py | 32 +++++++++++++++++++++++++++++++-
 2 files changed, 36 insertions(+), 5 deletions(-)

(limited to 'nova/api')

diff --git a/nova/api/openstack/__init__.py b/nova/api/openstack/__init__.py
index 28e2a1691..8f6076511 100644
--- a/nova/api/openstack/__init__.py
+++ b/nova/api/openstack/__init__.py
@@ -69,7 +69,7 @@ class APIRouter(wsgi.Router):
         """Simple paste factory, :class:`nova.wsgi.Router` doesn't have one"""
         return cls()
 
-    def __init__(self, ext_manager=None):
+    def __init__(self, ext_mgr=None):
         mapper = routes.Mapper()
 
         server_members = {'action': 'POST'}
@@ -112,9 +112,10 @@ class APIRouter(wsgi.Router):
                         collection={'detail': 'GET'},
                         controller=shared_ip_groups.Controller())
 
-        if ext_manager is None:
-            ext_manager = extensions.ExtensionManager()
-        for resource in ext_manager.get_resources():
+        if ext_mgr is None:
+            ext_mgr = extensions.ExtensionManager(FLAGS.osapi_extensions_path)
+        for resource in ext_mgr.get_resources():
+            print resource
             resource.add_routes(mapper)
 
         super(APIRouter, self).__init__(mapper)
diff --git a/nova/api/openstack/extensions.py b/nova/api/openstack/extensions.py
index 1c539c500..24846d9cd 100644
--- a/nova/api/openstack/extensions.py
+++ b/nova/api/openstack/extensions.py
@@ -15,15 +15,45 @@
 #    License for the specific language governing permissions and limitations
 #    under the License.
 
+import imp
+import os
+import sys
+
+
 class ExtensionManager(object):
 
+    def __init__(self, path):
+
+        self.path = path
+        self.extensions = []
+        self._load_extensions()
+
     def get_resources(self):
         """
         returns a list of ExtensionResource objects
         """
-        return []
+        resources = []
+        for ext in self.extensions:
+            resources.append(ext.get_resources())
+        return resources
+
+    def _load_extensions(self):
+        if not os.path.exists(self.path):
+            return
+
+        for f in os.listdir(self.path):
+            mod_name, file_ext = os.path.splitext(os.path.split(f)[-1])
+            ext_path = os.path.join(self.path, f)
+            if file_ext.lower() == '.py':
+                mod = imp.load_source(mod_name, ext_path)
+                self.extensions.append(getattr(mod, 'get_extension')())
+
 
 class ExtensionResource(object):
+    """
+    Example ExtensionResource object. All ExtensionResource objects should
+    adhere to this interface.
+    """
 
     def add_routes(self, mapper):
         pass
-- 
cgit