~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/plugin.py

  • Committer: Ian Clatworthy
  • Date: 2007-12-07 04:21:59 UTC
  • mto: This revision was merged to the branch mainline in revision 3092.
  • Revision ID: ian.clatworthy@internode.on.net-20071207042159-n9rmhanqid1l7olh
Better PDF for Qiock Start Card (Ian Clatworthy)

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, zero_ninetyone
 
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
68
59
    if DEFAULT_PLUGIN_PATH is None:
69
 
        DEFAULT_PLUGIN_PATH = osutils.pathjoin(config.config_dir(), 'plugins')
 
60
        path = [osutils.pathjoin(config.config_dir(), 'plugins')]
 
61
        if getattr(sys, 'frozen', None):    # bzr.exe
 
62
            # We need to use relative path to system-wide plugin
 
63
            # directory because bzrlib from standalone bzr.exe
 
64
            # could be imported by another standalone program
 
65
            # (e.g. bzr-config; or TortoiseBzr/Olive if/when they
 
66
            # will become standalone exe). [bialix 20071123]
 
67
            # __file__ typically is
 
68
            # C:\Program Files\Bazaar\lib\library.zip\bzrlib\plugin.pyc
 
69
            # then plugins directory is
 
70
            # C:\Program Files\Bazaar\plugins
 
71
            # so relative path is ../../../plugins
 
72
            path.append(osutils.abspath(osutils.pathjoin(
 
73
                osutils.dirname(__file__), '../../../plugins')))
 
74
        DEFAULT_PLUGIN_PATH = os.pathsep.join(path)
70
75
    return DEFAULT_PLUGIN_PATH
71
76
 
72
77
 
 
78
@deprecated_function(zero_ninetyone)
 
79
def all_plugins():
 
80
    """Return a dictionary of the plugins."""
 
81
    return dict((name, plugin.module) for name, plugin in plugins().items())
 
82
 
 
83
 
73
84
def disable_plugins():
74
85
    """Disable loading plugins.
75
86
 
76
87
    Future calls to load_plugins() will be ignored.
77
88
    """
78
 
    load_plugins([])
 
89
    # TODO: jam 20060131 This should probably also disable
 
90
    #       load_from_dirs()
 
91
    global _loaded
 
92
    _loaded = True
79
93
 
80
94
 
81
95
def _strip_trailing_sep(path):
82
96
    return path.rstrip("\\/")
83
97
 
84
98
 
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()
 
99
def set_plugins_path():
 
100
    """Set the path for plugins to be loaded from."""
 
101
    path = os.environ.get('BZR_PLUGIN_PATH',
 
102
                          get_default_plugin_path()).split(os.pathsep)
 
103
    # Get rid of trailing slashes, since Python can't handle them when
 
104
    # it tries to import modules.
 
105
    path = map(_strip_trailing_sep, path)
 
106
    # search the plugin path before the bzrlib installed dir
 
107
    path.append(os.path.dirname(_mod_plugins.__file__))
94
108
    _mod_plugins.__path__ = path
95
109
    return path
96
110
 
97
111
 
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):
 
112
def load_plugins():
199
113
    """Load bzrlib plugins.
200
114
 
201
115
    The environment variable BZR_PLUGIN_PATH is considered a delimited
205
119
 
206
120
    load_from_dirs() provides the underlying mechanism and is called with
207
121
    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
122
    """
213
123
    global _loaded
214
124
    if _loaded:
217
127
    _loaded = True
218
128
 
219
129
    # scan for all plugins in the path.
220
 
    load_from_path(set_plugins_path(path))
 
130
    load_from_path(set_plugins_path())
221
131
 
222
132
 
223
133
def load_from_path(dirs):
241
151
    for d in dirs:
242
152
        if not d:
243
153
            continue
244
 
        trace.mutter('looking for plugins in %s', d)
 
154
        mutter('looking for plugins in %s', d)
245
155
        if os.path.isdir(d):
246
156
            load_from_dir(d)
 
157
        else:
 
158
            # it might be a zip: try loading from the zip.
 
159
            load_from_zip(d)
247
160
 
248
161
 
249
162
# backwards compatability: load_from_dirs was the old name
252
165
 
253
166
 
254
167
def load_from_dir(d):
255
 
    """Load the plugins in directory d.
256
 
 
257
 
    d must be in the plugins module path already.
258
 
    """
 
168
    """Load the plugins in directory d."""
259
169
    # Get the list of valid python suffixes for __init__.py?
260
170
    # this includes .py, .pyc, and .pyo (depending on if we are running -O)
261
171
    # but it doesn't include compiled modules (.so, .dll, etc)
282
192
                    break
283
193
            else:
284
194
                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)
 
195
        if getattr(_mod_plugins, f, None):
 
196
            mutter('Plugin name %s already loaded', f)
289
197
        else:
290
 
            # trace.mutter('add plugin name %s', f)
 
198
            # mutter('add plugin name %s', f)
291
199
            plugin_names.add(f)
292
 
 
 
200
    
293
201
    for name in plugin_names:
294
202
        try:
295
203
            exec "import bzrlib.plugins.%s" % name in {}
296
204
        except KeyboardInterrupt:
297
205
            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
206
        except Exception, e:
304
 
            trace.warning("%s" % e)
305
207
            ## import pdb; pdb.set_trace()
306
208
            if re.search('\.|-| ', name):
307
209
                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))
 
210
                warning("Unable to load %r in %r as a plugin because file path"
 
211
                        " isn't a valid module name; try renaming it to %r."
 
212
                        % (name, d, sanitised_name))
313
213
            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)
 
214
                warning('Unable to load plugin %r from %r' % (name, d))
 
215
            log_exception_quietly()
 
216
 
 
217
 
 
218
def load_from_zip(zip_name):
 
219
    """Load all the plugins in a zip."""
 
220
    valid_suffixes = ('.py', '.pyc', '.pyo')    # only python modules/packages
 
221
                                                # is allowed
 
222
 
 
223
    try:
 
224
        index = zip_name.rindex('.zip')
 
225
    except ValueError:
 
226
        return
 
227
    archive = zip_name[:index+4]
 
228
    prefix = zip_name[index+5:]
 
229
 
 
230
    mutter('Looking for plugins in %r', zip_name)
 
231
 
 
232
    # use zipfile to get list of files/dirs inside zip
 
233
    try:
 
234
        z = zipfile.ZipFile(archive)
 
235
        namelist = z.namelist()
 
236
        z.close()
 
237
    except zipfile.error:
 
238
        # not a valid zip
 
239
        return
 
240
 
 
241
    if prefix:
 
242
        prefix = prefix.replace('\\','/')
 
243
        if prefix[-1] != '/':
 
244
            prefix += '/'
 
245
        ix = len(prefix)
 
246
        namelist = [name[ix:]
 
247
                    for name in namelist
 
248
                    if name.startswith(prefix)]
 
249
 
 
250
    mutter('Names in archive: %r', namelist)
 
251
    
 
252
    for name in namelist:
 
253
        if not name or name.endswith('/'):
 
254
            continue
 
255
    
 
256
        # '/' is used to separate pathname components inside zip archives
 
257
        ix = name.rfind('/')
 
258
        if ix == -1:
 
259
            head, tail = '', name
 
260
        else:
 
261
            head, tail = name.rsplit('/',1)
 
262
        if '/' in head:
 
263
            # we don't need looking in subdirectories
 
264
            continue
 
265
    
 
266
        base, suffix = osutils.splitext(tail)
 
267
        if suffix not in valid_suffixes:
 
268
            continue
 
269
    
 
270
        if base == '__init__':
 
271
            # package
 
272
            plugin_name = head
 
273
        elif head == '':
 
274
            # module
 
275
            plugin_name = base
 
276
        else:
 
277
            continue
 
278
    
 
279
        if not plugin_name:
 
280
            continue
 
281
        if getattr(_mod_plugins, plugin_name, None):
 
282
            mutter('Plugin name %s already loaded', plugin_name)
 
283
            continue
 
284
    
 
285
        try:
 
286
            exec "import bzrlib.plugins.%s" % plugin_name in {}
 
287
            mutter('Load plugin %s from zip %r', plugin_name, zip_name)
 
288
        except KeyboardInterrupt:
 
289
            raise
 
290
        except Exception, e:
 
291
            ## import pdb; pdb.set_trace()
 
292
            warning('Unable to load plugin %r from %r'
 
293
                    % (name, zip_name))
 
294
            log_exception_quietly()
318
295
 
319
296
 
320
297
def plugins():
321
298
    """Return a dictionary of the plugins.
322
 
 
 
299
    
323
300
    Each item in the dictionary is a PlugIn object.
324
301
    """
325
302
    result = {}
367
344
        """
368
345
        self.module = module
369
346
 
370
 
    def get_help_text(self, additional_see_also=None, verbose=True):
 
347
    def get_help_text(self, additional_see_also=None):
371
348
        """Return a string with the help for this topic.
372
349
 
373
350
        :param additional_see_also: Additional help topics to be
379
356
            result = self.module.__doc__
380
357
        if result[-1] != '\n':
381
358
            result += '\n'
382
 
        # there is code duplicated here and in bzrlib/help_topic.py's
 
359
        # there is code duplicated here and in bzrlib/help_topic.py's 
383
360
        # matching Topic code. This should probably be factored in
384
361
        # to a helper function and a common base class.
385
362
        if additional_see_also is not None:
413
390
        if getattr(self.module, '__path__', None) is not None:
414
391
            return os.path.abspath(self.module.__path__[0])
415
392
        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
 
393
            return os.path.abspath(self.module.__file__)
422
394
        else:
423
395
            return repr(self.module)
424
396
 
436
408
        else:
437
409
            return None
438
410
 
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
411
    def version_info(self):
452
412
        """Return the plugin's version_tuple or None if unknown."""
453
413
        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,)
 
414
        if version_info is not None and len(version_info) == 3:
 
415
            version_info = tuple(version_info) + ('final', 0)
464
416
        return version_info
465
 
 
 
417
    
466
418
    def _get__version__(self):
467
419
        version_info = self.version_info()
468
 
        if version_info is None or len(version_info) == 0:
 
420
        if version_info is None:
469
421
            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))
 
422
        if version_info[3] == 'final':
 
423
            version_string = '%d.%d.%d' % version_info[:3]
 
424
        else:
 
425
            version_string = '%d.%d.%d%s%d' % version_info
477
426
        return version_string
478
427
 
479
428
    __version__ = property(_get__version__)