78
70
Future calls to load_plugins() will be ignored.
80
global _plugins_disabled
81
_plugins_disabled = True
85
def describe_plugins(show_paths=False):
86
"""Generate text description of plugins.
88
Includes both those that have loaded, and those that failed to
91
:param show_paths: If true,
92
:returns: Iterator of text lines (including newlines.)
94
from inspect import getdoc
95
loaded_plugins = plugins()
96
all_names = sorted(list(set(
97
loaded_plugins.keys() + plugin_warnings.keys())))
98
for name in all_names:
99
if name in loaded_plugins:
100
plugin = loaded_plugins[name]
101
version = plugin.__version__
102
if version == 'unknown':
104
yield '%s %s\n' % (name, version)
105
d = getdoc(plugin.module)
107
doc = d.split('\n')[0]
109
doc = '(no description)'
110
yield (" %s\n" % doc)
112
yield (" %s\n" % plugin.path())
115
yield "%s (failed to load)\n" % name
116
if name in plugin_warnings:
117
for line in plugin_warnings[name]:
118
yield " ** " + line + '\n'
72
# TODO: jam 20060131 This should probably also disable
122
78
def _strip_trailing_sep(path):
123
79
return path.rstrip("\\/")
126
def _get_specific_plugin_paths(paths):
127
"""Returns the plugin paths from a string describing the associations.
129
:param paths: A string describing the paths associated with the plugins.
131
:returns: A list of (plugin name, path) tuples.
133
For example, if paths is my_plugin@/test/my-test:her_plugin@/production/her,
134
[('my_plugin', '/test/my-test'), ('her_plugin', '/production/her')]
137
Note that ':' in the example above depends on the os.
142
for spec in paths.split(os.pathsep):
144
name, path = spec.split('@')
146
raise errors.BzrCommandError(
147
'"%s" is not a valid <plugin_name>@<plugin_path> description '
149
specs.append((name, path))
153
def set_plugins_path(path=None):
154
"""Set the path for plugins to be loaded from.
156
:param path: The list of paths to search for plugins. By default,
157
path will be determined using get_standard_plugins_path.
158
if path is [], no plugins can be loaded.
161
path = get_standard_plugins_path()
162
_mod_plugins.__path__ = path
163
PluginImporter.reset()
164
# Set up a blacklist for disabled plugins
165
disabled_plugins = os.environ.get('BZR_DISABLE_PLUGINS', None)
166
if disabled_plugins is not None:
167
for name in disabled_plugins.split(os.pathsep):
168
PluginImporter.blacklist.add('bzrlib.plugins.' + name)
169
# Set up a the specific paths for plugins
170
for plugin_name, plugin_path in _get_specific_plugin_paths(os.environ.get(
171
'BZR_PLUGINS_AT', None)):
172
PluginImporter.specific_paths[
173
'bzrlib.plugins.%s' % plugin_name] = plugin_path
177
def _append_new_path(paths, new_path):
178
"""Append a new path if it set and not already known."""
179
if new_path is not None and new_path not in paths:
180
paths.append(new_path)
184
def get_core_plugin_path():
82
def set_plugins_path():
83
"""Set the path for plugins to be loaded from."""
84
path = os.environ.get('BZR_PLUGIN_PATH',
85
get_default_plugin_path()).split(os.pathsep)
186
86
bzr_exe = bool(getattr(sys, 'frozen', None))
187
87
if bzr_exe: # expand path for bzr.exe
188
88
# We need to use relative path to system-wide plugin
195
95
# then plugins directory is
196
96
# C:\Program Files\Bazaar\plugins
197
97
# so relative path is ../../../plugins
198
core_path = osutils.abspath(osutils.pathjoin(
199
osutils.dirname(__file__), '../../../plugins'))
200
else: # don't look inside library.zip
201
# search the plugin path before the bzrlib installed dir
202
core_path = os.path.dirname(_mod_plugins.__file__)
206
def get_site_plugin_path():
207
"""Returns the path for the site installed plugins."""
208
if sys.platform == 'win32':
209
# We don't have (yet) a good answer for windows since that is certainly
210
# related to the way we build the installers. -- vila20090821
214
from distutils.sysconfig import get_python_lib
216
# If distutuils is not available, we just don't know where they are
219
site_path = osutils.pathjoin(get_python_lib(), 'bzrlib', 'plugins')
223
def get_user_plugin_path():
224
return osutils.pathjoin(config.config_dir(), 'plugins')
227
def get_standard_plugins_path():
228
"""Determine a plugin path suitable for general use."""
229
# Ad-Hoc default: core is not overriden by site but user can overrides both
230
# The rationale is that:
231
# - 'site' comes last, because these plugins should always be available and
232
# are supposed to be in sync with the bzr installed on site.
233
# - 'core' comes before 'site' so that running bzr from sources or a user
234
# installed version overrides the site version.
235
# - 'user' comes first, because... user is always right.
236
# - the above rules clearly defines which plugin version will be loaded if
237
# several exist. Yet, it is sometimes desirable to disable some directory
238
# so that a set of plugins is disabled as once. This can be done via
239
# -site, -core, -user.
241
env_paths = os.environ.get('BZR_PLUGIN_PATH', '+user').split(os.pathsep)
242
defaults = ['+core', '+site']
244
# The predefined references
245
refs = dict(core=get_core_plugin_path(),
246
site=get_site_plugin_path(),
247
user=get_user_plugin_path())
249
# Unset paths that should be removed
250
for k,v in refs.iteritems():
252
# defaults can never mention removing paths as that will make it
253
# impossible for the user to revoke these removals.
254
if removed in env_paths:
255
env_paths.remove(removed)
260
for p in env_paths + defaults:
261
if p.startswith('+'):
262
# Resolve references if they are known
266
# Leave them untouched so user can still use paths starting
269
_append_new_path(paths, p)
98
path.append(osutils.abspath(osutils.pathjoin(
99
osutils.dirname(__file__), '../../../plugins')))
271
100
# Get rid of trailing slashes, since Python can't handle them when
272
101
# it tries to import modules.
273
paths = map(_strip_trailing_sep, paths)
277
def load_plugins(path=None):
102
path = map(_strip_trailing_sep, path)
103
if not bzr_exe: # don't look inside library.zip
104
# search the plugin path before the bzrlib installed dir
105
path.append(os.path.dirname(_mod_plugins.__file__))
106
# search the arch independent path if we can determine that and
107
# the plugin is found nowhere else
108
if sys.platform != 'win32':
110
from distutils.sysconfig import get_python_lib
112
# If distutuils is not available, we just won't add that path
115
archless_path = osutils.pathjoin(get_python_lib(), 'bzrlib',
117
if archless_path not in path:
118
path.append(archless_path)
119
_mod_plugins.__path__ = path
278
124
"""Load bzrlib plugins.
280
126
The environment variable BZR_PLUGIN_PATH is considered a delimited
335
172
load_from_dirs = load_from_path
338
def _find_plugin_module(dir, name):
339
"""Check if there is a valid python module that can be loaded as a plugin.
341
:param dir: The directory where the search is performed.
342
:param path: An existing file path, either a python file or a package
345
:return: (name, path, description) name is the module name, path is the
346
file to load and description is the tuple returned by
349
path = osutils.pathjoin(dir, name)
350
if os.path.isdir(path):
351
# Check for a valid __init__.py file, valid suffixes depends on -O and
352
# can be .py, .pyc and .pyo
353
for suffix, mode, kind in imp.get_suffixes():
354
if kind not in (imp.PY_SOURCE, imp.PY_COMPILED):
355
# We don't recognize compiled modules (.so, .dll, etc)
357
init_path = osutils.pathjoin(path, '__init__' + suffix)
358
if os.path.isfile(init_path):
359
return name, init_path, (suffix, mode, kind)
361
for suffix, mode, kind in imp.get_suffixes():
362
if name.endswith(suffix):
363
# Clean up the module name
364
name = name[:-len(suffix)]
365
if kind == imp.C_EXTENSION and name.endswith('module'):
366
name = name[:-len('module')]
367
return name, path, (suffix, mode, kind)
368
# There is no python module here
369
return None, None, (None, None, None)
372
def record_plugin_warning(plugin_name, warning_message):
373
trace.mutter(warning_message)
374
plugin_warnings.setdefault(plugin_name, []).append(warning_message)
377
def _load_plugin_module(name, dir):
378
"""Load plugin name from dir.
380
:param name: The plugin name in the bzrlib.plugins namespace.
381
:param dir: The directory the plugin is loaded from for error messages.
383
if ('bzrlib.plugins.%s' % name) in PluginImporter.blacklist:
386
exec "import bzrlib.plugins.%s" % name in {}
387
except KeyboardInterrupt:
389
except errors.IncompatibleAPI, e:
391
"Unable to load plugin %r. It requested API version "
392
"%s of module %s but the minimum exported version is %s, and "
393
"the maximum is %s" %
394
(name, e.wanted, e.api, e.minimum, e.current))
395
record_plugin_warning(name, warning_message)
397
trace.warning("%s" % e)
398
if re.search('\.|-| ', name):
399
sanitised_name = re.sub('[-. ]', '_', name)
400
if sanitised_name.startswith('bzr_'):
401
sanitised_name = sanitised_name[len('bzr_'):]
402
trace.warning("Unable to load %r in %r as a plugin because the "
403
"file path isn't a valid module name; try renaming "
404
"it to %r." % (name, dir, sanitised_name))
406
record_plugin_warning(
408
'Unable to load plugin %r from %r' % (name, dir))
409
trace.log_exception_quietly()
410
if 'error' in debug.debug_flags:
411
trace.print_exception(sys.exc_info(), sys.stderr)
414
175
def load_from_dir(d):
415
"""Load the plugins in directory d.
417
d must be in the plugins module path already.
418
This function is called once for each directory in the module path.
176
"""Load the plugins in directory d."""
177
# Get the list of valid python suffixes for __init__.py?
178
# this includes .py, .pyc, and .pyo (depending on if we are running -O)
179
# but it doesn't include compiled modules (.so, .dll, etc)
180
valid_suffixes = [suffix for suffix, mod_type, flags in imp.get_suffixes()
181
if flags in (imp.PY_SOURCE, imp.PY_COMPILED)]
182
package_entries = ['__init__'+suffix for suffix in valid_suffixes]
420
183
plugin_names = set()
421
for p in os.listdir(d):
422
name, path, desc = _find_plugin_module(d, p)
424
if name == '__init__':
425
# We do nothing with the __init__.py file in directories from
426
# the bzrlib.plugins module path, we may want to, one day
428
continue # We don't load __init__.py in the plugins dirs
429
elif getattr(_mod_plugins, name, None) is not None:
430
# The module has already been loaded from another directory
431
# during a previous call.
432
# FIXME: There should be a better way to report masked plugins
434
trace.mutter('Plugin name %s already loaded', name)
184
for f in os.listdir(d):
185
path = osutils.pathjoin(d, f)
186
if os.path.isdir(path):
187
for entry in package_entries:
188
# This directory should be a package, and thus added to
190
if os.path.isfile(osutils.pathjoin(path, entry)):
192
else: # This directory is not a package
195
for suffix_info in imp.get_suffixes():
196
if f.endswith(suffix_info[0]):
197
f = f[:-len(suffix_info[0])]
198
if suffix_info[2] == imp.C_EXTENSION and f.endswith('module'):
199
f = f[:-len('module')]
436
plugin_names.add(name)
203
if getattr(_mod_plugins, f, None):
204
mutter('Plugin name %s already loaded', f)
206
# mutter('add plugin name %s', f)
438
209
for name in plugin_names:
439
_load_plugin_module(name, d)
211
exec "import bzrlib.plugins.%s" % name in {}
212
except KeyboardInterrupt:
215
## import pdb; pdb.set_trace()
216
if re.search('\.|-| ', name):
217
sanitised_name = re.sub('[-. ]', '_', name)
218
if sanitised_name.startswith('bzr_'):
219
sanitised_name = sanitised_name[len('bzr_'):]
220
warning("Unable to load %r in %r as a plugin because the "
221
"file path isn't a valid module name; try renaming "
222
"it to %r." % (name, d, sanitised_name))
224
warning('Unable to load plugin %r from %r' % (name, d))
225
log_exception_quietly()
226
if 'error' in debug.debug_flags:
227
trace.print_exception(sys.exc_info(), sys.stderr)
230
@deprecated_function(one_three)
231
def load_from_zip(zip_name):
232
"""Load all the plugins in a zip."""
233
valid_suffixes = ('.py', '.pyc', '.pyo') # only python modules/packages
236
index = zip_name.rindex('.zip')
239
archive = zip_name[:index+4]
240
prefix = zip_name[index+5:]
242
mutter('Looking for plugins in %r', zip_name)
244
# use zipfile to get list of files/dirs inside zip
246
z = zipfile.ZipFile(archive)
247
namelist = z.namelist()
249
except zipfile.error:
254
prefix = prefix.replace('\\','/')
255
if prefix[-1] != '/':
258
namelist = [name[ix:]
260
if name.startswith(prefix)]
262
mutter('Names in archive: %r', namelist)
264
for name in namelist:
265
if not name or name.endswith('/'):
268
# '/' is used to separate pathname components inside zip archives
271
head, tail = '', name
273
head, tail = name.rsplit('/',1)
275
# we don't need looking in subdirectories
278
base, suffix = osutils.splitext(tail)
279
if suffix not in valid_suffixes:
282
if base == '__init__':
293
if getattr(_mod_plugins, plugin_name, None):
294
mutter('Plugin name %s already loaded', plugin_name)
298
exec "import bzrlib.plugins.%s" % plugin_name in {}
299
mutter('Load plugin %s from zip %r', plugin_name, zip_name)
300
except KeyboardInterrupt:
303
## import pdb; pdb.set_trace()
304
warning('Unable to load plugin %r from %r'
306
log_exception_quietly()
307
if 'error' in debug.debug_flags:
308
trace.print_exception(sys.exc_info(), sys.stderr)
443
312
"""Return a dictionary of the plugins.
445
314
Each item in the dictionary is a PlugIn object.
573
442
def version_info(self):
574
443
"""Return the plugin's version_tuple or None if unknown."""
575
444
version_info = getattr(self.module, 'version_info', None)
576
if version_info is not None:
578
if isinstance(version_info, types.StringType):
579
version_info = version_info.split('.')
580
elif len(version_info) == 3:
581
version_info = tuple(version_info) + ('final', 0)
583
# The given version_info isn't even iteratible
584
trace.log_exception_quietly()
585
version_info = (version_info,)
445
if version_info is not None and len(version_info) == 3:
446
version_info = tuple(version_info) + ('final', 0)
586
447
return version_info
588
449
def _get__version__(self):
589
450
version_info = self.version_info()
590
if version_info is None or len(version_info) == 0:
451
if version_info is None:
593
version_string = _format_version_tuple(version_info)
594
except (ValueError, TypeError, IndexError), e:
595
trace.log_exception_quietly()
596
# try to return something usefull for bad plugins, in stead of
598
version_string = '.'.join(map(str, version_info))
453
if version_info[3] == 'final':
454
version_string = '%d.%d.%d' % version_info[:3]
456
version_string = '%d.%d.%d%s%d' % version_info
599
457
return version_string
601
459
__version__ = property(_get__version__)
604
class _PluginImporter(object):
605
"""An importer tailored to bzr specific needs.
607
This is a singleton that takes care of:
608
- disabled plugins specified in 'blacklist',
609
- plugins that needs to be loaded from specific directories.
616
self.blacklist = set()
617
self.specific_paths = {}
619
def find_module(self, fullname, parent_path=None):
620
"""Search a plugin module.
622
Disabled plugins raise an import error, plugins with specific paths
623
returns a specific loader.
625
:return: None if the plugin doesn't need special handling, self
628
if not fullname.startswith('bzrlib.plugins.'):
630
if fullname in self.blacklist:
631
raise ImportError('%s is disabled' % fullname)
632
if fullname in self.specific_paths:
636
def load_module(self, fullname):
637
"""Load a plugin from a specific directory."""
638
# We are called only for specific paths
639
plugin_path = self.specific_paths[fullname]
641
if os.path.isdir(plugin_path):
642
for suffix, mode, kind in imp.get_suffixes():
643
if kind not in (imp.PY_SOURCE, imp.PY_COMPILED):
644
# We don't recognize compiled modules (.so, .dll, etc)
646
init_path = osutils.pathjoin(plugin_path, '__init__' + suffix)
647
if os.path.isfile(init_path):
648
# We've got a module here and load_module needs specific
650
loading_path = plugin_path
653
kind = imp.PKG_DIRECTORY
656
for suffix, mode, kind in imp.get_suffixes():
657
if plugin_path.endswith(suffix):
658
loading_path = plugin_path
660
if loading_path is None:
661
raise ImportError('%s cannot be loaded from %s'
662
% (fullname, plugin_path))
663
if kind is imp.PKG_DIRECTORY:
666
f = open(loading_path, mode)
668
mod = imp.load_module(fullname, f, loading_path,
669
(suffix, mode, kind))
670
mod.__package__ = fullname
677
# Install a dedicated importer for plugins requiring special handling
678
PluginImporter = _PluginImporter()
679
sys.meta_path.append(PluginImporter)
added
removed
bzrlib/plugin.py
