~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/plugin.py

  • Committer: Ian Clatworthy
  • Date: 2009-09-09 11:43:10 UTC
  • mto: (4634.37.2 prepare-2.0)
  • mto: This revision was merged to the branch mainline in revision 4689.
  • Revision ID: ian.clatworthy@canonical.com-20090909114310-glw7tv76i5gnx9pt
put rules back in Makefile supporting plain-style docs

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2004, 2005, 2007 Canonical Ltd
 
1
# Copyright (C) 2004, 2005, 2007, 2008 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
12
12
#
13
13
# You should have received a copy of the GNU General Public License
14
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
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
16
 
17
17
 
18
18
"""bzr python plugin support.
20
20
When load_plugins() is invoked, any python module in any directory in
21
21
$BZR_PLUGIN_PATH will be imported.  The module will be imported as
22
22
'bzrlib.plugins.$BASENAME(PLUGIN)'.  In the plugin's main body, it should
23
 
update any bzrlib registries it wants to extend; for example, to add new
24
 
commands, import bzrlib.commands and add your new command to the plugin_cmds
25
 
variable.
 
23
update any bzrlib registries it wants to extend.
 
24
 
 
25
See the plugin-api developer documentation for information about writing
 
26
plugins.
26
27
 
27
28
BZR_PLUGIN_PATH is also honoured for any plugins imported via
28
 
'import bzrlib.plugins.PLUGINNAME', as long as set_plugins_path has been 
 
29
'import bzrlib.plugins.PLUGINNAME', as long as set_plugins_path has been
29
30
called.
30
31
"""
31
32
 
32
33
import os
33
34
import sys
34
35
 
 
36
from bzrlib import osutils
 
37
 
35
38
from bzrlib.lazy_import import lazy_import
 
39
 
36
40
lazy_import(globals(), """
37
41
import imp
38
42
import re
39
43
import types
40
 
import zipfile
41
44
 
42
45
from bzrlib import (
 
46
    _format_version_tuple,
43
47
    config,
44
 
    osutils,
 
48
    debug,
 
49
    errors,
 
50
    trace,
45
51
    )
46
52
from bzrlib import plugins as _mod_plugins
47
53
""")
48
54
 
49
 
from bzrlib.symbol_versioning import deprecated_function, zero_ninetyone
50
 
from bzrlib.trace import mutter, warning, log_exception_quietly
 
55
from bzrlib.symbol_versioning import deprecated_function
51
56
 
52
57
 
53
58
DEFAULT_PLUGIN_PATH = None
61
66
    return DEFAULT_PLUGIN_PATH
62
67
 
63
68
 
64
 
@deprecated_function(zero_ninetyone)
65
 
def all_plugins():
66
 
    """Return a dictionary of the plugins."""
67
 
    return dict((name, plugin.module) for name, plugin in plugins().items())
68
 
 
69
 
 
70
69
def disable_plugins():
71
70
    """Disable loading plugins.
72
71
 
73
72
    Future calls to load_plugins() will be ignored.
74
73
    """
75
 
    # TODO: jam 20060131 This should probably also disable
76
 
    #       load_from_dirs()
77
 
    global _loaded
78
 
    _loaded = True
 
74
    load_plugins([])
 
75
 
79
76
 
80
77
def _strip_trailing_sep(path):
81
78
    return path.rstrip("\\/")
82
79
 
83
 
def set_plugins_path():
84
 
    """Set the path for plugins to be loaded from."""
 
80
 
 
81
def set_plugins_path(path=None):
 
82
    """Set the path for plugins to be loaded from.
 
83
 
 
84
    :param path: The list of paths to search for plugins.  By default,
 
85
        path will be determined using get_standard_plugins_path.
 
86
        if path is [], no plugins can be loaded.
 
87
    """
 
88
    if path is None:
 
89
        path = get_standard_plugins_path()
 
90
    _mod_plugins.__path__ = path
 
91
    return path
 
92
 
 
93
 
 
94
def get_standard_plugins_path():
 
95
    """Determine a plugin path suitable for general use."""
85
96
    path = os.environ.get('BZR_PLUGIN_PATH',
86
97
                          get_default_plugin_path()).split(os.pathsep)
87
98
    # Get rid of trailing slashes, since Python can't handle them when
88
99
    # it tries to import modules.
89
100
    path = map(_strip_trailing_sep, path)
90
 
    # search the plugin path before the bzrlib installed dir
91
 
    path.append(os.path.dirname(_mod_plugins.__file__))
92
 
    _mod_plugins.__path__ = path
 
101
    bzr_exe = bool(getattr(sys, 'frozen', None))
 
102
    if bzr_exe:    # expand path for bzr.exe
 
103
        # We need to use relative path to system-wide plugin
 
104
        # directory because bzrlib from standalone bzr.exe
 
105
        # could be imported by another standalone program
 
106
        # (e.g. bzr-config; or TortoiseBzr/Olive if/when they
 
107
        # will become standalone exe). [bialix 20071123]
 
108
        # __file__ typically is
 
109
        # C:\Program Files\Bazaar\lib\library.zip\bzrlib\plugin.pyc
 
110
        # then plugins directory is
 
111
        # C:\Program Files\Bazaar\plugins
 
112
        # so relative path is ../../../plugins
 
113
        path.append(osutils.abspath(osutils.pathjoin(
 
114
            osutils.dirname(__file__), '../../../plugins')))
 
115
    if not bzr_exe:     # don't look inside library.zip
 
116
        # search the plugin path before the bzrlib installed dir
 
117
        path.append(os.path.dirname(_mod_plugins.__file__))
 
118
    # search the arch independent path if we can determine that and
 
119
    # the plugin is found nowhere else
 
120
    if sys.platform != 'win32':
 
121
        try:
 
122
            from distutils.sysconfig import get_python_lib
 
123
        except ImportError:
 
124
            # If distutuils is not available, we just won't add that path
 
125
            pass
 
126
        else:
 
127
            archless_path = osutils.pathjoin(get_python_lib(), 'bzrlib',
 
128
                    'plugins')
 
129
            if archless_path not in path:
 
130
                path.append(archless_path)
93
131
    return path
94
132
 
95
133
 
96
 
def load_plugins():
 
134
def load_plugins(path=None):
97
135
    """Load bzrlib plugins.
98
136
 
99
137
    The environment variable BZR_PLUGIN_PATH is considered a delimited
103
141
 
104
142
    load_from_dirs() provides the underlying mechanism and is called with
105
143
    the default directory list to provide the normal behaviour.
 
144
 
 
145
    :param path: The list of paths to search for plugins.  By default,
 
146
        path will be determined using get_standard_plugins_path.
 
147
        if path is [], no plugins can be loaded.
106
148
    """
107
149
    global _loaded
108
150
    if _loaded:
111
153
    _loaded = True
112
154
 
113
155
    # scan for all plugins in the path.
114
 
    load_from_path(set_plugins_path())
 
156
    load_from_path(set_plugins_path(path))
115
157
 
116
158
 
117
159
def load_from_path(dirs):
135
177
    for d in dirs:
136
178
        if not d:
137
179
            continue
138
 
        mutter('looking for plugins in %s', d)
 
180
        trace.mutter('looking for plugins in %s', d)
139
181
        if os.path.isdir(d):
140
182
            load_from_dir(d)
141
 
        else:
142
 
            # it might be a zip: try loading from the zip.
143
 
            load_from_zip(d)
144
 
            continue
145
183
 
146
184
 
147
185
# backwards compatability: load_from_dirs was the old name
150
188
 
151
189
 
152
190
def load_from_dir(d):
153
 
    """Load the plugins in directory d."""
 
191
    """Load the plugins in directory d.
 
192
 
 
193
    d must be in the plugins module path already.
 
194
    """
154
195
    # Get the list of valid python suffixes for __init__.py?
155
196
    # this includes .py, .pyc, and .pyo (depending on if we are running -O)
156
197
    # but it doesn't include compiled modules (.so, .dll, etc)
177
218
                    break
178
219
            else:
179
220
                continue
180
 
        if getattr(_mod_plugins, f, None):
181
 
            mutter('Plugin name %s already loaded', f)
 
221
        if f == '__init__':
 
222
            continue # We don't load __init__.py again in the plugin dir
 
223
        elif getattr(_mod_plugins, f, None):
 
224
            trace.mutter('Plugin name %s already loaded', f)
182
225
        else:
183
 
            # mutter('add plugin name %s', f)
 
226
            # trace.mutter('add plugin name %s', f)
184
227
            plugin_names.add(f)
185
 
    
 
228
 
186
229
    for name in plugin_names:
187
230
        try:
188
231
            exec "import bzrlib.plugins.%s" % name in {}
189
232
        except KeyboardInterrupt:
190
233
            raise
 
234
        except errors.IncompatibleAPI, e:
 
235
            trace.warning("Unable to load plugin %r. It requested API version "
 
236
                "%s of module %s but the minimum exported version is %s, and "
 
237
                "the maximum is %s" %
 
238
                (name, e.wanted, e.api, e.minimum, e.current))
191
239
        except Exception, e:
 
240
            trace.warning("%s" % e)
192
241
            ## import pdb; pdb.set_trace()
193
242
            if re.search('\.|-| ', name):
194
 
                warning('Unable to load plugin %r from %r: '
195
 
                    'It is not a valid python module name.' % (name, d))
 
243
                sanitised_name = re.sub('[-. ]', '_', name)
 
244
                if sanitised_name.startswith('bzr_'):
 
245
                    sanitised_name = sanitised_name[len('bzr_'):]
 
246
                trace.warning("Unable to load %r in %r as a plugin because the "
 
247
                        "file path isn't a valid module name; try renaming "
 
248
                        "it to %r." % (name, d, sanitised_name))
196
249
            else:
197
 
                warning('Unable to load plugin %r from %r' % (name, d))
198
 
            log_exception_quietly()
199
 
 
200
 
 
201
 
def load_from_zip(zip_name):
202
 
    """Load all the plugins in a zip."""
203
 
    valid_suffixes = ('.py', '.pyc', '.pyo')    # only python modules/packages
204
 
                                                # is allowed
205
 
 
206
 
    try:
207
 
        index = zip_name.rindex('.zip')
208
 
    except ValueError:
209
 
        return
210
 
    archive = zip_name[:index+4]
211
 
    prefix = zip_name[index+5:]
212
 
 
213
 
    mutter('Looking for plugins in %r', zip_name)
214
 
 
215
 
    # use zipfile to get list of files/dirs inside zip
216
 
    try:
217
 
        z = zipfile.ZipFile(archive)
218
 
        namelist = z.namelist()
219
 
        z.close()
220
 
    except zipfile.error:
221
 
        # not a valid zip
222
 
        return
223
 
 
224
 
    if prefix:
225
 
        prefix = prefix.replace('\\','/')
226
 
        if prefix[-1] != '/':
227
 
            prefix += '/'
228
 
        ix = len(prefix)
229
 
        namelist = [name[ix:]
230
 
                    for name in namelist
231
 
                    if name.startswith(prefix)]
232
 
 
233
 
    mutter('Names in archive: %r', namelist)
234
 
    
235
 
    for name in namelist:
236
 
        if not name or name.endswith('/'):
237
 
            continue
238
 
    
239
 
        # '/' is used to separate pathname components inside zip archives
240
 
        ix = name.rfind('/')
241
 
        if ix == -1:
242
 
            head, tail = '', name
243
 
        else:
244
 
            head, tail = name.rsplit('/',1)
245
 
        if '/' in head:
246
 
            # we don't need looking in subdirectories
247
 
            continue
248
 
    
249
 
        base, suffix = osutils.splitext(tail)
250
 
        if suffix not in valid_suffixes:
251
 
            continue
252
 
    
253
 
        if base == '__init__':
254
 
            # package
255
 
            plugin_name = head
256
 
        elif head == '':
257
 
            # module
258
 
            plugin_name = base
259
 
        else:
260
 
            continue
261
 
    
262
 
        if not plugin_name:
263
 
            continue
264
 
        if getattr(_mod_plugins, plugin_name, None):
265
 
            mutter('Plugin name %s already loaded', plugin_name)
266
 
            continue
267
 
    
268
 
        try:
269
 
            exec "import bzrlib.plugins.%s" % plugin_name in {}
270
 
            mutter('Load plugin %s from zip %r', plugin_name, zip_name)
271
 
        except KeyboardInterrupt:
272
 
            raise
273
 
        except Exception, e:
274
 
            ## import pdb; pdb.set_trace()
275
 
            warning('Unable to load plugin %r from %r'
276
 
                    % (name, zip_name))
277
 
            log_exception_quietly()
 
250
                trace.warning('Unable to load plugin %r from %r' % (name, d))
 
251
            trace.log_exception_quietly()
 
252
            if 'error' in debug.debug_flags:
 
253
                trace.print_exception(sys.exc_info(), sys.stderr)
278
254
 
279
255
 
280
256
def plugins():
281
257
    """Return a dictionary of the plugins.
282
 
    
 
258
 
283
259
    Each item in the dictionary is a PlugIn object.
284
260
    """
285
261
    result = {}
327
303
        """
328
304
        self.module = module
329
305
 
330
 
    def get_help_text(self, additional_see_also=None):
 
306
    def get_help_text(self, additional_see_also=None, verbose=True):
331
307
        """Return a string with the help for this topic.
332
308
 
333
309
        :param additional_see_also: Additional help topics to be
339
315
            result = self.module.__doc__
340
316
        if result[-1] != '\n':
341
317
            result += '\n'
342
 
        # there is code duplicated here and in bzrlib/help_topic.py's 
 
318
        # there is code duplicated here and in bzrlib/help_topic.py's
343
319
        # matching Topic code. This should probably be factored in
344
320
        # to a helper function and a common base class.
345
321
        if additional_see_also is not None:
373
349
        if getattr(self.module, '__path__', None) is not None:
374
350
            return os.path.abspath(self.module.__path__[0])
375
351
        elif getattr(self.module, '__file__', None) is not None:
376
 
            return os.path.abspath(self.module.__file__)
 
352
            path = os.path.abspath(self.module.__file__)
 
353
            if path[-4:] in ('.pyc', '.pyo'):
 
354
                pypath = path[:-4] + '.py'
 
355
                if os.path.isfile(pypath):
 
356
                    path = pypath
 
357
            return path
377
358
        else:
378
359
            return repr(self.module)
379
360
 
391
372
        else:
392
373
            return None
393
374
 
 
375
    def load_plugin_tests(self, loader):
 
376
        """Return the adapted plugin's test suite.
 
377
 
 
378
        :param loader: The custom loader that should be used to load additional
 
379
            tests.
 
380
 
 
381
        """
 
382
        if getattr(self.module, 'load_tests', None) is not None:
 
383
            return loader.loadTestsFromModule(self.module)
 
384
        else:
 
385
            return None
 
386
 
394
387
    def version_info(self):
395
388
        """Return the plugin's version_tuple or None if unknown."""
396
389
        version_info = getattr(self.module, 'version_info', None)
397
 
        if version_info is not None and len(version_info) == 3:
398
 
            version_info = tuple(version_info) + ('final', 0)
 
390
        if version_info is not None:
 
391
            try:
 
392
                if isinstance(version_info, types.StringType):
 
393
                    version_info = version_info.split('.')
 
394
                elif len(version_info) == 3:
 
395
                    version_info = tuple(version_info) + ('final', 0)
 
396
            except TypeError, e:
 
397
                # The given version_info isn't even iteratible
 
398
                trace.log_exception_quietly()
 
399
                version_info = (version_info,)
399
400
        return version_info
400
 
    
 
401
 
401
402
    def _get__version__(self):
402
403
        version_info = self.version_info()
403
 
        if version_info is None:
 
404
        if version_info is None or len(version_info) == 0:
404
405
            return "unknown"
405
 
        if version_info[3] == 'final':
406
 
            version_string = '%d.%d.%d' % version_info[:3]
407
 
        else:
408
 
            version_string = '%d.%d.%d%s%d' % version_info
 
406
        try:
 
407
            version_string = _format_version_tuple(version_info)
 
408
        except (ValueError, TypeError, IndexError), e:
 
409
            trace.log_exception_quietly()
 
410
            # try to return something usefull for bad plugins, in stead of
 
411
            # stack tracing.
 
412
            version_string = '.'.join(map(str, version_info))
409
413
        return version_string
410
414
 
411
415
    __version__ = property(_get__version__)