1
# Copyright (C) 2004, 2005 by Canonical Ltd
3
# This program is free software; you can redistribute it and/or modify
4
# it under the terms of the GNU General Public License as published by
5
# the Free Software Foundation; either version 2 of the License, or
6
# (at your option) any later version.
8
# This program is distributed in the hope that it will be useful,
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11
# GNU General Public License for more details.
13
# You should have received a copy of the GNU General Public License
14
# along with this program; if not, write to the Free Software
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
18
# This module implements plug-in support.
19
# Any python module in $BZR_PLUGIN_PATH will be imported upon initialization
20
# of bzrlib (and then forgotten about). In the plugin's main body, it should
21
# update any bzrlib registries it wants to extend; for example, to add new
22
# commands, import bzrlib.commands and add your new command to the
23
# plugin_cmds variable.
29
from sets import Set as set
30
from bzrlib.trace import log_error
34
"""Find all python files which are plugins, and load them
36
The environment variable BZR_PLUGIN_PATH is considered a delimited set of
37
paths to look through. Each entry is searched for *.py files (and whatever
38
other extensions are used in the platform, such as *.pyd).
40
bzrpath = os.environ.get('BZR_PLUGIN_PATH', os.path.expanduser('~/.bzr/plugins'))
42
# The problem with imp.get_suffixes() is that it doesn't include
43
# .pyo which is technically valid
44
# It also means that "testmodule.so" will show up as both test and testmodule
45
# though it is only valid as 'test'
46
# but you should be careful, because "testmodule.py" loads as testmodule.
47
suffixes = imp.get_suffixes()
48
suffixes.append(('.pyo', 'rb', imp.PY_COMPILED))
49
package_entries = ['__init__.py', '__init__.pyc', '__init__.pyo']
50
for d in bzrpath.split(os.pathsep):
51
# going trough them one by one allows different plugins with the same
52
# filename in different directories in the path
56
if not os.path.isdir(d):
58
for f in os.listdir(d):
59
path = os.path.join(d, f)
60
if os.path.isdir(path):
61
for entry in package_entries:
62
# This directory should be a package, and thus added to
64
if os.path.isfile(os.path.join(path, entry)):
66
else: # This directory is not a package
69
for suffix_info in suffixes:
70
if f.endswith(suffix_info[0]):
71
f = f[:-len(suffix_info[0])]
72
if suffix_info[2] == imp.C_EXTENSION and f.endswith('module'):
73
f = f[:-len('module')]
79
plugin_names = list(plugin_names)
81
for name in plugin_names:
83
plugin_info = imp.find_module(name, [d])
85
plugin = imp.load_module('bzrlib.plugin.' + name,
88
if plugin_info[0] is not None:
89
plugin_info[0].close()
91
log_error('Unable to load plugin: %r from %r\n%s' % (name, d, e))
added
removed
bzrlib/plugin.py
