1 files changed, 68 insertions, 0 deletions
diff --git a/cobbler/module_loader.py b/cobbler/module_loader.py
new file mode 100644
index 0000000..98ca487
--- /dev/null
+++ b/cobbler/module_loader.py
@@ -0,0 +1,68 @@
+#!/usr/bin/python
+
+"""
+Module loader, adapted for cobbler usage
+
+Copyright 2006-2007, Red Hat, Inc
+Adrian Likins <alikins@redhat.com>
+Michael DeHaan <mdehaan@redhat.com>
+
+This software may be freely redistributed under the terms of the GNU
+general public license.
+
+You should have received a copy of the GNU General Public License
+along with this program; if not, write to the Free Software
+Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+"""
+
+
+import distutils.sysconfig
+import os
+import sys
+import glob
+from rhpl.translate import _, N_, textdomain, utf8
+
+plib = distutils.sysconfig.get_python_lib()
+mod_path="%s/cobbler/modules" % plib
+print "DEBUG: adding to python path: %s" % mod_path
+sys.path.insert(0, mod_path)
+sys.path.insert(1, "%s/cobbler" % plib)
+
+def load_modules(module_path=mod_path, blacklist=None):
+    filenames = glob.glob("%s/*.py" % module_path)
+    filenames = filenames + glob.glob("%s/*.pyc" % module_path)
+    filesnames = filenames + glob.glob("%s/*.pyo" % module_path)
+
+    mods = {}
+
+    print sys.path
+
+    for fn in filenames:
+        basename = os.path.basename(fn)
+        if basename == "__init__.py":
+            continue
+        if basename[-3:] == ".py":
+            modname = basename[:-3]
+        elif basename[-4:] in [".pyc", ".pyo"]:
+            modname = basename[:-4]
+
+
+        try:
+            blip =  __import__("modules.%s" % ( modname), globals(), locals(), [modname])
+            if not hasattr(blip, "register"):
+                errmsg = _("%(module_path)s/%(modname)s is not a proper module")
+                print errmsg % {'module_path': module_path, 'modname':modname}
+                continue
+            mods[modname] = blip
+        except ImportError, e:
+            print e
+            raise
+
+    return mods
+
+
+
+
+if __name__ == "__main__":
+    print load_modules(module_path)
+