You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
114 lines
4.3 KiB
114 lines
4.3 KiB
from couchpotato.core.event import fireEvent
|
|
from couchpotato.core.logger import CPLog
|
|
import glob
|
|
import os
|
|
import traceback
|
|
|
|
log = CPLog(__name__)
|
|
|
|
class Loader:
|
|
|
|
plugins = {}
|
|
providers = {}
|
|
|
|
modules = {}
|
|
|
|
def preload(self, root = ''):
|
|
|
|
core = os.path.join(root, 'couchpotato', 'core')
|
|
providers = os.path.join(root, 'couchpotato', 'core', 'providers')
|
|
|
|
self.paths = {
|
|
'core' : (0, 'couchpotato.core._base', os.path.join(core, '_base')),
|
|
'plugin' : (1, 'couchpotato.core.plugins', os.path.join(core, 'plugins')),
|
|
'notifications' : (20, 'couchpotato.core.notifications', os.path.join(core, 'notifications')),
|
|
'downloaders' : (20, 'couchpotato.core.downloaders', os.path.join(core, 'downloaders')),
|
|
'movie_provider' : (20, 'couchpotato.core.providers.movie', os.path.join(providers, 'movie')),
|
|
'nzb_provider' : (20, 'couchpotato.core.providers.nzb', os.path.join(providers, 'nzb')),
|
|
'torrent_provider' : (20, 'couchpotato.core.providers.torrent', os.path.join(providers, 'torrent')),
|
|
'trailer_provider' : (20, 'couchpotato.core.providers.trailer', os.path.join(providers, 'trailer')),
|
|
'subtitle_provider' : (20, 'couchpotato.core.providers.subtitle', os.path.join(providers, 'subtitle')),
|
|
'metadata_provider' : (25, 'couchpotato.core.providers.metadata', os.path.join(providers, 'metadata')),
|
|
}
|
|
|
|
for type, tuple in self.paths.iteritems():
|
|
priority, module, dir = tuple
|
|
self.addFromDir(type, priority, module, dir)
|
|
|
|
def run(self):
|
|
did_save = 0
|
|
|
|
for priority in self.modules:
|
|
for module_name, plugin in sorted(self.modules[priority].iteritems()):
|
|
# Load module
|
|
try:
|
|
m = getattr(self.loadModule(module_name), plugin.get('name'))
|
|
|
|
log.info("Loading %s: %s" % (plugin['type'], plugin['name']))
|
|
|
|
# Save default settings for plugin/provider
|
|
did_save += self.loadSettings(m, module_name, save = False)
|
|
|
|
self.loadPlugins(m, plugin.get('name'))
|
|
except Exception, e:
|
|
log.error('Can\'t import %s: %s' % (module_name, traceback.format_exc()))
|
|
|
|
if did_save:
|
|
fireEvent('settings.save')
|
|
|
|
def addFromDir(self, type, priority, module, dir):
|
|
|
|
for file in glob.glob(os.path.join(dir, '*')):
|
|
name = os.path.basename(file)
|
|
if os.path.isdir(os.path.join(dir, name)):
|
|
module_name = '%s.%s' % (module, name)
|
|
self.addModule(priority, type, module_name, name)
|
|
|
|
def loadSettings(self, module, name, save = True):
|
|
try:
|
|
for section in module.config:
|
|
fireEvent('settings.options', section['name'], section)
|
|
options = {}
|
|
for group in section['groups']:
|
|
for option in group['options']:
|
|
options[option['name']] = option.get('default', '')
|
|
fireEvent('settings.register', section_name = section['name'], options = options, save = save)
|
|
return True
|
|
except Exception, e:
|
|
log.debug("Failed loading settings for '%s': %s" % (name, traceback.format_exc()))
|
|
return False
|
|
|
|
def loadPlugins(self, module, name):
|
|
try:
|
|
klass = module.start()
|
|
klass.registerPlugin()
|
|
|
|
if klass and getattr(klass, 'auto_register_static'):
|
|
klass.registerStatic(module.__file__)
|
|
|
|
return True
|
|
except Exception, e:
|
|
log.error("Failed loading plugin '%s': %s" % (module.__file__, traceback.format_exc()))
|
|
return False
|
|
|
|
def addModule(self, priority, type, module, name):
|
|
|
|
if not self.modules.get(priority):
|
|
self.modules[priority] = {}
|
|
|
|
self.modules[priority][module] = {
|
|
'priority': priority,
|
|
'module': module,
|
|
'type': type,
|
|
'name': name,
|
|
}
|
|
|
|
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
|
|
|