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.
27
from bzrlib.osutils import config_dir
28
DEFAULT_PLUGIN_PATH = os.path.join(config_dir(), 'plugins')
35
Find all python plugins and load them.
37
Loading a plugin means importing it into the python interpreter.
38
The plugin is expected to make calls to register commands when
39
it's loaded (or perhaps access other hooks in future.)
41
A list of plugs is stored in bzrlib.plugin.all_plugins for future
44
The environment variable BZR_PLUGIN_PATH is considered a delimited
45
set of paths to look through. Each entry is searched for *.py
46
files (and whatever other extensions are used in the platform,
54
from sets import Set as set # python2.3
56
from bzrlib.trace import log_error, mutter, log_exception
57
from bzrlib.errors import BzrError
59
bzrpath = os.environ.get('BZR_PLUGIN_PATH')
61
bzrpath = DEFAULT_PLUGIN_PATH
65
raise BzrError("plugins already initialized")
67
# The problem with imp.get_suffixes() is that it doesn't include
68
# .pyo which is technically valid
69
# It also means that "testmodule.so" will show up as both test and testmodule
70
# though it is only valid as 'test'
71
# but you should be careful, because "testmodule.py" loads as testmodule.
72
suffixes = imp.get_suffixes()
73
suffixes.append(('.pyo', 'rb', imp.PY_COMPILED))
74
package_entries = ['__init__.py', '__init__.pyc', '__init__.pyo']
75
for d in bzrpath.split(os.pathsep):
76
# going through them one by one allows different plugins with the same
77
# filename in different directories in the path
78
mutter('looking for plugins in %s' % d)
82
if not os.path.isdir(d):
84
for f in os.listdir(d):
85
path = os.path.join(d, f)
86
if os.path.isdir(path):
87
for entry in package_entries:
88
# This directory should be a package, and thus added to
90
if os.path.isfile(os.path.join(path, entry)):
92
else: # This directory is not a package
95
for suffix_info in suffixes:
96
if f.endswith(suffix_info[0]):
97
f = f[:-len(suffix_info[0])]
98
if suffix_info[2] == imp.C_EXTENSION and f.endswith('module'):
99
f = f[:-len('module')]
103
mutter('add plugin name %s' % f)
106
plugin_names = list(plugin_names)
108
for name in plugin_names:
110
plugin_info = imp.find_module(name, [d])
111
mutter('load plugin %r' % (plugin_info,))
113
plugin = imp.load_module('bzrlib.plugin.' + name,
115
all_plugins.append(plugin_info)
117
if plugin_info[0] is not None:
118
plugin_info[0].close()
120
log_error('Unable to load plugin %r from %r' % (name, d))
added
removed
bzrlib/plugin.py
