← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/plugin.py

[merge] fix \t in commit messages

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/plugin.py
15
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
16
 
17
17
 
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.
24
 
 
25
 
 
 
18
"""bzr python plugin support
 
19
 
 
20
Any python module in $BZR_PLUGIN_PATH will be imported upon initialization of
 
21
bzrlib. The module will be imported as 'bzrlib.plugins.$BASENAME(PLUGIN)'.
 
22
In the plugin's main body, it should update any bzrlib registries it wants to
 
23
extend; for example, to add new commands, import bzrlib.commands and add your
 
24
new command to the plugin_cmds variable.
 
25
"""
 
26
 
 
27
# TODO: Refactor this to make it more testable.  The main problem at the
 
28
# moment is that loading plugins affects the global process state -- for bzr
 
29
# in general use it's a reasonable assumption that all plugins are loaded at
 
30
# startup and then stay loaded, but this is less good for testing.
 
31
# 
 
32
# Several specific issues:
 
33
#  - plugins can't be unloaded and will continue to effect later tests
 
34
#  - load_plugins does nothing if called a second time
 
35
#  - plugin hooks can't be removed
 
36
#
 
37
# Our options are either to remove these restrictions, or work around them by
 
38
# loading the plugins into a different space than the one running the tests.
 
39
# That could be either a separate Python interpreter or perhaps a new
 
40
# namespace inside this interpreter.
 
41
 
 
42
import imp
26
43
import os
27
 
from bzrlib.osutils import config_dir
 
44
import sys
 
45
 
 
46
import bzrlib
 
47
from bzrlib.config import config_dir
 
48
from bzrlib.trace import log_error, mutter, log_exception, warning, \
 
49
        log_exception_quietly
 
50
from bzrlib.errors import BzrError
 
51
from bzrlib import plugins
 
52
 
28
53
DEFAULT_PLUGIN_PATH = os.path.join(config_dir(), 'plugins')
29
54
 
30
55
all_plugins = []
32
57
 
33
58
 
34
59
def load_plugins():
35
 
    """
36
 
    Find all python plugins and load them.
 
60
    """Find all python plugins and load them.
37
61
 
38
62
    Loading a plugin means importing it into the python interpreter.
39
63
    The plugin is expected to make calls to register commands when
55
79
        #raise BzrError("plugins already initialized")
56
80
    _loaded = True
57
81
 
58
 
    import sys, os, imp
59
 
    
60
 
    from bzrlib.trace import log_error, mutter, log_exception
61
 
    from bzrlib.errors import BzrError
62
 
    from bzrlib import plugins
63
 
 
64
82
    dirs = os.environ.get('BZR_PLUGIN_PATH', DEFAULT_PLUGIN_PATH).split(":")
65
83
    dirs.insert(0, os.path.dirname(plugins.__file__))
66
84
 
75
93
    for d in dirs:
76
94
        # going through them one by one allows different plugins with the same
77
95
        # filename in different directories in the path
78
 
        mutter('looking for plugins in %s' % d)
 
96
        mutter('looking for plugins in %s', d)
79
97
        if not d:
80
98
            continue
81
99
        plugin_names = set()
100
118
                        break
101
119
                else:
102
120
                    continue
103
 
            mutter('add plugin name %s' % f)
 
121
            mutter('add plugin name %s', f)
104
122
            plugin_names.add(f)
105
123
 
106
124
        plugin_names = list(plugin_names)
108
126
        for name in plugin_names:
109
127
            try:
110
128
                plugin_info = imp.find_module(name, [d])
111
 
                mutter('load plugin %r' % (plugin_info,))
 
129
                mutter('load plugin %r', plugin_info)
112
130
                try:
113
 
                    plugin = imp.load_module('bzrlib.plugin.' + name,
 
131
                    plugin = imp.load_module('bzrlib.plugins.' + name,
114
132
                                             *plugin_info)
115
133
                    all_plugins.append(plugin)
 
134
                    setattr(bzrlib.plugins, name, plugin)
116
135
                finally:
117
136
                    if plugin_info[0] is not None:
118
137
                        plugin_info[0].close()
119
138
 
120
139
                mutter('loaded succesfully')
121
 
            except:
122
 
                log_error('Unable to load plugin %r from %r' % (name, d))
123
 
                log_exception()
124
 
 
 
140
            except KeyboardInterrupt:
 
141
                raise
 
142
            except Exception, e:
 
143
                ## import pdb; pdb.set_trace()
 
144
                warning('Unable to load plugin %r from %r' % (name, d))
 
145
                log_exception_quietly()