← 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:18:01 UTC
  • Revision ID: mbp@sourcefrog.net-20050622071801-916bbae5bd93cf0a
- new mkdir command

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
 
    from bzrlib import plugins
63
 
 
64
 
    dirs = os.environ.get('BZR_PLUGIN_PATH', DEFAULT_PLUGIN_PATH).split(":")
65
 
    dirs.insert(0, os.path.dirname(plugins.__file__))
 
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)
66
46
 
67
47
    # The problem with imp.get_suffixes() is that it doesn't include
68
48
    # .pyo which is technically valid
72
52
    suffixes = imp.get_suffixes()
73
53
    suffixes.append(('.pyo', 'rb', imp.PY_COMPILED))
74
54
    package_entries = ['__init__.py', '__init__.pyc', '__init__.pyo']
75
 
    for d in dirs:
76
 
        # going through them one by one allows different plugins with the same
 
55
    for d in bzrpath.split(os.pathsep):
 
56
        # going trough them one by one allows different plugins with the same
77
57
        # filename in different directories in the path
78
 
        mutter('looking for plugins in %s' % d)
79
58
        if not d:
80
59
            continue
81
60
        plugin_names = set()
100
79
                        break
101
80
                else:
102
81
                    continue
103
 
            mutter('add plugin name %s' % f)
104
82
            plugin_names.add(f)
105
83
 
106
84
        plugin_names = list(plugin_names)
108
86
        for name in plugin_names:
109
87
            try:
110
88
                plugin_info = imp.find_module(name, [d])
111
 
                mutter('load plugin %r' % (plugin_info,))
112
89
                try:
113
90
                    plugin = imp.load_module('bzrlib.plugin.' + name,
114
91
                                             *plugin_info)
115
 
                    all_plugins.append(plugin)
116
92
                finally:
117
93
                    if plugin_info[0] is not None:
118
94
                        plugin_info[0].close()
119
95
            except Exception, e:
120
 
                log_error('Unable to load plugin %r from %r' % (name, d))
121
 
                log_error(str(e))
122
 
                log_exception()
 
96
                log_error('Unable to load plugin: %r from %r\n%s' % (name, d, e))
123
97