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.
28
from bzrlib.config import config_dir
29
DEFAULT_PLUGIN_PATH = os.path.join(config_dir(), 'plugins')
37
Find all python plugins and load them.
39
Loading a plugin means importing it into the python interpreter.
40
The plugin is expected to make calls to register commands when
41
it's loaded (or perhaps access other hooks in future.)
43
A list of plugs is stored in bzrlib.plugin.all_plugins for future
46
The environment variable BZR_PLUGIN_PATH is considered a delimited
47
set of paths to look through. Each entry is searched for *.py
48
files (and whatever other extensions are used in the platform,
52
global all_plugins, _loaded
54
# People can make sure plugins are loaded, they just won't be twice
56
#raise BzrError("plugins already initialized")
61
from bzrlib.trace import log_error, mutter, log_exception
62
from bzrlib.errors import BzrError
63
from bzrlib import plugins
65
dirs = os.environ.get('BZR_PLUGIN_PATH', DEFAULT_PLUGIN_PATH).split(":")
66
dirs.insert(0, os.path.dirname(plugins.__file__))
68
# The problem with imp.get_suffixes() is that it doesn't include
69
# .pyo which is technically valid
70
# It also means that "testmodule.so" will show up as both test and testmodule
71
# though it is only valid as 'test'
72
# but you should be careful, because "testmodule.py" loads as testmodule.
73
suffixes = imp.get_suffixes()
74
suffixes.append(('.pyo', 'rb', imp.PY_COMPILED))
75
package_entries = ['__init__.py', '__init__.pyc', '__init__.pyo']
77
# going through them one by one allows different plugins with the same
78
# filename in different directories in the path
79
mutter('looking for plugins in %s' % d)
83
if not os.path.isdir(d):
85
for f in os.listdir(d):
86
path = os.path.join(d, f)
87
if os.path.isdir(path):
88
for entry in package_entries:
89
# This directory should be a package, and thus added to
91
if os.path.isfile(os.path.join(path, entry)):
93
else: # This directory is not a package
96
for suffix_info in suffixes:
97
if f.endswith(suffix_info[0]):
98
f = f[:-len(suffix_info[0])]
99
if suffix_info[2] == imp.C_EXTENSION and f.endswith('module'):
100
f = f[:-len('module')]
104
mutter('add plugin name %s' % f)
107
plugin_names = list(plugin_names)
109
for name in plugin_names:
111
plugin_info = imp.find_module(name, [d])
112
mutter('load plugin %r' % (plugin_info,))
114
plugin = imp.load_module('bzrlib.plugins.' + name,
116
all_plugins.append(plugin)
117
setattr(bzrlib.plugins, name, plugin)
119
if plugin_info[0] is not None:
120
plugin_info[0].close()
122
mutter('loaded succesfully')
124
log_error('Unable to load plugin %r from %r' % (name, d))
added
removed
bzrlib/plugin.py
