73
72
Future calls to load_plugins() will be ignored.
75
global _plugins_disabled
76
_plugins_disabled = True
74
# TODO: jam 20060131 This should probably also disable
80
80
def _strip_trailing_sep(path):
81
81
return path.rstrip("\\/")
84
def set_plugins_path(path=None):
85
"""Set the path for plugins to be loaded from.
87
:param path: The list of paths to search for plugins. By default,
88
path will be determined using get_standard_plugins_path.
89
if path is [], no plugins can be loaded.
92
path = get_standard_plugins_path()
93
_mod_plugins.__path__ = path
94
PluginImporter.reset()
95
# Set up a blacklist for disabled plugins
96
disabled_plugins = os.environ.get('BZR_DISABLE_PLUGINS', None)
97
if disabled_plugins is not None:
98
for name in disabled_plugins.split(os.pathsep):
99
PluginImporter.blacklist.add('bzrlib.plugins.' + name)
100
# Set up a the specific paths for plugins
101
specific_plugins = os.environ.get('BZR_PLUGINS_AT', None)
102
if specific_plugins is not None:
103
for spec in specific_plugins.split(os.pathsep):
104
plugin_name, plugin_path = spec.split('@')
105
PluginImporter.specific_paths[
106
'bzrlib.plugins.%s' % plugin_name] = plugin_path
110
def _append_new_path(paths, new_path):
111
"""Append a new path if it set and not already known."""
112
if new_path is not None and new_path not in paths:
113
paths.append(new_path)
117
def get_core_plugin_path():
84
def set_plugins_path():
85
"""Set the path for plugins to be loaded from."""
86
path = os.environ.get('BZR_PLUGIN_PATH',
87
get_default_plugin_path()).split(os.pathsep)
119
88
bzr_exe = bool(getattr(sys, 'frozen', None))
120
89
if bzr_exe: # expand path for bzr.exe
121
90
# We need to use relative path to system-wide plugin
128
97
# then plugins directory is
129
98
# C:\Program Files\Bazaar\plugins
130
99
# so relative path is ../../../plugins
131
core_path = osutils.abspath(osutils.pathjoin(
132
osutils.dirname(__file__), '../../../plugins'))
133
else: # don't look inside library.zip
134
# search the plugin path before the bzrlib installed dir
135
core_path = os.path.dirname(_mod_plugins.__file__)
139
def get_site_plugin_path():
140
"""Returns the path for the site installed plugins."""
141
if sys.platform == 'win32':
142
# We don't have (yet) a good answer for windows since that is certainly
143
# related to the way we build the installers. -- vila20090821
147
from distutils.sysconfig import get_python_lib
149
# If distutuils is not available, we just don't know where they are
152
site_path = osutils.pathjoin(get_python_lib(), 'bzrlib', 'plugins')
156
def get_user_plugin_path():
157
return osutils.pathjoin(config.config_dir(), 'plugins')
160
def get_standard_plugins_path():
161
"""Determine a plugin path suitable for general use."""
162
# Ad-Hoc default: core is not overriden by site but user can overrides both
163
# The rationale is that:
164
# - 'site' comes last, because these plugins should always be available and
165
# are supposed to be in sync with the bzr installed on site.
166
# - 'core' comes before 'site' so that running bzr from sources or a user
167
# installed version overrides the site version.
168
# - 'user' comes first, because... user is always right.
169
# - the above rules clearly defines which plugin version will be loaded if
170
# several exist. Yet, it is sometimes desirable to disable some directory
171
# so that a set of plugins is disabled as once. This can be done via
172
# -site, -core, -user.
174
env_paths = os.environ.get('BZR_PLUGIN_PATH', '+user').split(os.pathsep)
175
defaults = ['+core', '+site']
177
# The predefined references
178
refs = dict(core=get_core_plugin_path(),
179
site=get_site_plugin_path(),
180
user=get_user_plugin_path())
182
# Unset paths that should be removed
183
for k,v in refs.iteritems():
185
# defaults can never mention removing paths as that will make it
186
# impossible for the user to revoke these removals.
187
if removed in env_paths:
188
env_paths.remove(removed)
193
for p in env_paths + defaults:
194
if p.startswith('+'):
195
# Resolve references if they are known
199
# Leave them untouched so user can still use paths starting
202
_append_new_path(paths, p)
100
path.append(osutils.abspath(osutils.pathjoin(
101
osutils.dirname(__file__), '../../../plugins')))
204
102
# Get rid of trailing slashes, since Python can't handle them when
205
103
# it tries to import modules.
206
paths = map(_strip_trailing_sep, paths)
210
def load_plugins(path=None):
104
path = map(_strip_trailing_sep, path)
105
if not bzr_exe: # don't look inside library.zip
106
# search the plugin path before the bzrlib installed dir
107
path.append(os.path.dirname(_mod_plugins.__file__))
108
# search the arch independent path if we can determine that and
109
# the plugin is found nowhere else
110
if sys.platform != 'win32':
112
from distutils.sysconfig import get_python_lib
114
# If distutuils is not available, we just won't add that path
117
archless_path = osutils.pathjoin(get_python_lib(), 'bzrlib',
119
if archless_path not in path:
120
path.append(archless_path)
121
_mod_plugins.__path__ = path
211
126
"""Load bzrlib plugins.
213
128
The environment variable BZR_PLUGIN_PATH is considered a delimited
268
174
load_from_dirs = load_from_path
271
def _find_plugin_module(dir, name):
272
"""Check if there is a valid python module that can be loaded as a plugin.
274
:param dir: The directory where the search is performed.
275
:param path: An existing file path, either a python file or a package
278
:return: (name, path, description) name is the module name, path is the
279
file to load and description is the tuple returned by
282
path = osutils.pathjoin(dir, name)
283
if os.path.isdir(path):
284
# Check for a valid __init__.py file, valid suffixes depends on -O and
285
# can be .py, .pyc and .pyo
286
for suffix, mode, kind in imp.get_suffixes():
287
if kind not in (imp.PY_SOURCE, imp.PY_COMPILED):
288
# We don't recognize compiled modules (.so, .dll, etc)
290
init_path = osutils.pathjoin(path, '__init__' + suffix)
291
if os.path.isfile(init_path):
292
return name, init_path, (suffix, mode, kind)
294
for suffix, mode, kind in imp.get_suffixes():
295
if name.endswith(suffix):
296
# Clean up the module name
297
name = name[:-len(suffix)]
298
if kind == imp.C_EXTENSION and name.endswith('module'):
299
name = name[:-len('module')]
300
return name, path, (suffix, mode, kind)
301
# There is no python module here
302
return None, None, (None, None, None)
305
def _load_plugin_module(name, dir):
306
"""Load plugin name from dir.
308
:param name: The plugin name in the bzrlib.plugins namespace.
309
:param dir: The directory the plugin is loaded from for error messages.
311
if ('bzrlib.plugins.%s' % name) in PluginImporter.blacklist:
314
exec "import bzrlib.plugins.%s" % name in {}
315
except KeyboardInterrupt:
317
except errors.IncompatibleAPI, e:
318
trace.warning("Unable to load plugin %r. It requested API version "
319
"%s of module %s but the minimum exported version is %s, and "
320
"the maximum is %s" %
321
(name, e.wanted, e.api, e.minimum, e.current))
323
trace.warning("%s" % e)
324
if re.search('\.|-| ', name):
325
sanitised_name = re.sub('[-. ]', '_', name)
326
if sanitised_name.startswith('bzr_'):
327
sanitised_name = sanitised_name[len('bzr_'):]
328
trace.warning("Unable to load %r in %r as a plugin because the "
329
"file path isn't a valid module name; try renaming "
330
"it to %r." % (name, dir, sanitised_name))
332
trace.warning('Unable to load plugin %r from %r' % (name, dir))
333
trace.log_exception_quietly()
334
if 'error' in debug.debug_flags:
335
trace.print_exception(sys.exc_info(), sys.stderr)
338
177
def load_from_dir(d):
339
178
"""Load the plugins in directory d.
341
180
d must be in the plugins module path already.
342
This function is called once for each directory in the module path.
182
# Get the list of valid python suffixes for __init__.py?
183
# this includes .py, .pyc, and .pyo (depending on if we are running -O)
184
# but it doesn't include compiled modules (.so, .dll, etc)
185
valid_suffixes = [suffix for suffix, mod_type, flags in imp.get_suffixes()
186
if flags in (imp.PY_SOURCE, imp.PY_COMPILED)]
187
package_entries = ['__init__'+suffix for suffix in valid_suffixes]
344
188
plugin_names = set()
345
for p in os.listdir(d):
346
name, path, desc = _find_plugin_module(d, p)
348
if name == '__init__':
349
# We do nothing with the __init__.py file in directories from
350
# the bzrlib.plugins module path, we may want to, one day
352
continue # We don't load __init__.py in the plugins dirs
353
elif getattr(_mod_plugins, name, None) is not None:
354
# The module has already been loaded from another directory
355
# during a previous call.
356
# FIXME: There should be a better way to report masked plugins
358
trace.mutter('Plugin name %s already loaded', name)
189
for f in os.listdir(d):
190
path = osutils.pathjoin(d, f)
191
if os.path.isdir(path):
192
for entry in package_entries:
193
# This directory should be a package, and thus added to
195
if os.path.isfile(osutils.pathjoin(path, entry)):
197
else: # This directory is not a package
200
for suffix_info in imp.get_suffixes():
201
if f.endswith(suffix_info[0]):
202
f = f[:-len(suffix_info[0])]
203
if suffix_info[2] == imp.C_EXTENSION and f.endswith('module'):
204
f = f[:-len('module')]
360
plugin_names.add(name)
208
if getattr(_mod_plugins, f, None):
209
trace.mutter('Plugin name %s already loaded', f)
211
# trace.mutter('add plugin name %s', f)
362
214
for name in plugin_names:
363
_load_plugin_module(name, d)
216
exec "import bzrlib.plugins.%s" % name in {}
217
except KeyboardInterrupt:
219
except errors.IncompatibleAPI, e:
220
trace.warning("Unable to load plugin %r. It requested API version "
221
"%s of module %s but the minimum exported version is %s, and "
222
"the maximum is %s" %
223
(name, e.wanted, e.api, e.minimum, e.current))
225
trace.warning("%s" % e)
226
## import pdb; pdb.set_trace()
227
if re.search('\.|-| ', name):
228
sanitised_name = re.sub('[-. ]', '_', name)
229
if sanitised_name.startswith('bzr_'):
230
sanitised_name = sanitised_name[len('bzr_'):]
231
trace.warning("Unable to load %r in %r as a plugin because the "
232
"file path isn't a valid module name; try renaming "
233
"it to %r." % (name, d, sanitised_name))
235
trace.warning('Unable to load plugin %r from %r' % (name, d))
236
trace.log_exception_quietly()
237
if 'error' in debug.debug_flags:
238
trace.print_exception(sys.exc_info(), sys.stderr)
241
@deprecated_function(one_three)
242
def load_from_zip(zip_name):
243
"""Load all the plugins in a zip."""
244
valid_suffixes = ('.py', '.pyc', '.pyo') # only python modules/packages
247
index = zip_name.rindex('.zip')
250
archive = zip_name[:index+4]
251
prefix = zip_name[index+5:]
253
trace.mutter('Looking for plugins in %r', zip_name)
255
# use zipfile to get list of files/dirs inside zip
257
z = zipfile.ZipFile(archive)
258
namelist = z.namelist()
260
except zipfile.error:
265
prefix = prefix.replace('\\','/')
266
if prefix[-1] != '/':
269
namelist = [name[ix:]
271
if name.startswith(prefix)]
273
trace.mutter('Names in archive: %r', namelist)
275
for name in namelist:
276
if not name or name.endswith('/'):
279
# '/' is used to separate pathname components inside zip archives
282
head, tail = '', name
284
head, tail = name.rsplit('/',1)
286
# we don't need looking in subdirectories
289
base, suffix = osutils.splitext(tail)
290
if suffix not in valid_suffixes:
293
if base == '__init__':
304
if getattr(_mod_plugins, plugin_name, None):
305
trace.mutter('Plugin name %s already loaded', plugin_name)
309
exec "import bzrlib.plugins.%s" % plugin_name in {}
310
trace.mutter('Load plugin %s from zip %r', plugin_name, zip_name)
311
except KeyboardInterrupt:
314
## import pdb; pdb.set_trace()
315
trace.warning('Unable to load plugin %r from %r'
317
trace.log_exception_quietly()
318
if 'error' in debug.debug_flags:
319
trace.print_exception(sys.exc_info(), sys.stderr)
367
323
"""Return a dictionary of the plugins.
369
325
Each item in the dictionary is a PlugIn object.
497
453
def version_info(self):
498
454
"""Return the plugin's version_tuple or None if unknown."""
499
455
version_info = getattr(self.module, 'version_info', None)
500
if version_info is not None:
502
if isinstance(version_info, types.StringType):
503
version_info = version_info.split('.')
504
elif len(version_info) == 3:
505
version_info = tuple(version_info) + ('final', 0)
507
# The given version_info isn't even iteratible
508
trace.log_exception_quietly()
509
version_info = (version_info,)
456
if version_info is not None and len(version_info) == 3:
457
version_info = tuple(version_info) + ('final', 0)
510
458
return version_info
512
460
def _get__version__(self):
513
461
version_info = self.version_info()
514
if version_info is None or len(version_info) == 0:
462
if version_info is None:
517
version_string = _format_version_tuple(version_info)
518
except (ValueError, TypeError, IndexError), e:
519
trace.log_exception_quietly()
520
# try to return something usefull for bad plugins, in stead of
522
version_string = '.'.join(map(str, version_info))
464
if version_info[3] == 'final':
465
version_string = '%d.%d.%d' % version_info[:3]
467
version_string = '%d.%d.%d%s%d' % version_info
523
468
return version_string
525
470
__version__ = property(_get__version__)
528
class _PluginImporter(object):
529
"""An importer tailored to bzr specific needs.
531
This is a singleton that takes care of:
532
- disabled plugins specified in 'blacklist',
533
- plugins that needs to be loaded from specific directories.
540
self.blacklist = set()
541
self.specific_paths = {}
543
def find_module(self, fullname, parent_path=None):
544
"""Search a plugin module.
546
Disabled plugins raise an import error, plugins with specific paths
547
returns a specific loader.
549
:return: None if the plugin doesn't need special handling, self
552
if not fullname.startswith('bzrlib.plugins.'):
554
if fullname in self.blacklist:
555
raise ImportError('%s is disabled' % fullname)
556
if fullname in self.specific_paths:
560
def load_module(self, fullname):
561
"""Load a plugin from a specific directory."""
562
# We are called only for specific paths
563
plugin_path = self.specific_paths[fullname]
566
if os.path.isdir(plugin_path):
567
for suffix, mode, kind in imp.get_suffixes():
568
if kind not in (imp.PY_SOURCE, imp.PY_COMPILED):
569
# We don't recognize compiled modules (.so, .dll, etc)
571
init_path = osutils.pathjoin(plugin_path, '__init__' + suffix)
572
if os.path.isfile(init_path):
573
loading_path = init_path
577
for suffix, mode, kind in imp.get_suffixes():
578
if plugin_path.endswith(suffix):
579
loading_path = plugin_path
581
if loading_path is None:
582
raise ImportError('%s cannot be loaded from %s'
583
% (fullname, plugin_path))
584
f = open(loading_path, mode)
586
mod = imp.load_module(fullname, f, loading_path,
587
(suffix, mode, kind))
589
# The plugin can contain modules, so be ready
590
mod.__path__ = [plugin_path]
591
mod.__package__ = fullname
597
# Install a dedicated importer for plugins requiring special handling
598
PluginImporter = _PluginImporter()
599
sys.meta_path.append(PluginImporter)
added
removed
bzrlib/plugin.py
