Browse Source

Refactor plugin loading

pull/227/head
Oliverpool 9 years ago
parent
commit
1ed28ef2b4
  1. 8
      patacrep/build.py
  2. 17
      patacrep/files.py
  3. 7
      patacrep/tools/convert/__main__.py
  4. 8
      test/test_content/test_content.py
  5. 4
      test/test_song/test_parser.py

8
patacrep/build.py

@ -80,15 +80,11 @@ class Songbook:
) )
# Loading custom plugins # Loading custom plugins
self._config['_content_plugins'] = files.load_plugins( self._config['_content_plugins'] = files.load_plugins_content(
datadirs=self._config['_datadir'], datadirs=self._config['_datadir'],
root_modules=['content'],
keyword='CONTENT_PLUGINS',
) )
self._config['_song_plugins'] = files.load_plugins( self._config['_song_plugins'] = files.load_plugins_songs(
datadirs=self._config['_datadir'], datadirs=self._config['_datadir'],
root_modules=['songs'],
keyword='SONG_RENDERERS',
)['tsg'] )['tsg']
# Configuration set # Configuration set

17
patacrep/files.py

@ -93,6 +93,22 @@ def iter_modules(path, prefix):
LOGGER.debug("[plugins] Could not load module {}: {}".format(name, str(error))) LOGGER.debug("[plugins] Could not load module {}: {}".format(name, str(error)))
continue continue
def load_plugins_content(datadirs=()):
"""Load the content plugins, and return a dictionary of those plugins."""
return load_plugins(
datadirs=datadirs,
root_modules=('content',),
keyword='CONTENT_PLUGINS',
)
def load_plugins_songs(datadirs=()):
"""Load the song renderer plugins, and return a dictionary of those plugins."""
return load_plugins(
datadirs=datadirs,
root_modules=('songs',),
keyword='SONG_RENDERERS',
)
def load_plugins(datadirs, root_modules, keyword): def load_plugins(datadirs, root_modules, keyword):
"""Load all plugins, and return a dictionary of those plugins. """Load all plugins, and return a dictionary of those plugins.
@ -102,6 +118,7 @@ def load_plugins(datadirs, root_modules, keyword):
Arguments: Arguments:
- datadirs: List of directories in which plugins are to be searched. - datadirs: List of directories in which plugins are to be searched.
The plugins will also be searched in the patacrep modules.
- root_modules: the submodule in which plugins are to be searched, as a - root_modules: the submodule in which plugins are to be searched, as a
list of modules (e.g. ["some", "deep", "module"] for list of modules (e.g. ["some", "deep", "module"] for
"some.deep.module"). "some.deep.module").

7
patacrep/tools/convert/__main__.py

@ -35,12 +35,7 @@ def main(args=None):
dest = args[2] dest = args[2]
song_files = args[3:] song_files = args[3:]
# todo : what is the datadir argument used for? renderers = files.load_plugins_songs()
renderers = files.load_plugins(
datadirs=[],
root_modules=['songs'],
keyword='SONG_RENDERERS',
)
if dest not in renderers: if dest not in renderers:
LOGGER.error( LOGGER.error(

8
test/test_content/test_content.py

@ -108,15 +108,11 @@ class FileTest(unittest.TestCase, metaclass=dynamic.DynamicTest):
) )
# Load the plugins # Load the plugins
config['_content_plugins'] = files.load_plugins( config['_content_plugins'] = files.load_plugins_content(
datadirs=config['_datadir'], datadirs=config['_datadir'],
root_modules=['content'],
keyword='CONTENT_PLUGINS',
) )
config['_song_plugins'] = files.load_plugins( config['_song_plugins'] = files.load_plugins_songs(
datadirs=config['_datadir'], datadirs=config['_datadir'],
root_modules=['songs'],
keyword='SONG_RENDERERS',
)['tsg'] )['tsg']
return config return config

4
test/test_song/test_parser.py

@ -81,10 +81,8 @@ class FileTest(unittest.TestCase, metaclass=dynamic.DynamicTest):
cls.config['_datadir'] = [] cls.config['_datadir'] = []
cls.config['_datadir'].append('datadir') cls.config['_datadir'].append('datadir')
cls.song_plugins = files.load_plugins( cls.song_plugins = files.load_plugins_songs(
datadirs=cls.config['_datadir'], datadirs=cls.config['_datadir'],
root_modules=['songs'],
keyword='SONG_RENDERERS',
) )
with cls.chdir(): with cls.chdir():
for source in sorted(glob.glob('*.*.source')): for source in sorted(glob.glob('*.*.source')):

Loading…
Cancel
Save