← 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 07:30:45 UTC
  • Revision ID: mbp@sourcefrog.net-20050622073045-80d336b7e0b859aa
- redirect stdout/stderr while running tests

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         # python2.3
 
30
from bzrlib.trace import log_error
 
31
 
 
32
 
 
33
DEFAULT_PLUGIN_PATH = '~/.bzr.conf/plugins'
32
34
 
33
35
 
34
36
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
 
    
60
 
    from bzrlib.trace import log_error, mutter, log_exception
61
 
    from bzrlib.errors import BzrError
62
 
 
63
 
    bzrpath = os.environ.get('BZR_PLUGIN_PATH', DEFAULT_PLUGIN_PATH)
 
37
    """Find all python files which are plugins, and load them
 
38
 
 
39
    The environment variable BZR_PLUGIN_PATH is considered a delimited set of
 
40
    paths to look through. Each entry is searched for *.py files (and whatever
 
41
    other extensions are used in the platform, such as *.pyd).
 
42
    """
 
43
    bzrpath = os.environ.get('BZR_PLUGIN_PATH')
 
44
    if not bzrpath:
 
45
        bzrpath = os.path.expanduser(DEFAULT_PLUGIN_PATH)
64
46
 
65
47
    # The problem with imp.get_suffixes() is that it doesn't include
66
48
    # .pyo which is technically valid
71
53
    suffixes.append(('.pyo', 'rb', imp.PY_COMPILED))
72
54
    package_entries = ['__init__.py', '__init__.pyc', '__init__.pyo']
73
55
    for d in bzrpath.split(os.pathsep):
74
 
        # going through them one by one allows different plugins with the same
 
56
        # going trough them one by one allows different plugins with the same
75
57
        # filename in different directories in the path
76
 
        mutter('looking for plugins in %s' % d)
77
58
        if not d:
78
59
            continue
79
60
        plugin_names = set()
98
79
                        break
99
80
                else:
100
81
                    continue
101
 
            mutter('add plugin name %s' % f)
102
82
            plugin_names.add(f)
103
83
 
104
84
        plugin_names = list(plugin_names)
106
86
        for name in plugin_names:
107
87
            try:
108
88
                plugin_info = imp.find_module(name, [d])
109
 
                mutter('load plugin %r' % (plugin_info,))
110
89
                try:
111
90
                    plugin = imp.load_module('bzrlib.plugin.' + name,
112
91
                                             *plugin_info)
113
 
                    all_plugins.append(plugin)
114
92
                finally:
115
93
                    if plugin_info[0] is not None:
116
94
                        plugin_info[0].close()
117
95
            except Exception, e:
118
 
                log_error('Unable to load plugin %r from %r' % (name, d))
119
 
                log_error(str(e))
120
 
                log_exception()
 
96
                log_error('Unable to load plugin: %r from %r\n%s' % (name, d, e))
121
97