~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/plugin.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2008-03-16 14:01:20 UTC
  • mfrom: (3280.2.5 integration)
  • Revision ID: pqm@pqm.ubuntu.com-20080316140120-i3yq8yr1l66m11h7
Start 1.4 development

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2004, 2005, 2007, 2008 Canonical Ltd
 
1
# Copyright (C) 2004, 2005, 2007 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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  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.
24
 
 
25
 
See the plugin-api developer documentation for information about writing
26
 
plugins.
 
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.
27
26
 
28
27
BZR_PLUGIN_PATH is also honoured for any plugins imported via
29
 
'import bzrlib.plugins.PLUGINNAME', as long as set_plugins_path has been
 
28
'import bzrlib.plugins.PLUGINNAME', as long as set_plugins_path has been 
30
29
called.
31
30
"""
32
31
 
33
32
import os
34
33
import sys
35
34
 
36
 
from bzrlib import osutils
37
 
 
38
35
from bzrlib.lazy_import import lazy_import
39
 
 
40
36
lazy_import(globals(), """
41
37
import imp
42
38
import re
43
39
import types
 
40
import zipfile
44
41
 
45
42
from bzrlib import (
46
 
    _format_version_tuple,
47
43
    config,
48
 
    debug,
49
 
    errors,
50
 
    trace,
 
44
    osutils,
51
45
    )
52
46
from bzrlib import plugins as _mod_plugins
53
47
""")
54
48
 
55
 
from bzrlib.symbol_versioning import (
56
 
    deprecated_function,
57
 
    deprecated_in,
58
 
    )
 
49
from bzrlib.symbol_versioning import deprecated_function, one_three
 
50
from bzrlib.trace import mutter, warning, log_exception_quietly
59
51
 
60
52
 
61
53
DEFAULT_PLUGIN_PATH = None
62
54
_loaded = False
63
55
 
64
 
@deprecated_function(deprecated_in((2, 0, 0)))
65
56
def get_default_plugin_path():
66
57
    """Get the DEFAULT_PLUGIN_PATH"""
67
58
    global DEFAULT_PLUGIN_PATH
75
66
 
76
67
    Future calls to load_plugins() will be ignored.
77
68
    """
78
 
    load_plugins([])
 
69
    # TODO: jam 20060131 This should probably also disable
 
70
    #       load_from_dirs()
 
71
    global _loaded
 
72
    _loaded = True
79
73
 
80
74
 
81
75
def _strip_trailing_sep(path):
82
76
    return path.rstrip("\\/")
83
77
 
84
78
 
85
 
def set_plugins_path(path=None):
86
 
    """Set the path for plugins to be loaded from.
87
 
 
88
 
    :param path: The list of paths to search for plugins.  By default,
89
 
        path will be determined using get_standard_plugins_path.
90
 
        if path is [], no plugins can be loaded.
91
 
    """
92
 
    if path is None:
93
 
        path = get_standard_plugins_path()
94
 
    _mod_plugins.__path__ = path
95
 
    return path
96
 
 
97
 
 
98
 
def _append_new_path(paths, new_path):
99
 
    """Append a new path if it set and not already known."""
100
 
    if new_path is not None and new_path not in paths:
101
 
        paths.append(new_path)
102
 
    return paths
103
 
 
104
 
 
105
 
def get_core_plugin_path():
106
 
    core_path = None
 
79
def set_plugins_path():
 
80
    """Set the path for plugins to be loaded from."""
 
81
    path = os.environ.get('BZR_PLUGIN_PATH',
 
82
                          get_default_plugin_path()).split(os.pathsep)
107
83
    bzr_exe = bool(getattr(sys, 'frozen', None))
108
84
    if bzr_exe:    # expand path for bzr.exe
109
85
        # We need to use relative path to system-wide plugin
116
92
        # then plugins directory is
117
93
        # C:\Program Files\Bazaar\plugins
118
94
        # so relative path is ../../../plugins
119
 
        core_path = osutils.abspath(osutils.pathjoin(
120
 
                osutils.dirname(__file__), '../../../plugins'))
121
 
    else:     # don't look inside library.zip
122
 
        # search the plugin path before the bzrlib installed dir
123
 
        core_path = os.path.dirname(_mod_plugins.__file__)
124
 
    return core_path
125
 
 
126
 
 
127
 
def get_site_plugin_path():
128
 
    """Returns the path for the site installed plugins."""
129
 
    if sys.platform == 'win32':
130
 
        # We don't have (yet) a good answer for windows since that is certainly
131
 
        # related to the way we build the installers. -- vila20090821
132
 
        return None
133
 
    site_path = None
134
 
    try:
135
 
        from distutils.sysconfig import get_python_lib
136
 
    except ImportError:
137
 
        # If distutuils is not available, we just don't know where they are
138
 
        pass
139
 
    else:
140
 
        site_path = osutils.pathjoin(get_python_lib(), 'bzrlib', 'plugins')
141
 
    return site_path
142
 
 
143
 
 
144
 
def get_user_plugin_path():
145
 
    return osutils.pathjoin(config.config_dir(), 'plugins')
146
 
 
147
 
 
148
 
def get_standard_plugins_path():
149
 
    """Determine a plugin path suitable for general use."""
150
 
    # Ad-Hoc default: core is not overriden by site but user can overrides both
151
 
    # The rationale is that:
152
 
    # - 'site' comes last, because these plugins should always be available and
153
 
    #   are supposed to be in sync with the bzr installed on site.
154
 
    # - 'core' comes before 'site' so that running bzr from sources or a user
155
 
    #   installed version overrides the site version.
156
 
    # - 'user' comes first, because... user is always right.
157
 
    # - the above rules clearly defines which plugin version will be loaded if
158
 
    #   several exist. Yet, it is sometimes desirable to disable some directory
159
 
    #   so that a set of plugins is disabled as once. This can be done via
160
 
    #   -site, -core, -user.
161
 
 
162
 
    env_paths = os.environ.get('BZR_PLUGIN_PATH', '+user').split(os.pathsep)
163
 
    defaults = ['+core', '+site']
164
 
 
165
 
    # The predefined references
166
 
    refs = dict(core=get_core_plugin_path(),
167
 
                site=get_site_plugin_path(),
168
 
                user=get_user_plugin_path())
169
 
 
170
 
    # Unset paths that should be removed
171
 
    for k,v in refs.iteritems():
172
 
        removed = '-%s' % k
173
 
        # defaults can never mention removing paths as that will make it
174
 
        # impossible for the user to revoke these removals.
175
 
        if removed in env_paths:
176
 
            env_paths.remove(removed)
177
 
            refs[k] = None
178
 
 
179
 
    # Expand references
180
 
    paths = []
181
 
    for p in env_paths + defaults:
182
 
        if p.startswith('+'):
183
 
            # Resolve reference if they are known
184
 
            try:
185
 
                p = refs[p[1:]]
186
 
            except KeyError:
187
 
                # Leave them untouched otherwise, user may have paths starting
188
 
                # with '+'...
189
 
                pass
190
 
        _append_new_path(paths, p)
191
 
 
 
95
        path.append(osutils.abspath(osutils.pathjoin(
 
96
            osutils.dirname(__file__), '../../../plugins')))
192
97
    # Get rid of trailing slashes, since Python can't handle them when
193
98
    # it tries to import modules.
194
 
    paths = map(_strip_trailing_sep, paths)
195
 
    return paths
196
 
 
197
 
 
198
 
def load_plugins(path=None):
 
99
    path = map(_strip_trailing_sep, path)
 
100
    if not bzr_exe:     # don't look inside library.zip
 
101
        # search the plugin path before the bzrlib installed dir
 
102
        path.append(os.path.dirname(_mod_plugins.__file__))
 
103
    _mod_plugins.__path__ = path
 
104
    return path
 
105
 
 
106
 
 
107
def load_plugins():
199
108
    """Load bzrlib plugins.
200
109
 
201
110
    The environment variable BZR_PLUGIN_PATH is considered a delimited
205
114
 
206
115
    load_from_dirs() provides the underlying mechanism and is called with
207
116
    the default directory list to provide the normal behaviour.
208
 
 
209
 
    :param path: The list of paths to search for plugins.  By default,
210
 
        path will be determined using get_standard_plugins_path.
211
 
        if path is [], no plugins can be loaded.
212
117
    """
213
118
    global _loaded
214
119
    if _loaded:
217
122
    _loaded = True
218
123
 
219
124
    # scan for all plugins in the path.
220
 
    load_from_path(set_plugins_path(path))
 
125
    load_from_path(set_plugins_path())
221
126
 
222
127
 
223
128
def load_from_path(dirs):
241
146
    for d in dirs:
242
147
        if not d:
243
148
            continue
244
 
        trace.mutter('looking for plugins in %s', d)
 
149
        mutter('looking for plugins in %s', d)
245
150
        if os.path.isdir(d):
246
151
            load_from_dir(d)
247
152
 
252
157
 
253
158
 
254
159
def load_from_dir(d):
255
 
    """Load the plugins in directory d.
256
 
 
257
 
    d must be in the plugins module path already.
258
 
    """
 
160
    """Load the plugins in directory d."""
259
161
    # Get the list of valid python suffixes for __init__.py?
260
162
    # this includes .py, .pyc, and .pyo (depending on if we are running -O)
261
163
    # but it doesn't include compiled modules (.so, .dll, etc)
282
184
                    break
283
185
            else:
284
186
                continue
285
 
        if f == '__init__':
286
 
            continue # We don't load __init__.py again in the plugin dir
287
 
        elif getattr(_mod_plugins, f, None):
288
 
            trace.mutter('Plugin name %s already loaded', f)
 
187
        if getattr(_mod_plugins, f, None):
 
188
            mutter('Plugin name %s already loaded', f)
289
189
        else:
290
 
            # trace.mutter('add plugin name %s', f)
 
190
            # mutter('add plugin name %s', f)
291
191
            plugin_names.add(f)
292
 
 
 
192
    
293
193
    for name in plugin_names:
294
194
        try:
295
195
            exec "import bzrlib.plugins.%s" % name in {}
296
196
        except KeyboardInterrupt:
297
197
            raise
298
 
        except errors.IncompatibleAPI, e:
299
 
            trace.warning("Unable to load plugin %r. It requested API version "
300
 
                "%s of module %s but the minimum exported version is %s, and "
301
 
                "the maximum is %s" %
302
 
                (name, e.wanted, e.api, e.minimum, e.current))
303
198
        except Exception, e:
304
 
            trace.warning("%s" % e)
305
199
            ## import pdb; pdb.set_trace()
306
200
            if re.search('\.|-| ', name):
307
201
                sanitised_name = re.sub('[-. ]', '_', name)
308
 
                if sanitised_name.startswith('bzr_'):
309
 
                    sanitised_name = sanitised_name[len('bzr_'):]
310
 
                trace.warning("Unable to load %r in %r as a plugin because the "
311
 
                        "file path isn't a valid module name; try renaming "
312
 
                        "it to %r." % (name, d, sanitised_name))
 
202
                warning("Unable to load %r in %r as a plugin because file path"
 
203
                        " isn't a valid module name; try renaming it to %r."
 
204
                        % (name, d, sanitised_name))
313
205
            else:
314
 
                trace.warning('Unable to load plugin %r from %r' % (name, d))
315
 
            trace.log_exception_quietly()
316
 
            if 'error' in debug.debug_flags:
317
 
                trace.print_exception(sys.exc_info(), sys.stderr)
 
206
                warning('Unable to load plugin %r from %r' % (name, d))
 
207
            log_exception_quietly()
 
208
 
 
209
 
 
210
@deprecated_function(one_three)
 
211
def load_from_zip(zip_name):
 
212
    """Load all the plugins in a zip."""
 
213
    valid_suffixes = ('.py', '.pyc', '.pyo')    # only python modules/packages
 
214
                                                # is allowed
 
215
    try:
 
216
        index = zip_name.rindex('.zip')
 
217
    except ValueError:
 
218
        return
 
219
    archive = zip_name[:index+4]
 
220
    prefix = zip_name[index+5:]
 
221
 
 
222
    mutter('Looking for plugins in %r', zip_name)
 
223
 
 
224
    # use zipfile to get list of files/dirs inside zip
 
225
    try:
 
226
        z = zipfile.ZipFile(archive)
 
227
        namelist = z.namelist()
 
228
        z.close()
 
229
    except zipfile.error:
 
230
        # not a valid zip
 
231
        return
 
232
 
 
233
    if prefix:
 
234
        prefix = prefix.replace('\\','/')
 
235
        if prefix[-1] != '/':
 
236
            prefix += '/'
 
237
        ix = len(prefix)
 
238
        namelist = [name[ix:]
 
239
                    for name in namelist
 
240
                    if name.startswith(prefix)]
 
241
 
 
242
    mutter('Names in archive: %r', namelist)
 
243
    
 
244
    for name in namelist:
 
245
        if not name or name.endswith('/'):
 
246
            continue
 
247
    
 
248
        # '/' is used to separate pathname components inside zip archives
 
249
        ix = name.rfind('/')
 
250
        if ix == -1:
 
251
            head, tail = '', name
 
252
        else:
 
253
            head, tail = name.rsplit('/',1)
 
254
        if '/' in head:
 
255
            # we don't need looking in subdirectories
 
256
            continue
 
257
    
 
258
        base, suffix = osutils.splitext(tail)
 
259
        if suffix not in valid_suffixes:
 
260
            continue
 
261
    
 
262
        if base == '__init__':
 
263
            # package
 
264
            plugin_name = head
 
265
        elif head == '':
 
266
            # module
 
267
            plugin_name = base
 
268
        else:
 
269
            continue
 
270
    
 
271
        if not plugin_name:
 
272
            continue
 
273
        if getattr(_mod_plugins, plugin_name, None):
 
274
            mutter('Plugin name %s already loaded', plugin_name)
 
275
            continue
 
276
    
 
277
        try:
 
278
            exec "import bzrlib.plugins.%s" % plugin_name in {}
 
279
            mutter('Load plugin %s from zip %r', plugin_name, zip_name)
 
280
        except KeyboardInterrupt:
 
281
            raise
 
282
        except Exception, e:
 
283
            ## import pdb; pdb.set_trace()
 
284
            warning('Unable to load plugin %r from %r'
 
285
                    % (name, zip_name))
 
286
            log_exception_quietly()
318
287
 
319
288
 
320
289
def plugins():
321
290
    """Return a dictionary of the plugins.
322
 
 
 
291
    
323
292
    Each item in the dictionary is a PlugIn object.
324
293
    """
325
294
    result = {}
367
336
        """
368
337
        self.module = module
369
338
 
370
 
    def get_help_text(self, additional_see_also=None, verbose=True):
 
339
    def get_help_text(self, additional_see_also=None):
371
340
        """Return a string with the help for this topic.
372
341
 
373
342
        :param additional_see_also: Additional help topics to be
379
348
            result = self.module.__doc__
380
349
        if result[-1] != '\n':
381
350
            result += '\n'
382
 
        # there is code duplicated here and in bzrlib/help_topic.py's
 
351
        # there is code duplicated here and in bzrlib/help_topic.py's 
383
352
        # matching Topic code. This should probably be factored in
384
353
        # to a helper function and a common base class.
385
354
        if additional_see_also is not None:
436
405
        else:
437
406
            return None
438
407
 
439
 
    def load_plugin_tests(self, loader):
440
 
        """Return the adapted plugin's test suite.
441
 
 
442
 
        :param loader: The custom loader that should be used to load additional
443
 
            tests.
444
 
 
445
 
        """
446
 
        if getattr(self.module, 'load_tests', None) is not None:
447
 
            return loader.loadTestsFromModule(self.module)
448
 
        else:
449
 
            return None
450
 
 
451
408
    def version_info(self):
452
409
        """Return the plugin's version_tuple or None if unknown."""
453
410
        version_info = getattr(self.module, 'version_info', None)
454
 
        if version_info is not None:
455
 
            try:
456
 
                if isinstance(version_info, types.StringType):
457
 
                    version_info = version_info.split('.')
458
 
                elif len(version_info) == 3:
459
 
                    version_info = tuple(version_info) + ('final', 0)
460
 
            except TypeError, e:
461
 
                # The given version_info isn't even iteratible
462
 
                trace.log_exception_quietly()
463
 
                version_info = (version_info,)
 
411
        if version_info is not None and len(version_info) == 3:
 
412
            version_info = tuple(version_info) + ('final', 0)
464
413
        return version_info
465
 
 
 
414
    
466
415
    def _get__version__(self):
467
416
        version_info = self.version_info()
468
 
        if version_info is None or len(version_info) == 0:
 
417
        if version_info is None:
469
418
            return "unknown"
470
 
        try:
471
 
            version_string = _format_version_tuple(version_info)
472
 
        except (ValueError, TypeError, IndexError), e:
473
 
            trace.log_exception_quietly()
474
 
            # try to return something usefull for bad plugins, in stead of
475
 
            # stack tracing.
476
 
            version_string = '.'.join(map(str, version_info))
 
419
        if version_info[3] == 'final':
 
420
            version_string = '%d.%d.%d' % version_info[:3]
 
421
        else:
 
422
            version_string = '%d.%d.%d%s%d' % version_info
477
423
        return version_string
478
424
 
479
425
    __version__ = property(_get__version__)