~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/plugin.py

  • Committer: Aaron Bentley
  • Date: 2007-07-17 13:27:14 UTC
  • mfrom: (2624 +trunk)
  • mto: This revision was merged to the branch mainline in revision 2631.
  • Revision ID: abentley@panoramicfeedback.com-20070717132714-tmzx9khmg9501k51
Merge from bzr.dev

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,
 
45
    plugins,
51
46
    )
52
 
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.trace import mutter, warning, log_exception_quietly
59
50
 
60
51
 
61
52
DEFAULT_PLUGIN_PATH = None
62
53
_loaded = False
63
54
 
64
 
@deprecated_function(deprecated_in((2, 0, 0)))
65
55
def get_default_plugin_path():
66
56
    """Get the DEFAULT_PLUGIN_PATH"""
67
57
    global DEFAULT_PLUGIN_PATH
70
60
    return DEFAULT_PLUGIN_PATH
71
61
 
72
62
 
 
63
def all_plugins():
 
64
    """Return a dictionary of the plugins."""
 
65
    result = {}
 
66
    for name, plugin in plugins.__dict__.items():
 
67
        if isinstance(plugin, types.ModuleType):
 
68
            result[name] = plugin
 
69
    return result
 
70
 
 
71
 
73
72
def disable_plugins():
74
73
    """Disable loading plugins.
75
74
 
76
75
    Future calls to load_plugins() will be ignored.
77
76
    """
78
 
    load_plugins([])
79
 
 
80
 
 
81
 
def _strip_trailing_sep(path):
82
 
    return path.rstrip("\\/")
83
 
 
84
 
 
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
 
77
    # TODO: jam 20060131 This should probably also disable
 
78
    #       load_from_dirs()
 
79
    global _loaded
 
80
    _loaded = True
 
81
 
 
82
 
 
83
def set_plugins_path():
 
84
    """Set the path for plugins to be loaded from."""
 
85
    path = os.environ.get('BZR_PLUGIN_PATH',
 
86
                          get_default_plugin_path()).split(os.pathsep)
 
87
    # search the plugin path before the bzrlib installed dir
 
88
    path.append(os.path.dirname(plugins.__file__))
 
89
    plugins.__path__ = path
95
90
    return path
96
91
 
97
92
 
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
107
 
    bzr_exe = bool(getattr(sys, 'frozen', None))
108
 
    if bzr_exe:    # expand path for bzr.exe
109
 
        # We need to use relative path to system-wide plugin
110
 
        # directory because bzrlib from standalone bzr.exe
111
 
        # could be imported by another standalone program
112
 
        # (e.g. bzr-config; or TortoiseBzr/Olive if/when they
113
 
        # will become standalone exe). [bialix 20071123]
114
 
        # __file__ typically is
115
 
        # C:\Program Files\Bazaar\lib\library.zip\bzrlib\plugin.pyc
116
 
        # then plugins directory is
117
 
        # C:\Program Files\Bazaar\plugins
118
 
        # 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
 
 
192
 
    # Get rid of trailing slashes, since Python can't handle them when
193
 
    # it tries to import modules.
194
 
    paths = map(_strip_trailing_sep, paths)
195
 
    return paths
196
 
 
197
 
 
198
 
def load_plugins(path=None):
 
93
def load_plugins():
199
94
    """Load bzrlib plugins.
200
95
 
201
96
    The environment variable BZR_PLUGIN_PATH is considered a delimited
205
100
 
206
101
    load_from_dirs() provides the underlying mechanism and is called with
207
102
    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
103
    """
213
104
    global _loaded
214
105
    if _loaded:
217
108
    _loaded = True
218
109
 
219
110
    # scan for all plugins in the path.
220
 
    load_from_path(set_plugins_path(path))
 
111
    load_from_path(set_plugins_path())
221
112
 
222
113
 
223
114
def load_from_path(dirs):
232
123
 
233
124
    The python module path for bzrlib.plugins will be modified to be 'dirs'.
234
125
    """
235
 
    # We need to strip the trailing separators here as well as in the
236
 
    # set_plugins_path function because calling code can pass anything in to
237
 
    # this function, and since it sets plugins.__path__, it should set it to
238
 
    # something that will be valid for Python to use (in case people try to
239
 
    # run "import bzrlib.plugins.PLUGINNAME" after calling this function).
240
 
    _mod_plugins.__path__ = map(_strip_trailing_sep, dirs)
 
126
    plugins.__path__ = dirs
241
127
    for d in dirs:
242
128
        if not d:
243
129
            continue
244
 
        trace.mutter('looking for plugins in %s', d)
 
130
        mutter('looking for plugins in %s', d)
245
131
        if os.path.isdir(d):
246
132
            load_from_dir(d)
 
133
        else:
 
134
            # it might be a zip: try loading from the zip.
 
135
            load_from_zip(d)
 
136
            continue
247
137
 
248
138
 
249
139
# backwards compatability: load_from_dirs was the old name
252
142
 
253
143
 
254
144
def load_from_dir(d):
255
 
    """Load the plugins in directory d.
256
 
 
257
 
    d must be in the plugins module path already.
258
 
    """
 
145
    """Load the plugins in directory d."""
259
146
    # Get the list of valid python suffixes for __init__.py?
260
147
    # this includes .py, .pyc, and .pyo (depending on if we are running -O)
261
148
    # but it doesn't include compiled modules (.so, .dll, etc)
282
169
                    break
283
170
            else:
284
171
                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)
 
172
        if getattr(plugins, f, None):
 
173
            mutter('Plugin name %s already loaded', f)
289
174
        else:
290
 
            # trace.mutter('add plugin name %s', f)
 
175
            # mutter('add plugin name %s', f)
291
176
            plugin_names.add(f)
292
 
 
 
177
    
293
178
    for name in plugin_names:
294
179
        try:
295
180
            exec "import bzrlib.plugins.%s" % name in {}
296
181
        except KeyboardInterrupt:
297
182
            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
183
        except Exception, e:
304
 
            trace.warning("%s" % e)
305
184
            ## import pdb; pdb.set_trace()
306
185
            if re.search('\.|-| ', name):
307
 
                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))
 
186
                warning('Unable to load plugin %r from %r: '
 
187
                    'It is not a valid python module name.' % (name, d))
313
188
            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)
318
 
 
319
 
 
320
 
def plugins():
321
 
    """Return a dictionary of the plugins.
322
 
 
323
 
    Each item in the dictionary is a PlugIn object.
324
 
    """
325
 
    result = {}
326
 
    for name, plugin in _mod_plugins.__dict__.items():
327
 
        if isinstance(plugin, types.ModuleType):
328
 
            result[name] = PlugIn(name, plugin)
329
 
    return result
 
189
                warning('Unable to load plugin %r from %r' % (name, d))
 
190
            log_exception_quietly()
 
191
 
 
192
 
 
193
def load_from_zip(zip_name):
 
194
    """Load all the plugins in a zip."""
 
195
    valid_suffixes = ('.py', '.pyc', '.pyo')    # only python modules/packages
 
196
                                                # is allowed
 
197
 
 
198
    try:
 
199
        index = zip_name.rindex('.zip')
 
200
    except ValueError:
 
201
        return
 
202
    archive = zip_name[:index+4]
 
203
    prefix = zip_name[index+5:]
 
204
 
 
205
    mutter('Looking for plugins in %r', zip_name)
 
206
 
 
207
    # use zipfile to get list of files/dirs inside zip
 
208
    try:
 
209
        z = zipfile.ZipFile(archive)
 
210
        namelist = z.namelist()
 
211
        z.close()
 
212
    except zipfile.error:
 
213
        # not a valid zip
 
214
        return
 
215
 
 
216
    if prefix:
 
217
        prefix = prefix.replace('\\','/')
 
218
        if prefix[-1] != '/':
 
219
            prefix += '/'
 
220
        ix = len(prefix)
 
221
        namelist = [name[ix:]
 
222
                    for name in namelist
 
223
                    if name.startswith(prefix)]
 
224
 
 
225
    mutter('Names in archive: %r', namelist)
 
226
    
 
227
    for name in namelist:
 
228
        if not name or name.endswith('/'):
 
229
            continue
 
230
    
 
231
        # '/' is used to separate pathname components inside zip archives
 
232
        ix = name.rfind('/')
 
233
        if ix == -1:
 
234
            head, tail = '', name
 
235
        else:
 
236
            head, tail = name.rsplit('/',1)
 
237
        if '/' in head:
 
238
            # we don't need looking in subdirectories
 
239
            continue
 
240
    
 
241
        base, suffix = osutils.splitext(tail)
 
242
        if suffix not in valid_suffixes:
 
243
            continue
 
244
    
 
245
        if base == '__init__':
 
246
            # package
 
247
            plugin_name = head
 
248
        elif head == '':
 
249
            # module
 
250
            plugin_name = base
 
251
        else:
 
252
            continue
 
253
    
 
254
        if not plugin_name:
 
255
            continue
 
256
        if getattr(plugins, plugin_name, None):
 
257
            mutter('Plugin name %s already loaded', plugin_name)
 
258
            continue
 
259
    
 
260
        try:
 
261
            exec "import bzrlib.plugins.%s" % plugin_name in {}
 
262
            mutter('Load plugin %s from zip %r', plugin_name, zip_name)
 
263
        except KeyboardInterrupt:
 
264
            raise
 
265
        except Exception, e:
 
266
            ## import pdb; pdb.set_trace()
 
267
            warning('Unable to load plugin %r from %r'
 
268
                    % (name, zip_name))
 
269
            log_exception_quietly()
330
270
 
331
271
 
332
272
class PluginsHelpIndex(object):
367
307
        """
368
308
        self.module = module
369
309
 
370
 
    def get_help_text(self, additional_see_also=None, verbose=True):
 
310
    def get_help_text(self, additional_see_also=None):
371
311
        """Return a string with the help for this topic.
372
312
 
373
313
        :param additional_see_also: Additional help topics to be
379
319
            result = self.module.__doc__
380
320
        if result[-1] != '\n':
381
321
            result += '\n'
382
 
        # there is code duplicated here and in bzrlib/help_topic.py's
 
322
        # there is code duplicated here and in bzrlib/help_topic.py's 
383
323
        # matching Topic code. This should probably be factored in
384
324
        # to a helper function and a common base class.
385
325
        if additional_see_also is not None:
395
335
    def get_help_topic(self):
396
336
        """Return the modules help topic - its __name__ after bzrlib.plugins.."""
397
337
        return self.module.__name__[len('bzrlib.plugins.'):]
398
 
 
399
 
 
400
 
class PlugIn(object):
401
 
    """The bzrlib representation of a plugin.
402
 
 
403
 
    The PlugIn object provides a way to manipulate a given plugin module.
404
 
    """
405
 
 
406
 
    def __init__(self, name, module):
407
 
        """Construct a plugin for module."""
408
 
        self.name = name
409
 
        self.module = module
410
 
 
411
 
    def path(self):
412
 
        """Get the path that this plugin was loaded from."""
413
 
        if getattr(self.module, '__path__', None) is not None:
414
 
            return os.path.abspath(self.module.__path__[0])
415
 
        elif getattr(self.module, '__file__', None) is not None:
416
 
            path = os.path.abspath(self.module.__file__)
417
 
            if path[-4:] in ('.pyc', '.pyo'):
418
 
                pypath = path[:-4] + '.py'
419
 
                if os.path.isfile(pypath):
420
 
                    path = pypath
421
 
            return path
422
 
        else:
423
 
            return repr(self.module)
424
 
 
425
 
    def __str__(self):
426
 
        return "<%s.%s object at %s, name=%s, module=%s>" % (
427
 
            self.__class__.__module__, self.__class__.__name__, id(self),
428
 
            self.name, self.module)
429
 
 
430
 
    __repr__ = __str__
431
 
 
432
 
    def test_suite(self):
433
 
        """Return the plugin's test suite."""
434
 
        if getattr(self.module, 'test_suite', None) is not None:
435
 
            return self.module.test_suite()
436
 
        else:
437
 
            return None
438
 
 
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
 
    def version_info(self):
452
 
        """Return the plugin's version_tuple or None if unknown."""
453
 
        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,)
464
 
        return version_info
465
 
 
466
 
    def _get__version__(self):
467
 
        version_info = self.version_info()
468
 
        if version_info is None or len(version_info) == 0:
469
 
            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))
477
 
        return version_string
478
 
 
479
 
    __version__ = property(_get__version__)