← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/plugin.py

  • Committer: Martin Pool
  • Date: 2005-06-22 06:37:43 UTC
  • Revision ID: mbp@sourcefrog.net-20050622063743-e395f04c4db8977f
- move old blackbox code from testbzr into bzrlib.selftest.blackbox

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/plugin.py
22
22
# commands, import bzrlib.commands and add your new command to the
23
23
# plugin_cmds variable.
24
24
 
25
 
 
26
 
import os
27
 
from bzrlib.osutils import config_dir
28
 
DEFAULT_PLUGIN_PATH = os.path.join(config_dir(), 'plugins')
29
 
 
30
 
all_plugins = []
31
 
_loaded = False
 
25
import sys, os, imp
 
26
try:
 
27
    set
 
28
except NameError:
 
29
    from sets import Set as set
 
30
from bzrlib.trace import log_error
32
31
 
33
32
 
34
33
def load_plugins():
35
 
    """
36
 
    Find all python plugins and load them.
37
 
 
38
 
    Loading a plugin means importing it into the python interpreter.
39
 
    The plugin is expected to make calls to register commands when
40
 
    it's loaded (or perhaps access other hooks in future.)
41
 
 
42
 
    A list of plugs is stored in bzrlib.plugin.all_plugins for future
43
 
    reference.
44
 
 
45
 
    The environment variable BZR_PLUGIN_PATH is considered a delimited
46
 
    set of paths to look through. Each entry is searched for *.py
47
 
    files (and whatever other extensions are used in the platform,
48
 
    such as *.pyd).
49
 
    """
50
 
 
51
 
    global all_plugins, _loaded
52
 
    if _loaded:
53
 
        # People can make sure plugins are loaded, they just won't be twice
54
 
        return
55
 
        #raise BzrError("plugins already initialized")
56
 
    _loaded = True
57
 
 
58
 
    import sys, os, imp
59
 
    try:
60
 
        set
61
 
    except NameError:
62
 
        from sets import Set as set         # python2.3
63
 
 
64
 
    from bzrlib.trace import log_error, mutter, log_exception
65
 
    from bzrlib.errors import BzrError
66
 
 
67
 
    bzrpath = os.environ.get('BZR_PLUGIN_PATH', DEFAULT_PLUGIN_PATH)
 
34
    """Find all python files which are plugins, and load them
 
35
 
 
36
    The environment variable BZR_PLUGIN_PATH is considered a delimited set of
 
37
    paths to look through. Each entry is searched for *.py files (and whatever
 
38
    other extensions are used in the platform, such as *.pyd).
 
39
    """
 
40
    bzrpath = os.environ.get('BZR_PLUGIN_PATH', os.path.expanduser('~/.bzr/plugins'))
68
41
 
69
42
    # The problem with imp.get_suffixes() is that it doesn't include
70
43
    # .pyo which is technically valid
75
48
    suffixes.append(('.pyo', 'rb', imp.PY_COMPILED))
76
49
    package_entries = ['__init__.py', '__init__.pyc', '__init__.pyo']
77
50
    for d in bzrpath.split(os.pathsep):
78
 
        # going through them one by one allows different plugins with the same
 
51
        # going trough them one by one allows different plugins with the same
79
52
        # filename in different directories in the path
80
 
        mutter('looking for plugins in %s' % d)
81
53
        if not d:
82
54
            continue
83
55
        plugin_names = set()
102
74
                        break
103
75
                else:
104
76
                    continue
105
 
            mutter('add plugin name %s' % f)
106
77
            plugin_names.add(f)
107
78
 
108
79
        plugin_names = list(plugin_names)
110
81
        for name in plugin_names:
111
82
            try:
112
83
                plugin_info = imp.find_module(name, [d])
113
 
                mutter('load plugin %r' % (plugin_info,))
114
84
                try:
115
85
                    plugin = imp.load_module('bzrlib.plugin.' + name,
116
86
                                             *plugin_info)
117
 
                    all_plugins.append(plugin_info)
118
87
                finally:
119
88
                    if plugin_info[0] is not None:
120
89
                        plugin_info[0].close()
121
90
            except Exception, e:
122
 
                log_error('Unable to load plugin %r from %r' % (name, d))
123
 
                log_error(str(e))
124
 
                log_exception()
 
91
                log_error('Unable to load plugin: %r from %r\n%s' % (name, d, e))
125
92