CouchPotatoServer/couchpotato/settings/loader.py

from blinker import signal
from couchpotato.core.logger import CPLog
import glob
import os

log = CPLog(__name__)

class SettingsLoader:

    def __init__(self, root = ''):

        self.settings_register = signal('settings.register')
        self.settings_save = signal('settings.save')

        self.paths = {
            'plugins' : ('couchpotato.core.plugins', os.path.join(root, 'couchpotato', 'core', 'plugins')),
            'providers' : ('couchpotato.core.providers', os.path.join(root, 'couchpotato', 'core', 'providers')),
        }

        for type, tuple in self.paths.iteritems():
            self.loadFromDir(tuple[0], tuple[1])

    def loadFromDir(self, module, dir):
        did_save = 0
        for file in glob.glob(os.path.join(dir, '*')):
            plugin_name = os.path.basename(file)
            plugin_dir = os.path.join(dir, plugin_name)
            if os.path.isdir(plugin_dir):
                did_save += self.loadConfig(module, plugin_name, save = False)

        if did_save:
            self.settings_save.send()

    def loadConfig(self, module, name, save = True):
        module_name = '%s.%s' % (module, name)
        print module_name
        try:
            m = getattr(self.loadModule(module_name), name)
            (section, options) = m.config
            self.settings_register.send(section, options = options, save = save)

            return True
        except Exception, e:
            log.error("Failed loading config for %s: %s" % (name, e))

    def loadModule(self, name):
        try:
            m = __import__(name)
            splitted = name.split('.')
            for sub in splitted[1:-1]:
                m = getattr(m, sub)
            return m
        except:
            raise